Utilities/common
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix: 'click' wait argument

Browse Source
This commit is contained in:
break27
2026-06-03 15:38:51 +08:00
parent 996bda2298
commit 772a8eb368
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
src/common/utils/selenium.py
View File

@@ -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