selenium中的备胎

2019-07-11 10:13:32 浏览数 (1)

漫漫人生路,总会错几步。在跑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次都失败,也就是备胎都不起作用,那也就无力回天,得考虑优化代码了。

0 人点赞