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)
转载于:https://blog.51cto.com/12965114/2324729