Chromedriver进程不会自动结束该怎么强制关闭掉进程

原创 码农  2020-01-31 11:43:15  阅读 1211 次 评论 0 条

Python开发测试中,当我们使用Selenium测试工具通过 Chromedriver 启动 Chrome 浏览网页时,可能会由于某些异常情况导致程序崩溃,但 Chromedriver 进程不会退出。

例如,我们编写一段显然有问题的代码:

from selenium.webdriver import Chrome


driver = Chrome('./chromedriver')
driver.get('https://www.xxx.com')

1 + 'a' # 这一行代码必定导致程序崩溃

代码报错以后,弹出的 Chrome 窗口不会自动关闭。并且 chromedriver 的进程也不会自动结束,如下图所示:

Chromedriver进程不会自动结束该怎么强制关闭掉进程 编程代码 第1张

这就会导致系统中出现越来越多的 chromedriver 进程,从而占用大量的内存。

为了防止这种情况,我们必须想办法,在任何情况下都需要保证退出 chromedriver。

你可能会使用一个超大型的 try ... except...把所有与 selenium 相关的代码都包起来:

from selenium.webdriver import Chrome


driver = Chrome('./chromedriver')
try:
    driver.get('https://www.xxx.com')
 #第2行
 #第3行
 #第4行
 #。。。
 #第 n 行
except Exception:
    driver.quit()

当然你也可以把具体的操作步骤放在函数里面,然后 try...except... 把函数包住。但本质上是一样的。

但这种超大型的 try...except...一是会导致程序速度减慢,二是程序出现了其他异常的时候,真正的报错信息无法正常打印出来:

>>> a = {}
>>> try:
>>>    a['k']
>>> except Exception as e:
>>>    print(e)
Chromedriver进程不会自动结束该怎么强制关闭掉进程 编程代码 第2张

当你看到这个没头没尾的'k',你不知道是哪一行有问题,也不知道具体有什么问题。

那么,我们有没有办法,既不使用 try ... except ...,但是又能在程序崩溃的时候自动退出 chromedriver 呢?

这个时候我们就可以使用上下文管理器。

我们先来包装一下 Selenium,实现一个带有上下文管理器的类。创建一个SafeDriver.py文件:

from selenium.webdriver import Chrome


class SafeDriver:
    def __init__(self):
        self.driver = Chrome('./chromedriver')

    def __enter__(self):
        return self

    def __exit__(self, exc_type, exc_val, exc_tb):
        if self.driver:
            self.driver.quit()

然后,我们在另一个程序里面调用它:

from SafeDriver import SafeDriver

safe_driver = SafeDriver()
with safe_driver as driver:
    driver.driver.get('https://www.xxx.com')
    a = {}
    a['k']

程序进入到with safe_driver as driver的缩进里面,我们会得到一个driver变量,它可以用来操作浏览器。

我们只需要在缩进里面正常写代码即可。一旦由于某种原因导致缩进里面的代码报错,Python 自动会进入SafeDriver类的__exit__方法中,执行里面的代码。在这个方法里面,我们就可以关闭 chromedriver。从而保证只要程序异常退出,浏览器一定会被关闭,不会遗留进程。

运行效果如下图所示:

Chromedriver进程不会自动结束该怎么强制关闭掉进程 编程代码 第3张

报错信息和出错的行数都能正常打印出来了。

我们来看看如何实现一个包含上下文管理器的程序:

  1. 随意定义一个类,里面写好你需要执行的逻辑
  2. 增加__enter__(self)方法,定义进入上下文管理器时返回的内容
  3. 增加__exit__(self, exc_type, exc_val, exc_tb)方法,定义退出上下文管理器时需要执行的代码

需要注意的是,__enter____exit__需要成对使用,不能单独使用其中一个。

在上面的代码中,__enter__方法仅仅返回了self,于是,下面两段代码

safe_driver = SafeDriver()with safe_driver as instance:pass

仅仅从功能上来说,instance 变量与safe_driver变量完全一样,都可以使用safe_driver.driverinstance.driver

所不同的是,使用with启用上下文管理器以后,在退出缩进的时候会执行__exit__中的内容。

为了简便起见,我们可以使用with safe_driver.driver as driver,直接拿到对象中的self.driver属性,这样可以直接使用类似于driver.get('https://www.xxx.com')访问网站,而不是instance.driver.get('https://www.xxx.com')。少敲几次键盘

本文地址:https://www.itcodeit.com/post/66.html
版权声明:本文为原创文章,版权归 码农 所有,欢迎分享本文,转载请保留出处!

发表评论


表情

还没有留言,还不快点抢沙发?