1. 程式人生 > >WebDriverWait智慧等待查詢元素,提高程式碼的執行效率,不浪費時間,減少程式碼量

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)