Python Webdriver 从新使用已经打开的浏览器实例

由于Webdriver每次实例化都会新开一个全新的浏览器会话，在有些状况下须要复用以前打开未关闭的会话。好比爬虫，但愿结束脚本时，让浏览器处于空闲状态。当脚本从新运行时，它将继续使用这个会话工做。还就是在作自动化测试时，前面作了一大推操做，可是因为程序出错，重启时不用再继续前面复杂的操做。

我的以为这种功能很是有用，可是官方竟然没有提供这种功能的API，苦苦搜搜，在网上找了两个java版的http://blog.csdn.net/wwwqjpcom/article/details/51232302 和 http://woxiangbo.iteye.com/blog/2372683
看了下源码其实java和python的驱动原理过程都很是类似。

打开一个Chrome会话:

from selenium import webdriver
driver = webdriver.Chrome()

运行上面的脚本，它将启动浏览器并退出。由于没有调用quit()方法，因此浏览器会话仍会存在。可是代码里建立的driver对象已经不在了，理论上不能用脚本控制这个浏览器。它将变成一个僵尸浏览器，只能手动杀死它。

经过webdriver启动一个浏览器会话大概会有这样三个阶段:

• 一、启动的浏览器驱动代理(hromedriver，Firefox的驱动程序，等等)；
• 二、建立一个命令执行器。用来向代理发送操做命令；
• 三、使用代理创建一个新的浏览器会话，该代理将与浏览器进行通讯。用sessionId来标识会话。

所以只要拿到阶段2中的执行器和阶段3中的sessionID就能恢复上次的会话。这两个有api能够直接获取:

from selenium import webdriver

driver = webdriver.Chrome()
executor_url = driver.command_executor._url
session_id = driver.session_id
print(session_id)
print(executor_url)
driver.get("http://www.spiderpy.cn/")

获得相似这样的输出（第一个是会话的sessionId,第二个就是命令执行器链接）：

397d725f042a076f7d4a82f7d3fead13
http://127.0.0.1:52869

一切就绪，下面就开始实现复用以前会话的功能，在Stack Overflow上面讲的实现是这样的:

from selenium import webdriver

driver = webdriver.Chrome()
executor_url = driver.command_executor._url
session_id = driver.session_id
driver.get("http://www.spiderpy.cn/")

print(session_id)
print(executor_url)

driver2 = webdriver.Remote(command_executor=executor_url, desired_capabilities={})
driver2.session_id = session_id
print(driver2.current_url)

多是由于版本缘由吧，反正在我环境中运行时，效果是实现了，可以从新链接到上一个会话，可是却打开了一个新的空白会话。看了下Remote类的源码，发现是由于每次实例化都会调用start_session这个方法新建一个会话。因此解决方法就是继承并重写这个类。自定义一个ReuseChrome这个类重写start_session方法使它再也不新建session，使用传入的session_id：

from selenium.webdriver import Remote
from selenium.webdriver.chrome import options
from selenium.common.exceptions import InvalidArgumentException

class ReuseChrome(Remote):

    def __init__(self, command_executor, session_id):
        self.r_session_id = session_id
        Remote.__init__(self, command_executor=command_executor, desired_capabilities={})

    def start_session(self, capabilities, browser_profile=None):
        """
        重写start_session方法
        """
        if not isinstance(capabilities, dict):
            raise InvalidArgumentException("Capabilities must be a dictionary")
        if browser_profile:
            if "moz:firefoxOptions" in capabilities:
                capabilities["moz:firefoxOptions"]["profile"] = browser_profile.encoded
            else:
                capabilities.update({'firefox_profile': browser_profile.encoded})

        self.capabilities = options.Options().to_capabilities()
        self.session_id = self.r_session_id
        self.w3c = False

而后在第二次链接是使用重写的ReuseChrome类：

from selenium import webdriver

#  第一次使用Chrome() 新建浏览器会话
driver = webdriver.Chrome()

# 记录 executor_url 和 session_id 以便复用session
executor_url = driver.command_executor._url
session_id = driver.session_id
# 访问百度
driver.get("http://www.spiderpy.cn/")

print(session_id)
print(executor_url)

# 假如driver对象不存在，但浏览器未关闭
del driver

# 使用ReuseChrome()复用上次的session
driver2 = ReuseChrome(command_executor=executor_url, session_id=session_id)

# 打印current_url为百度的地址，说明复用成功
print(driver2.current_url)
driver2.get("https://www.baidu.com")

这样就能顺利链接到上次没关闭的浏览器会话。

文章地址: http://www.spiderpy.cn/blog/detail/36