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