如何让Python上下文管理器作用域内的类无需传参自动获取当前上下文实例

可以通过Python标准库的contextvars模块实现并发安全的上下文自动传递，完全满足并行测试的隔离性要求，不需要手动传递webapp实例。

实现原理 #

contextvars是Python3.7+内置的上下文变量模块，会为不同的执行流（线程、协程、多进程）维护独立的变量副本，天然避免不同测试用例的上下文互相污染，刚好适配并行测试场景。我们只需要在webapp上下文管理器进入时将实例存入上下文变量，FileLoader初始化时自动从上下文变量取值即可。

代码实现 #

1. 定义上下文变量并改造webapp上下文管理器 #

import contextvars
from selenium import webdriver

# 全局上下文变量，存储当前执行流对应的webapp实例
current_webapp = contextvars.ContextVar('current_webapp', default=None)

class webapp:
    def __enter__(self):
        # 原有启动浏览器、初始化驱动的逻辑
        self.driver = webdriver.Chrome()
        # 存入当前webapp实例到上下文变量，记录token用于后续重置
        self._context_token = current_webapp.set(self)
        return self

    def __exit__(self, exc_type, exc_val, exc_tb):
        # 原有关闭浏览器的逻辑
        self.driver.quit()
        # 重置上下文变量，避免污染后续执行流
        current_webapp.reset(self._context_token)

2. 改造FileLoader自动获取上下文 #

class FileLoader:
    def __init__(self, id: str):
        # 自动从当前上下文获取webapp实例
        self.webapp_instance = current_webapp.get()
        if not self.webapp_instance:
            raise RuntimeError("FileLoader必须在webapp上下文内部使用")
        self.id = id
    
    def upload(self, path: str):
        # 原有文件上传逻辑，直接使用self.webapp_instance操作驱动即可
        pass

3. 测试用例写法 #

和期望的写法完全一致：

def test_xxx():
    with webapp():                                                                    
        file_loader = FileLoader(id="loader").upload(path)

方案优势 #

• 完全隔离：多进程、多线程、协程级别的并行测试都不会出现上下文串用的问题，每个测试用例拿到的都是自己的webapp实例
• 侵入性低：只需要修改webapp上下文管理器和FileLoader的初始化逻辑，原有业务代码不需要调整
• 错误易排查：如果在上下文外使用FileLoader会直接抛出明确的运行时错误，方便调试

内容的提问来源于stack exchange，提问作者Stefano Borini