Python + Selenium 小技巧之等待

引言

在自动化测试过程中有一种情况很常见，明明你的定位没有问题，代码执行时却直接报了元素不可见。这时就需要考虑下是不是因为程序运行太快，或者由于页面加载太慢造成了元素还没出来就已经报错了。要知道浏览器的渲染加载比不上你代码执行的速度。

为了解决这种情况，就用到了等待，在Selenium中有三种等待方式，分别为强制等待、隐式等待、显式等待。

强制等待

第一种等待方式：强制等待，其实指的就是time模块的sleep，示例如下：

import time
from selenium import webdriver

browser = webdriver.Chrome()
browser.get('https://HGzhao.github.io')
time.sleep(3) # 单位为秒(s)
browser.quit()

优缺点说明

优点
简单明了，需要等待的时候直接time.sleep()，等待的时长根据执行Case的需要来定。
缺点
不智能。若设置的时间短，元素还没有加载出来，照样会报错；若设置的时间过长，便会加长Case的执行时间，影响整体的运行速度。

隐式等待

第二种等待方式：隐式等待，使用的是webdriver中implicitly_wait()函数，示例如下：

from selenium import webdriver

browser = webdriver.Chrome()  
browser.implicitly_wait(20) # 最长等待20秒  
browser.get('https://HGzhao.github.io')
browser.quit()

这里有两点需要注意：

隐式等待的设置时全局性的，在开头设置过之后，整个的程序运行过程中都会有效，都会等待页面加载完成；不需要每次设置一遍。
页面加载完成后新打开的弹窗是没有等待效果的。

优缺点说明

优点
相比sleep() ，隐性更加智能，它会在你设置的时间内判断网页是否加载完成，如果完成了，就进行下一步；在设置的时间内没有加载完成，则会报超时加载。
缺点
随着ajax技术的广泛应用，页面的元素往往都可以局部加载，也就是在整个页面没有加载完的时候，可能我们需要的元素已经加载完成了，那就么有必要再等待整个页面的加载。这就引出了第三种等待的方式—显式等待。

显式等待

第三种等待方式：显式等待，使用的是WebDriverWait，配合该类的until()和until_not()方法，就能够根据判断条件而进行灵活地等待了。示例如下：

from datetime import datetime
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

browser = webdriver.Chrome()
browser.get("http://www.baidu.com")
browser.maximize_window()
# 为了更好的对比效果，首先我们设置了一个存在的元素，然后在去找一个不存在的元素，同样设置了10s的等待时间
try:
# kw元素存在
    print(datetime.now())
    element = WebDriverWait(browser, 10).until(EC.presence_of_element_located((By.ID, "kw")))
# 不存在的kwqq元素
    print(datetime.now())
    element = WebDriverWait(browser, 10).until(EC.presence_of_element_located((By.ID, "kwqq")))
finally:
    print(datetime.now())
    browser.quit()

说明：
1、until也属于WebDriverWait，代表一直等待，直到某元素可见。而until_not与其相反，判断某个元素直到不存在。
2、presence_of_element_located主要判断页面元素kw在页面中存在。

优缺点说明

优点
它会不断的根据你设定的条件去判断，直到超过你设置的等待时间，如果设置的条件满足，然后进行下一步操作。
如果没有满足会报一个selenium.common.exceptions.TimeoutException: Message:错误。

多种等待混合使用

隐性等待和显性等待可以同时用，但要注意：等待的最长时间取两者之中的大者。示例如下：

from selenium import webdriver
from selenium.webdriver.support.wait import WebDriverWait
from selenium.webdriver.support import expected_conditions as EC
from selenium.webdriver.common.by import By

browser = webdriver.Chrome()
browser.implicitly_wait(10)
browser.get('https://HGzhao.github.io')
locator = (By.LINK_TEXT, 'CSDN')

try:
    WebDriverWait(browser, 20, 0.5).until(EC.presence_of_element_located(locator))
    print(browser.find_element_by_link_text('CSDN').get_attribute('href'))
finally:
    browser.close()

结语

我们在实际使用selenium或appium执行一些自动化操作时，对一个待定位元素的出现特别是web端加载的过程，等待是一种行之有效的方式。设置一种合理地等待方式，是保证脚本稳定有效运行的一个非常重要的手段。此外需要注意的是，对于元素的选择，最好将其独立封装成方法，在每个方法中可以加上日志、显式等待。

＆娴敲棋子＆

https://HGzhao.github.io/2019/09/14/python-selenium-waiting/