selenium之python源码解读-WebDriverWait

一、显示等待

所谓显示等待，是针对某一个特定的元素设置等待时间，如果在规定的时间内找到了该元素，就执行相关的操作，如果在规定的时间内没有找到该元素，在抛出异常

PS:注意显示等待和隐身等待的区别，隐身等待是对页面中的所有元素设置加载时间。

二、WebDriverWait类

在selenium框架中显示等待是通过WebDriverWait来实现的

WebDriverWait所在位置：在support包的wait.py文件中，selenium\webdriver\support\wait.py

构成函数：

class WebDriverWait(object):
    def __init__(self, driver, timeout, poll_frequency=POLL_FREQUENCY, ignored_exceptions=None):
        pass

构造函数参数解读：

driver：webdriver实例，可以是Ie, Firefox, Chrome or Remote

timeout：查找特定元素时的最大超时时间，单位seconds

poll_frequency：关键字参数，查找元素的频率，即每隔多少时间回调一次method，默认为 0.5 second

ignored_exceptions：关键字参数，在回调method过程中忽略哪些异常，默认为NoSuchElementException，如果有需要忽略多个异常，可以以iterable的形式传入参数，如tuple

实例方法：

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

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

方法解读：

until：循环调用method的方法，直到method返回true()非false，返回结果：method方法返回的结果

until_not：循环调用method的方法，直到method返回false
在这两个方法中，参数methon是必传参数，且传入的是一个functhon对象(注意是函数对象，不带括号的)，该method对象的参数是driver

def until(self, method, message=''):
    screen = None
    stacktrace = None

    # 设置超时时间
    end_time = time.time() + self._timeout
    while True: # 循环调用method方法
        try:
            # 执行传入的methon方法，且参数为driver
            value = method(self._driver)
            if value: # 如果method方法执行的结果为非false，则停止循环
                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 selenium.webdriver.support.ui import WebDriverWait # 导入WebDriverWait
# method：参数为driver的lambda匿名函数
# 功能：找到id为someId的元素
# 返回值：x.find_element_by_id("someId")的WebElement对象
element = WebDriverWait(driver, 10).until(lambda x: x.find_element_by_id("someId"))

总结：

until / until_not方法

参数：method对象(注意是对象，而非调用方法)，可以传入lambda匿名函数

method方法的参数为driver