显式等待until传入自定义方法

WebDriverWait(driver,10).until(expected_conditions.element_to_be_clickable(ele))

通过追踪代码，可以发现上面的element_to_be_clickable其实也是一个方法，只不过是selenium定义好的方法。

既然可以传方法，那其实也可以传入自定义的方法。

class element_to_be_clickable(object):
    """ An Expectation for checking an element is visible and enabled such that
    you can click it."""
    def __init__(self, locator):
        self.locator = locator

    def __call__(self, driver):
        element = visibility_of_element_located(self.locator)(driver)
        if element and element.is_enabled():
            return element
        else:
            return False

until方法源码

 def until(self, method, message=''):
        """Calls the method provided with the driver as an argument until the 
        return value is not False."""
        screen = None
        stacktrace = None

        end_time = time.time() + self._timeout
        while True:
            try:
                value = method(self._driver)
                if value:
                    return value
            except self._ignored_exceptions as exc:
                screen = getattr(exc, 'screen', None)
                stacktrace = getattr(exc, 'stacktrace', None)
            time.sleep(self._poll)
            if time.time() > end_time:
                break
        raise TimeoutException(message, screen, stacktrace)

举例

from time import sleep
from selenium import webdriver
from selenium.webdriver.chrome.options import Options
from selenium.webdriver.common.by import By
from selenium.webdriver.support.wait import WebDriverWait

class TestHome():
    def setup(self):
        # 使用Options，需要from selenium.webdriver.chrome.options import Options
        chromeOptions = Options()
        #9222是端口号，只要是不被占用的端口号都可以
        chromeOptions.add_experimental_option('debuggerAddress', '127.0.0.1:9222')
        # 使用webdriver，需要from selenium import webdriver
        self.driver = webdriver.Chrome(options=chromeOptions)
        # 隐式等待，是全局生效的
        self.driver.implicitly_wait(3)

    def teardown(self):
        sleep(5)  # from time import sleep
        self.driver.quit()

    def test_home(self):
        self.driver.find_element(By.ID, 'menu_contacts').click()  # 使用By，需要from selenium.webdriver.common.by import By
        # self.driver.find_element_by_id('menu_contacts').click()
        # sleep(2)
        # 显式等待
        # WebDriverWait(self.driver,10).until(expected_conditions.element_to_be_clickable((By.CSS_SELECTOR, '.js_has_member div:nth-child(1) .js_add_member')))

        # self.driver.find_element(By.CSS_SELECTOR, '.js_has_member div:nth-child(1) .js_add_member').click()

        # 循环去点击“添加成员”按钮
        # 在until方法种传入自定义方法wait_element
        WebDriverWait(self.driver, 10).until(self.wait_element)  # 不是WebDriverWait(self.driver,10).until(self.wait_element())

        '''
        while True:
            self.wait_element()
            
        '''
        self.driver.find_element(By.ID, 'username').send_keys('Tom')
        self.driver.find_element(By.ID, 'memberAdd_acctid').send_keys('Tom111')
        self.driver.find_element(By.ID, 'memberAdd_phone').send_keys('13838383388')
        self.driver.find_element(By.CSS_SELECTOR, '.js_btn_save').click()

    # 根据页面是否有“姓名”输入框，判断页面是否已经跳转到录入信息页面
    def wait_element(self, x):  # 复用until方法，所以得加x。
        size = len(self.driver.find_elements(By.ID, 'username'))  # 调用这个方法时，self.driver.find_elements也会受到隐式等待的影响
        if size < 1:
            self.driver.find_element(By.CSS_SELECTOR, '.js_has_member div:nth-child(1) .js_add_member').click()
        return size >= 1