漫漫人生路,总会错几步。在跑selenium的时候,或多或少,总有那么几次会fail掉。如同我们汽车有备胎,selenium中我们也可以弄个备胎。
在日常的selenium的自动化测试中,我们总能得到这样的error.
代码语言:javascript复制 raise TimeoutException(message, screen, stacktrace)
selenium.common.exceptions.TimeoutException: Message: None
基本上就是超时的error. 如果页面超时,会导致后面的所有都不执行。 selenium是单进程在执行,我们考虑搞一个备用的进程,如果主进程死了,就可以切换过来,如同我们生活中的备胎。 这样就不需要重新启动selenium。
前面我们用过retry, 是需要重新启动的。这个不需要重启,就是耗费了一点点资源,可以忽略不计。 原理很简单,就是我们备用一个windows handle, 如果异常了,我们就切换过去,重来一遍。
代码语言:javascript复制from selenium import webdriver
from selenium.webdriver.chrome.options import Options
import traceback
url = "http://new/"
chrome_options = Options()
prefs = {
'profile.default_content_setting_values': {
'images': 2, # 禁用图片的加载
'javascript': 2 ##禁用js,可能会导致通过js加载的互动数抓取失效
}
}
chrome_options.add_experimental_option("prefs", prefs)
# chrome_options.add_argument("--headless") # 不弹出浏览器
browser = webdriver.Chrome(chrome_options=chrome_options)
browser.implicitly_wait(5) # 操作、获取元素时的隐式等待时间
browser.set_page_load_timeout(10) # 页面加载超时等待时间
main_win = browser.current_window_handle # 记录当前窗口的句柄
all_win = browser.window_handles
try:
if len(all_win) == 1:
print('开启备胎')
js = 'window.open("https://www.baidu.com");'
browser.execute_script(js) # 开启保护
all_win = browser.window_handles
browser.get(url) # 此处访问你需要的URL
except:
# 超时
print('Time out')
if 'time' in str(traceback.format_exc()):
# 切换新的浏览器窗口
for win in all_win:
if main_win != win:
print('WIN', win, 'Main', main_win)
print('切换到备胎')
browser.close()
browser.switch_to.window(win)
main_win = win
browser.get(url) # 此处访问你需要的URL
browser.find_element_by_id('new-activate-btn').click() #后续操作
测试: 我们可以把需要打开的页面,用抓包工具给断点着,制造超时的场景。 测试通过。
这如同我们生活中,如果上天再给一次机会,我一定会好好争取。
这样我们的测试就稳定多了,如果2次都失败,也就是备胎都不起作用,那也就无力回天,得考虑优化代码了。