selenium之python源码解读-WebDriverWait

2018/04/30 21:58
阅读数 122

一、显示等待

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

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

 

展开阅读全文
加载中
点击引领话题📣 发布并加入讨论🔥
打赏
0 评论
0 收藏
0
分享
返回顶部
顶部