WebDriverWait智能等待查找元素,提高代码的执行效率,不浪费时间,减少代码量

from selenium.webdriver.support.wait import WebDriverWait
智能等待10s之后获取元素,获取的是单个元素
def find_element(self, locator):
WebDriverWait(self.driver, 10).until(lambda the_driver: the_driver.find_element(
locator))

    return self.driver.find_element(*locator)

# 智能等待10s之后获取元素,获取的是多个元素
def find_elements(self, *locator):
    WebDriverWait(self.driver, 10).until(lambda the_driver: len(the_driver.find_elements(*locator)) > 0)

猜你喜欢

转载自blog.51cto.com/12965114/2324729