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)