Python中如何线程安全地检测函数是否处于指定ContextManager中？

要实现你想要的效果，核心是利用线程本地存储（threading.local()）——它能为每个线程维护独立的上下文数据，天然保证线程安全。下面是具体的实现步骤：

1. 定义线程本地存储对象 #

首先创建一个线程本地存储实例，用来保存当前线程的MyContext上下文：

import threading

# 线程本地存储，每个线程有独立的存储空间
_local = threading.local()

2. 实现MyContext上下文管理器 #

在上下文的__enter__方法中把当前上下文实例存入线程本地存储，__exit__方法中恢复之前的上下文（支持嵌套上下文场景）：

class MyContext:
    def __enter__(self):
        # 保存之前的上下文，处理嵌套上下文的情况
        self.prev_ctx = getattr(_local, "current_my_context", None)
        # 将当前上下文设为线程本地存储的活跃上下文
        _local.current_my_context = self
        # 可选：如果需要兼容`with MyContext() as ctx`的写法，返回self即可
        return self

    def __exit__(self, exc_type, exc_val, exc_tb):
        # 恢复之前的上下文，保证退出后不影响后续代码逻辑
        _local.current_my_context = self.prev_ctx

3. 实现some_function的上下文检测 #

在函数内部通过线程本地存储获取当前活跃的上下文，判断是否处于MyContext环境中：

def some_function():
    # 从线程本地存储获取当前上下文
    current_ctx = getattr(_local, "current_my_context", None)
    # 检测是否处于MyContext的上下文环境
    if not isinstance(current_ctx, MyContext):
        raise RuntimeError("some_function 必须在 MyContext 上下文环境中调用")
    
    # 这里编写原`ctx.some_function()`中的业务逻辑
    print("已在MyContext上下文中执行操作")

4. 测试验证 #

# 正常调用：在上下文内执行
with MyContext():
    some_function()  # 输出：已在MyContext上下文中执行操作

# 错误调用：不在上下文内执行
try:
    some_function()
except RuntimeError as e:
    print(e)  # 输出：some_function 必须在 MyContext 上下文环境中调用

原理说明 #

threading.local()创建的对象，每个线程访问时都会拿到属于自己的独立数据副本，不同线程的上下文状态完全隔离，所以天然满足线程安全要求。Python内置的decimal.localcontext也是基于同样的原理实现——通过线程本地存储维护当前线程的Decimal上下文，让你在with块内无需显式传递ctx就能访问上下文配置。

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