python + slenium自动化测试设置元素等待

WebDriver 提供了两种类型的等待：显式等待和隐式等待。

显式等待

显式等待使 WebdDriver 等待某个条件成立时继续执行，否则在达到最大时长时抛出超时异常

（TimeoutException）。

from selenium import webdriver
from selenium.webdriver.common.by import By #里面包含xpath、id、name等
from selenium.webdriver.support.ui import WebDriverWait
from selenium.webdriver.support import expected_conditions as EC
driver = 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, ctime
driver = 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 out
Fri Oct 23 22:51:35 2015

隐式等待

WebDriver 提供了 implicitly_wait()方法来实现隐式等待，默认设置为 0。它的用法相对来说要简单得多。

from selenium import webdriver
from 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 秒）还没有定位到元素，则抛出异常。