fix: added type hints

This commit is contained in:
2026-08-14 13:57:54 +08:00
parent 31a3167cff
commit 1923d44542
+2 -1
View File
@@ -1,6 +1,7 @@
import random import random
import string import string
from typing import Callable
from selenium.webdriver.remote.webdriver import WebDriver from selenium.webdriver.remote.webdriver import WebDriver
from selenium.webdriver.remote.webelement import WebElement from selenium.webdriver.remote.webelement import WebElement
from selenium.webdriver.common.by import By from selenium.webdriver.common.by import By
@@ -11,7 +12,7 @@ from selenium.common.exceptions import StaleElementReferenceException, NoAlertPr
driver: WebDriver|None = None driver: WebDriver|None = None
unique, parameters = None, dict() unique, parameters = None, dict()
def until(condition, timeout=None): def until[T](condition: Callable[[WebDriver], T], timeout=None) -> T:
try: return WebDriverWait(driver, timeout or parameters.get('timeout') or 0).until(condition) try: return WebDriverWait(driver, timeout or parameters.get('timeout') or 0).until(condition)
except: pass except: pass