我们前面介绍的Selenium的WebDriver测试方法是基于网页来进行的。之前的例子都是用简单的网页来操作，可能体会不到网页的加载过程，但实际应用过程中，网页加载是要消耗一定的时间的。你的脚本已经开始运行，但你要定位的元素却还没加载出来，此时就会报出一个找不到元素的错误。显然，没有考虑加载时间的脚本不是一个成功的脚本。今天我们就来介绍一下如何设置等待时间。
等待的三种方法
time.sleep(n)
强制等待n秒。Python本身的函数，包含在time包中，使用前需要导入time包。我们之前的实例中经常会使用这种等待，方便大家能看清脚本执行的结果。这种等待方法十分笨拙，无论页面加载得如何，都必须等待n秒，可以说是十分不智能了。
implicitly_wait(n)
最长等待n秒，若n秒内页面加载完成，则提前结束等待。WebDriver提供的等待方法，也被称为隐性等待，比强制等待聪明了一点，但如果页面本身包含一个超大视频之类的文件，就算我们需要定位的元素在最开始已经加载出来，却依旧要等待所有文件加载结束之后，脚本才能继续执行，依旧算是有些弊端。
WebDriverWait(n)
最长等待n秒，n秒内每隔一段时间去检查需要定位的元素是否存在，若存在则提前结束等待。也是WebDriver提供的等待方法，也被称为显性等待，这种等待比隐性等待更智能了一些，无视整个页面的加载，只要需要的元素存在了，就结束等待。
实例
强制等待在之前例子中都有使用过了，我们就来看一看WebDriver提供的两种等待方法
隐性等待
其实隐性等待在之前的介绍中也有使用过，但没有特别提到使用的方法。我们这次来打开网易的首页，这是一个加载内容很多的门户网站，根据网速不同，目测大概要10秒左右才能加载完毕，我们把等待时间设置为60秒，然后计算一下从打开页面到点击页面导航栏中的【公开课】按钮需要多久。
|
1
2
3
4
5
6
7
8
9
10
11
12
13
|
# coding = utf-8
import time
driver = webdriver.Chrome()
driver.implicitly_wait(60) #隐性等待时间60秒
time_start = time.time() #记录打开页面时的时间
driver.get('https://www.163.com/')
driver.find_element_by_id('js_love_url').click()
time_end = time.time() #记录点击按钮后的时间
print('access time is : ', time_end - time_start) #打印时间差，即实际消耗时间
time.sleep(2) #强制等待2秒，为了观察我们确实打开了【公开课】页面
driver.quit()
|
脚本执行结束可以看到，虽然我们设置了隐性时间为60秒，但5秒左右（看我下面的执行结果）页面已加载完毕，可以点击【公开课】按钮。下面是我的一次执行结果，显示了整个加载时间。
|
1
|
>>>access time is : 5.717327117919922
|
显性等待
显性等待使用时需要导入selenium.webdriver.support.wait.WebDriverWait类，API如下：
|
1
|
WebDriverWait(driver, timeout, poll_frequency=0.5, ignored_exceptions=None)
|
- driver：不用说了，你定义的WebDriver浏览器（Chrome，Firefox等）
- timeout：最长等待时间，单位为秒
- poll_frequency：间隔查找元素的时间，默认为0.5秒（不设定就是0.5秒），也就是说默认0.5秒去查看一次要查找的元素是否存在，找到了就结束整个显性等待，否则继续等待0.5秒再查找一次
- ignored_exceptions=None：超时发送的异常信息，默认发送NoSuchElementException
由于显性等待可能需要确认元素是否存在，一般还要配合以下两个方法使用
|
1
2
|
until(method, message='')
until_not(method, message='')
|
- method：until()的method意为将驱动程序提供的方法作为参数调用，直到返回为not False，until_not()的method就是直到返回为False
- message：超时传的异常信息
注意，method()必须是可调用的方法，要有__call__()方法。上面的例子，我们重新写一下。
|
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
|
# coding = utf-8
from selenium import webdriver
from selenium.webdriver.support.wait import WebDriverWait
import time
driver = webdriver.Chrome()
class button():
def __call__(self, driver):
if driver.find_element_by_id('js_love_url'):
return True
else:
return False
driver.implicitly_wait(60)
time_start = time.time()
driver.get('https://www.163.com/')
# driver.find_element_by_id('js_love_url').click()
WebDriverWait(driver,2,0.5).until(button())
time_end = time.time()
print('access time is : ', time_end - time_start)
time.sleep(2)
driver.quit()
|
看完这个例子可能会有疑问，我显性等待明明设置了是2秒，为什么没有报错呢。因为我们同时也设置了隐性等待时间，两者时间取最长的等待时间为实际等待时间，因此在这里例子里，等待的时间还是60秒。
总结
1、Selenium可以采取三种等待，最智能的为显性等待WebDriverWait()
2、隐性等待和显性等待同时存在时，取两者中最长的等待时间为有效等待时间
3、显性等待中的until(method())的method()为可调用方法，可以自己定义，也可以用匿名函数等方法，这个我们以后细谈
4、隐性等待设置一次，即贯穿整个脚本始终，强制等待则必须在每个需要等待之处设置
以上就是本文的全部内容，希望对大家的学习有所帮助，也希望大家多多支持服务器之家。
原文链接：https://blog.csdn.net/kelanmomo/article/details/82886474