Python selenium如何设置等待时间

我们前面介绍的Selenium的WebDriver测试方法是基于网页来进行的。之前的例子都是用简单的网页来操作，可能体会不到网页的加载过程，但实际应用过程中，网页加载是要消耗一定的时间的。你的脚本已经开始运行，但你要定位的元素却还没加载出来，此时就会报出一个找不到元素的错误。显然，没有考虑加载时间的脚本不是一个成功的脚本。今天我们就来介绍一下如何设置等待时间。

等待的三种方法

time.sleep(n)

强制等待n秒。Python本身的函数，包含在time包中，使用前需要导入time包。我们之前的实例中经常会使用这种等待，方便大家能看清脚本执行的结果。这种等待方法十分笨拙，无论页面加载得如何，都必须等待n秒，可以说是十分不智能了。

implicitly_wait(n)

最长等待n秒，若n秒内页面加载完成，则提前结束等待。WebDriver提供的等待方法，也被称为隐性等待，比强制等待聪明了一点，但如果页面本身包含一个超大视频之类的文件，就算我们需要定位的元素在最开始已经加载出来，却依旧要等待所有文件加载结束之后，脚本才能继续执行，依旧算是有些弊端。

WebDriverWait(n)

最长等待n秒，n秒内每隔一段时间去检查需要定位的元素是否存在，若存在则提前结束等待。也是WebDriver提供的等待方法，也被称为显性等待，这种等待比隐性等待更智能了一些，无视整个页面的加载，只要需要的元素存在了，就结束等待。

实例

强制等待在之前例子中都有使用过了，我们就来看一看WebDriver提供的两种等待方法

隐性等待

其实隐性等待在之前的介绍中也有使用过，但没有特别提到使用的方法。我们这次来打开网易的首页，这是一个加载内容很多的门户网站，根据网速不同，目测大概要10秒左右才能加载完毕，我们把等待时间设置为60秒，然后计算一下从打开页面到点击页面导航栏中的【公开课】按钮需要多久。

# coding = utf-8
from selenium import webdriver
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秒左右（看我下面的执行结果）页面已加载完毕，可以点击【公开课】按钮。下面是我的一次执行结果，显示了整个加载时间。

>>>access time is : 5.717327117919922

显性等待

显性等待使用时需要导入selenium.webdriver.support.wait.WebDriverWait类，API如下：

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
  

由于显性等待可能需要确认元素是否存在，一般还要配合以下两个方法使用

until(method, message='')
until_not(method, message='')

• method：until()的method意为将驱动程序提供的方法作为参数调用，直到返回为not False，until_not()的method就是直到返回为False
• message：超时传的异常信息
  

注意，method()必须是可调用的方法，要有__call__()方法。上面的例子，我们重新写一下。

# 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、隐性等待设置一次，即贯穿整个脚本始终，强制等待则必须在每个需要等待之处设置

以上就是本文的全部内容，希望对大家的学习有所帮助，也希望大家多多支持脚本之家。

最新评论