Python中能否用with open()语法管理Selenium Driver实现自动关闭？

当然可以利用with语句来管理Selenium Driver，核心是让Driver实现上下文管理器协议（即包含__enter__和__exit__方法），这样with块执行完毕后会自动触发资源清理操作（关闭Driver）。以下是两种实用方案：

方案一：自定义上下文管理器类 #

通过包装Selenium Driver实现上下文管理器，逻辑清晰直观：

from selenium import webdriver
from selenium.webdriver.chrome.service import Service

class ChromeDriverManager:
    def __enter__(self):
        # 初始化Driver
        self.driver = webdriver.Chrome(service=Service())
        return self.driver
    
    def __exit__(self, exc_type, exc_val, exc_tb):
        # 不管是否出现异常，都关闭Driver
        self.driver.quit()

# 使用示例
with ChromeDriverManager() as driver:
    driver.get("https://example.com")
    # 在这里执行你的页面操作，比如定位元素、点击等

方案二：用contextlib简化实现 #

借助contextlib.contextmanager装饰器，用生成器函数快速实现上下文管理器，代码更简洁：

from selenium import webdriver
from selenium.webdriver.chrome.service import Service
from contextlib import contextmanager

@contextmanager
def chrome_driver():
    driver = webdriver.Chrome(service=Service())
    try:
        yield driver  # 将Driver返回给with语句的变量
    finally:
        driver.quit()  # 确保Driver被关闭

# 使用示例
with chrome_driver() as driver:
    driver.get("https://example.com")
    # 执行你的操作

两种方案的核心优势是：无论with块内的代码是否抛出异常，Driver都会自动调用quit()方法关闭，彻底避免手动关闭遗漏的问题。

内容的提问来源于stack exchange，提问作者54starr