读者问题:selenium里的等待

682 阅读4分钟
原文链接: zhuanlan.zhihu.com

提问:selenium中,智能等待怎么整,网上查的看的有点儿晕

答:

1.直接Sleep:

例1.直接sleep指定秒数。

from selenium import webdriver
import time
time_start = time.time()
browser = webdriver.Firefox()
browser.get("http://www.baidu.com")
time.sleep(5) #写死的等待5秒
browser.find_element_by_id("kw").send_keys("test")
browser.find_element_by_id("su").click()
browser.close()
time_end = time.time()
print("一共 {}秒".format(time_end-time_start))


参考运行结果:

一共 11.42103099822998秒


这里花了11秒是因为打开 百度并做搜索一共花了6秒,然后中间的sleep等待了5秒。

这种方式是最简单的。不推荐使用,除非其他方式都不行。


2. implicit Wait隐式等待

例2. 隐式等待5秒

from selenium import webdriver
import time
time_start = time.time()
browser = webdriver.Firefox()
browser.implicitly_wait(5)
browser.get("http://www.baidu.com")
browser.find_element_by_id("kw").send_keys("test")
browser.find_element_by_id("su").click()
browser.close()
time_end = time.time()
print("一共 {}秒".format(time_end-time_start))


参考运行结果:

一共 6.454062223434448秒


这里的6秒正是打开百度并做搜索的时间。


这个隐式等待的概念来自 watir 这个框架。意思是当selenium 找不到某个元素而要抛出异常前会重新在dom对象中查找这个元素,直到达到隐式等待里指定的时间,也就是指定一个最长等待时间。

这个等待方式,对 webdriver 对象整个生命周期都生效。也就是只要指定一次,后续各种元素查找操作都会有这个最长等待时间。因此,这里 implicit_wait() 只调用了一次。


3. Explicit Wait 显式等待

例3. 显式等待10秒

from selenium import webdriver
from selenium.webdriver.common.by import By
from selenium.webdriver.support.ui import WebDriverWait
from selenium.webdriver.support import expected_conditions as EC
import time
time_start = time.time()


browser = webdriver.Firefox()
browser.get("https://www.baidu.com")
wait = WebDriverWait(browser, 10)  # 设置显示等待时间为10s
element = wait.until(EC.element_to_be_clickable((By.ID, 'kw')))  # 等待判断条件
element.send_keys("test")
element = wait.until(EC.element_to_be_clickable((By.ID, 'su')))  # 等待判断条件
element.click()
browser.close()
time_end = time.time()


print("一共 {}秒".format(time_end-time_start))


参考运行结果:

一共 6.216423034667969秒


这里我设置了显式等待时间为10秒,显式等待并不会真的等满10秒,而是在这10秒内取寻找元素,提前找到就提前结束等待。


显式等待的使用比较复杂,首先,需要定义一个WebDriverWait 对象,里面设置要等待的时间上限。 然后在这个对象上调用until方法,而until方法接受的参数是一个expected_conditions 对象,这个对象表示要等待的条件。 在上面例子中我们调用了 element_to_be_clickable 来判断元素是否可以点击,并把这个方法的返回值作为等待条件传给until 方法。


虽然显式复杂,但归根结底,用户要输入的就是:

1.要等待的元素的寻找方法和值,比如例子中的 By.ID 和 kw

2.要等待这个元素的具体条件,比如等到它可以点击 element_to_be_clickable

3.要等待的最大时间,比如WebDriverWait(browser, 10) 这里的10秒

4.要等待的webdriver 对象,比如WebDriverWait(browser, 10) 这里的browser


4. implicit Wait VS Explicit Wait

这两种方式对比,显然 隐式 比较简单,而 显式 比较灵活。实际工作中,隐式 并不能每次都等待成功,这是因为前端技术日新月异,动态的页面越来越多。实际工作中,这两种应该混合使用。


但除此之外,还更推荐一种做法:

自己封装一个包括等待操作的类和方法。


有些框架作者会这样做:

在selenium提供的find_element_by_xxx 基础上,自己封装一个 find_element_by_xxx。

然后在自己封装的这个方法里加上显式等待,并让用户在调用这个方法时,传入等待时间,如果用户不传,就用默认的。


比如这样:

例4.自己封装一个find element by id

from selenium.webdriver.common.by import By
from selenium.webdriver.support.ui import WebDriverWait
from selenium.webdriver.support import expected_conditions as EC


def find_element_by_id(driver, id, timeout=10):
      wait = WebDriverWait(driver, timeout)
      element = wait.until(EC.element_to_be_clickable((By.ID,id)))  # 等待判断条件
      return element


这个方法实现的就是如果用户没有传入等待时间,则显式等待 10秒,如果传入了等待时间,则把用户输入值设置为显示等待的最大时长。


扩展阅读(选做,需要一定基础才能比较好的理解):

1. 搜索关键字 robotframework Wait Until Keyword Succeeds

Wait Until Keyword Succeeds 这是 robot 里做等待的一个关键字,提供了很赞的等待机制:


它接受的参数有:timeout, retry_interval, name, *args

分别是,等待时间上限, 重试间隔, 要执行的方法,要执行的方法的参数表。


相当于这是一个通用的等待方法,可以等待各种普通方法的执行,而不仅仅局限于等待selenium 方法。


2.搜索关键字 selenium fluentwait

这个是 selenium java 版里提供的类似 robotframework 的等待的机制的方法。

也可以设置 等待时间上限和 重试间隔


首发于公众号:测试进阶(test_up)

原问题讨论来自以下付费社群,扫码即可加入。