脚本强制退出时Chrome浏览器无法关闭的技术求助

问题核心 #

在多文件Selenium项目中，try/except/finally和atexit模块能实现脚本异常退出时自动关闭Firefox，但Chrome仅在VS Code调试环境下生效，强制终止（如Ctrl+C、进程杀死）时无法关闭。

原因分析 #

1. Chrome进程分离特性：Chrome与ChromeDriver是独立进程，强制终止Python脚本时，仅杀死Python进程，未处理ChromeDriver和Chrome的进程链。
2. 多文件资源管理分散：初始化与销毁逻辑分布在不同模块，atexit注册时机不当，导致清理函数未被触发。
3. 调试环境特殊处理：VS Code调试时会捕获进程退出信号，确保清理逻辑执行；但正常运行或强制终止时，信号未被处理。

解决方案 #

1. 捕获退出信号，强制执行清理逻辑 #

针对Ctrl+C（SIGINT）、kill命令（SIGTERM）等终止信号，注册处理函数，确保关闭浏览器及驱动进程。

# initialize.py
import signal
from selenium import webdriver

driver = None

def cleanup_driver(signal_num, frame):
    global driver
    if driver:
        driver.quit()  # 使用quit()而非close()，彻底终止驱动与Chrome进程
    exit(0)

# 注册信号处理
signal.signal(signal.SIGINT, cleanup_driver)
signal.signal(signal.SIGTERM, cleanup_driver)

def init_chrome():
    global driver
    options = webdriver.ChromeOptions()
    # 添加你的Chrome配置（如headless、禁用GPU等）
    driver = webdriver.Chrome(options=options)
    return driver

2. 用上下文管理器封装驱动实例 #

将Chrome驱动的初始化与销毁逻辑封装为上下文管理器，确保with代码块结束时（含异常退出）自动执行清理。

# initialize.py
from selenium import webdriver
from contextlib import contextmanager

@contextmanager
def chrome_driver():
    options = webdriver.ChromeOptions()
    # 自定义配置
    options.add_argument("--headless=new")
    options.add_argument("--disable-gpu")
    
    driver = webdriver.Chrome(options=options)
    try:
        yield driver
    finally:
        driver.quit()

在业务模块中调用：

# scraper.py
from initialize import chrome_driver

def run_scrape():
    with chrome_driver() as driver:
        driver.get("https://target-site.com")
        # 爬取逻辑

3. 极端情况：强制清理Chrome相关进程 #

若前两种方法失效，可直接通过系统命令杀死Chrome及ChromeDriver进程（注意：可能误杀用户手动打开的Chrome，建议结合进程ID精准清理）。

import os
import platform
from selenium import webdriver

def kill_chrome_processes(driver_pid=None):
    system = platform.system()
    if system == "Windows":
        if driver_pid:
            os.system(f"taskkill /f /t /pid {driver_pid}")
        else:
            os.system("taskkill /f /im chrome.exe /im chromedriver.exe")
    elif system in ["Linux", "Darwin"]:
        if driver_pid:
            os.system(f"pkill -P {driver_pid}")
            os.system(f"kill -9 {driver_pid}")
        else:
            os.system("pkill -f chrome")
            os.system("pkill -f chromedriver")

# 在初始化时记录驱动进程ID
driver = webdriver.Chrome()
driver_pid = driver.service.process.pid

# 注册到atexit或信号处理函数
import atexit
atexit.register(lambda: kill_chrome_processes(driver_pid))

4. 调整ChromeOptions配置，绑定进程 #

添加detach配置，确保ChromeDriver退出时自动关闭Chrome进程。

options = webdriver.ChromeOptions()
options.add_experimental_option("detach", False)  # 默认False，若被修改需显式设置
# 其他配置
driver = webdriver.Chrome(options=options)

推荐组合 #

优先使用信号处理+上下文管理器的组合，覆盖绝大多数异常退出场景；极端情况补充进程清理逻辑。

内容的提问来源于stack exchange，提问作者farhan jatt