def loop(self, func): trigger = False tries = 100 while ((trigger == False) & (tries > 0)): tries -= 1 try: func #passed method or function executes here trigger = True except: print("loop failed ("+str(tries)+")") self.driver.save_screenshot(self.screenshot_name) trigger = False finally: if tries == 100: self.driver.close()