webdriver使用已打开过的chrome

Posted 2020-11-16 自动化代码美学

基本功能：

执行脚本a，打开一个chrome，脚本a执行完成，chrome未关闭。
执行脚本b，继续使用a打开的chrome，不新启浏览器。

附加：

如果已打开的chrome未关闭，则在chrome中新建标签页来打开新的页面。
如果已打开的chrome已关闭，则新启浏览器。

最近用python+selenium+pytest，写了个测试小工具用来自动化登陆浏览器，一方面是方便管理网址、账号、密码，存放在脚本中，另一方面也省去了频繁输入登陆网站的操作，节省了不少时间。

但这个小工具用起来存在明显问题：每次都新启一个浏览器，多了后就是这样的

根本不知道谁是谁。

于是就想到要实现前面提到的这些功能。

首先要解决的第一个问题就是，怎么重新使用已打开的chrome。百度后知晓，是通过session_id。浏览器都有一个session_id，拿到这个session_id就可以通过webdriver.Remote去调chrome。

driver = webdriver.Chrome()
params["session_id"] = driver.session_id
params["server_url"] = driver.command_executor._url

driver = webdriver.Remote(command_executor=params["server_url"])
driver.session_id = params["session_id"]

经过试验，python存在一个坑是每次初始化就会调start_session去新开一个空白的浏览器。网上有解决方案是继承Remote，重写start_session方法。然而，仔细看完代码就发现，何必多此一举，直接quit()就搞定。

driver = webdriver.Remote(command_executor=params["server_url"])
driver.quit()  # 退出start_session新开的空白浏览器
driver.session_id = params["session_id"]

quit是quit了，但driver还存在呀，所以还是多理清思路，才能避免走冤枉路。

然后要解决的第二个问题就是，如何在执行脚本b的时候再接着用session。当然就是存本地咯。这里用到的是pickle，能很方便的在本地存取变量。

存

with open(session_file, \'wb\') as f:
    pickle.dump(params, f)

取

with open(session_file, \'rb\') as f:
    params = pickle.load(f)

第三个问题就是新开标签页和切换窗口。

driver.execute_script(\'window.open("");\')  # 调js
driver.switch_to.window(driver.window_handles[-1])  # 切换到最后一个页签

最后要解决的一个问题就是，如果已经打开的chrome关掉了，从本地文件读取的session就会过时。拿这个过时session去用，就会”chrome not reachable“。解决思路就是，捕获driver抛出的WebDriverException，重新创建新的driver。

try:
    driver = webdriver.Remote(command_executor=params["server_url"])
    driver.quit()  # 退出start_session新开的空白浏览器
    driver.session_id = params["session_id"]
    driver.execute_script(\'window.open("");\')
    driver.switch_to.window(driver.window_handles[-1])
except:
    driver = create_driver()

完整代码

session_file = \'browser_session.data\'


def create_driver():
    driver = webdriver.Chrome()
    with open(session_file, \'wb\') as f:
        params = {"session_id": driver.session_id, "server_url": driver.command_executor._url}
        pickle.dump(params, f)
    return driver

  
if not Path(session_file).exists():
    driver = create_driver()
else:
    with open(session_file, \'rb\') as f:
        params = pickle.load(f)
        try:
            driver = webdriver.Remote(command_executor=params["server_url"])
            driver.quit()  # 退出start_session新开的空白浏览器
            driver.session_id = params["session_id"]
            driver.execute_script(\'window.open("");\')
            driver.switch_to.window(driver.window_handles[-1])
        except:
            driver = create_driver()