diff --git a/src/common/utils/selenium.py b/src/common/utils/selenium.py index 71bb04e..b67fb1b 100644 --- a/src/common/utils/selenium.py +++ b/src/common/utils/selenium.py @@ -52,7 +52,7 @@ def click(selector: str|WebElement, wait=True, condition=False): element = locate(selector, wait, predicate) if isinstance(selector, str) else selector counter = lambda: int(element.get_attribute(identity) or '0') value = counter() - driver.execute_script("window.__%s__ = () => { arguments[0].setAttribute('%s', arguments[1] + 1) }" % ((identity,) * 2), element, value) + driver.execute_script("window.__%s__ = () => { arguments[0].setAttribute('%s', arguments[1] + 1) };" % ((identity,) * 2), element, value) driver.execute_script("arguments[0].addEventListener('click', __%s__);" % identity, element) for _ in range(parameters.get('attempts', 0)): @@ -65,7 +65,7 @@ def click(selector: str|WebElement, wait=True, condition=False): error = True continue try: - WebDriverWait(driver, parameters.get('interval', 0)).until(lambda _: counter() > value) + if wait: WebDriverWait(driver, parameters.get('interval', 0)).until(lambda _: counter() > value) break except TimeoutException: continue except: break