WebDriver 提供了两种类型的等待:显式等待和隐式等待。
显式等待
显式等待使 WebdDriver 等待某个条件成立时继续执行,否则在达到最大时长时抛出超时异常
(TimeoutException)。from selenium import webdriverfrom selenium.webdriver.common.by import By #里面包含xpath、id、name等from selenium.webdriver.support.ui import WebDriverWaitfrom selenium.webdriver.support import expected_conditions as ECdriver = webdriver.Firefox()driver.get("http://www.baidu.com")element = WebDriverWait(driver, 5, 0.5).until( EC.presence_of_element_located((By.ID, "kw")) )#by里面也包含xpath、name等element.send_keys('selenium')driver.quit()
WebDriverWait 类是由 WebDirver 提供的等待方法。在设置时间内,默认每隔一段时间检测一次当前页面元素是否存在,如果超过设置时间检测不到则抛出异常。具体格式如下:
WebDriverWait(driver, timeout, poll_frequency=0.5, ignored_exceptions=None)
driver :浏览器驱动。 timeout :最长超时时间,默认以秒为单位。 ignored_exceptions :超时后的异常信息,默认情况下抛 NoSuchElementException 异常。 WebDriverWait()一般由 until()或 until_not()方法配合使用,下面是 until()和 until_not()方法的说明。
until(method, message=‘ ’)
调用该方法提供的驱动程序作为一个参数,直到返回值为 True。
until_not(method, message=’ ’)
调用该方法提供的驱动程序作为一个参数,直到返回值为 False。
在本例中,通过 as 关键字将 expected_conditions 重命名为 EC,并调用 presence_of_element_located()方法判断元素是否存在。 expected_conditions 类所提供的预期条件判断的方法如表 4.3 所示。除 expected_conditions 所提供的丰富的预期条件判断方法外,还可以使用前面学过的 is_displayed()方法来判断元素是否可见。
from selenium import webdriver from time import sleep, ctimedriver = webdriver.Firefox()driver.get("http://www.baidu.com")print(ctime())for i in range(10): try: el = driver.find_element_by_id("kw22") if el.is_displayed(): break except: pass sleep(1)else: print("time out") driver.close() print(ctime())
相对来说,这种方式更容易理解,通过 for 循环 10 次,每次循环判断元素的 is_displayed()状态是否为 True:
如果为 True,则 break 跳出循环;否则 sleep(1)后继续循环判断,直到 10 次循环结束后,打印“time out”信息。执行结果如下:========================= RESTART: D:/pyse/baidu.py =========================Fri Oct 23 22:51:25 2015 time outFri Oct 23 22:51:35 2015
隐式等待
WebDriver 提供了 implicitly_wait()方法来实现隐式等待,默认设置为 0。它的用法相对来说要简单得多。
from selenium import webdriverfrom selenium.common.exceptions import NoSuchElementException from time import ctime driver = webdriver.Firefox()# 设置隐式等待为10秒driver.implicitly_wait(10) driver.get("http://www.baidu.com")try: print(ctime()) driver.find_element_by_id("kw22").send_keys('selenium')except NoSuchElementException as e: print(e)finally: print(ctime()) driver.quit()
implicitly_wait()默认参数的单位为秒,本例中设置等待时长为 10 秒。首先这 10 秒并非一个固定的等待时间,它并不影响脚本的执行速度。其次,它并不针对页面上的某一元素进行等待。当脚本执行到某个元素定位时,如果元素可以定位,则继续执行;如果元素定位不到,则它将以轮询的方式不断地判断元素是否被定位到。假设在第 6 秒定位到了元素则继续执行,若直到超出设置时长(10 秒)还没有定位到元素,则抛出异常。