设置元素等待时间

阅读：236560492 分享到

我们通过 WebDriver 查找控件或者操作控件的时候，如果被查找或被操作的控件不存在，就会报异常。其实很多种情况下由于网速问题，我们查找或操作的控件还没来得及出现，这时候 WebDriver 的代码就会误以为控件不存在而报异常。如果我们的代码能等待一段时间就好了，就是说在一定时间内能查找到或操作到这个控件是合理的，超越我们定的时间就认为控件不存在，然后报异常。

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

WebDriverWait 是由 WebDirver 提供的显示等待类。在设置时间内，默认每隔一段时间检测一次当前页面元素是否存在，如果超过设置时间检测不到则抛出异常。

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

显式等待

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

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

driver = webdriver.Chrome()
driver.get("http://www.baidu.com")

'''
类：
WebDriverWait(driver, timeout, poll_frequency=0.5, ignored_exceptions=None)

参数：
driver ：浏览器驱动。
timeout ：最长超时时间，默认以秒为单位。
poll_frequency ：检测的间隔（步长）时间，默认为0.5S。
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() 方法判断元素是否存在。

'''
element = WebDriverWait(driver, 5, 0.5).until(EC.presence_of_element_located((By.ID, "kw")))
element.send_keys('selenium')

driver.quit()

隐式等待

implicitly_wait() 默认参数的单位为秒，我们下面的例子中设置等待时长为 10 秒。首先这 10 秒并非一个固定的等待时间，它并不影响脚本的执行速度。其次，它并不针对页面上的某一元素进行等待。当脚本执行到某个元素定位时，如果元素可以定位，则继续执行；如果元素定位不到，则它将以轮询的方式不断地判断元素是否被定位到。假设在第 6 秒定位到了元素则继续执行，若直到超出设置时长（10秒）还没有定位到元素，则抛出异常。

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()

自定义等待

如果我们不想使用 selenium 自带的显示等待或隐式等待方法，我们完全可以自己写出等待的逻辑代码。

import time

'''
功能：
securityfind 函数在规定时间查找到控件并返回该控件，否则返回 None，默认查找 30 次，每次查找后等待 1 秒。
参数说明：
driver：控制浏览器的 webdriver。
tagname：要查找的控件名称，打印出来，给用户自己自己看的。
strtag：查找语法。
flag：查找方法，比如 by_id，by_class_name 等等。
intertime：查找一次等待多久，默认 1 秒。
count：总共查找多少次，默认 30 次。
'''
def securityfind(driver, tagname, strtag, flag, intertime = 1, count = 30):
nowcount = 0
while nowcount < count:
    time.sleep(intertime)
    try:
        if flag == "by_id":
            result = driver.find_element_by_id(strtag)
        elif flag == "by_name":
            result = driver.find_element_by_name(strtag)
        elif flag == "by_class_name":
            result = driver.find_element_by_class_name(strtag)
        elif flag == "by_tag_name":
            result = driver.find_element_by_tag_name(strtag)
        elif flag == "by_link_text":
            result = driver.find_element_by_link_text(strtag)
        elif flag == "by_partial_link_text":
            result = driver.find_element_by_partial_link_text(strtag)
        elif flag == "by_xpath":
            result = driver.find_element_by_xpath(strtag)
        elif flag == "by_css_selector":
            result = driver.find_element_by_css_selector(strtag)
        print "规定时间内找到：" + tagname
        return result  # 找到 tag 就返回
    except:
        nowcount += 1
        print("没找到" + tagname + str(nowcount) + "次")

    print("规定时间次数内没找到：" + tagname)
    return None

我们现在举例来使用上面我们自己写的查找等待函数。

from selenium import webdriver
import securityfind  # 导入我们自己写的函数

driver = webdriver.Chrome()
driver.get("http://www.baidu.com")

element = securityfind(driver, "百度搜索框", "kw", "by_id", 1, 10)
if element != None:   
    element.send_keys('selenium')
else:
    print("控件不存在")

driver.quit()