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)