Python ContextDecorator 重复调用时怎么避免上下文状态串线

同一个 ContextDecorator 实例被多个函数调用时，最容易出问题的不是装饰器语法，而是把“本次调用的状态”存进了共享实例。解决办法是把配置留在装饰器对象里，把计时、临时标记、异常信息等运行态放进每次进入时创建的上下文实例；如果上下文本身是一次性的，就实现 _recreate_cm()，让每次装饰器调用重新得到一份状态。

先看清状态串线从哪里发生

继承 ContextDecorator 后，实例既能放进 with，也能写成 @实例。装饰器调用时，Python 会在函数外层进入和退出上下文；官方文档特别要求底层上下文管理器支持多次 with。如果对象里有一个会在 __enter__ 中改写的字段，第二次调用就可能读到第一次留下的值。

例如计时器、当前请求编号、资源句柄和异常标记都属于调用态；名称、采样开关、日志级别这类不随调用变化的内容才适合放在共享装饰器实例中。

用每次创建的上下文承载可变数据

下面的实现让装饰器对象只保存标签，真正的开始时间、调用编号和异常信息放在一次调用对应的 _InvocationContext 中。代码注释只解释关键边界，便于迁移到日志、指标或临时资源场景。

from contextlib import ContextDecorator
from time import monotonic

class _InvocationContext:
    def __init__(self, label):
        self.label = label          # 稳定配置由外层传入，调用态不回写装饰器
        self.started_at = None      # 每次进入都重新设置
        self.failed = False         # 异常退出只影响当前上下文实例

    def __enter__(self):
        self.started_at = monotonic()
        return self

    def __exit__(self, exc_type, exc, traceback):
        self.failed = exc_type is not None
        elapsed = monotonic() - self.started_at
        print(self.label, round(elapsed, 4), self.failed)
        return False                # 不吞掉业务异常

class TraceCall(ContextDecorator):
    def __init__(self, label):
        self.label = label            # 只保存可安全共享的配置

    def _recreate_cm(self):
        return _InvocationContext(self.label)  # 每次装饰器调用生成独立状态

    def __enter__(self):
        self._current = self._recreate_cm()
        return self._current.__enter__()

    def __exit__(self, exc_type, exc, traceback):
        current = self._current
        self._current = None          # 先断开引用，避免把上次状态带到下一次
        return current.__exit__(exc_type, exc, traceback)

@TraceCall("load-config")
def load_config():
    return {"ready": True}

这里的关键不是把所有逻辑拆成两个类，而是明确生命周期：TraceCall 负责稳定配置和装饰器协议，_InvocationContext 负责一次进入到一次退出之间的状态。实际项目中也可以直接让 TraceCall._recreate_cm() 返回一个完整的新实例，但不要让并发调用共享 started_at、failed 或资源句柄。

一次性上下文要覆盖 _recreate_cm

contextlib.contextmanager 返回的生成器上下文通常只能走完一次。它之所以可以作为装饰器，是因为装饰器每次调用时会隐式创建新的生成器实例。自定义一次性上下文也应采用同样思路：保留创建所需的参数，在 _recreate_cm() 中返回新对象，而不是复制已经进入过的对象。

如果装饰器函数需要拿到 __enter__ 的返回值，例如事务对象或临时目录句柄，就不要强行用 @decorator，把 with TraceCall(...) as state 写在函数体内更清楚。

异常退出与重复调用的检查清单

先连续调用两次，再让第二次调用抛出异常，观察每次退出打印的标记是否只对应当前调用。__exit__ 返回 False 或 None 才会继续传播业务异常；资源清理应放在退出路径，不要依赖下一次 __enter__ 覆盖旧值。

@TraceCall("batch")
def handle(item):
    if item == "bad":
        raise ValueError("invalid item")  # 用异常路径确认当前状态会被清理
    return item.upper()

for item in ("ok", "bad", "next"):
    try:
        handle(item)                     # 每次调用都应拥有自己的上下文实例
    except ValueError:
        pass                              # 业务层决定如何处理异常，不由装饰器吞掉

检查时重点看三件事：前一次的句柄是否在退出后释放；异常是否仍能到达调用方；第三次调用是否从干净状态开始。不要把“输出顺序看起来正常”当作状态隔离证明，实际资源对象和异常字段更值得检查。

常见问题

ContextDecorator 的实例能不能直接重复使用

可以，但前提是它支持多次上下文进入，并且每次调用的可变状态都能独立初始化。一次性生成器上下文应通过重建机制使用。

为什么装饰器里拿不到 __enter__ 的返回值

装饰器只包住函数调用，没有位置接收 with ... as value 的绑定结果；需要这个值时改用函数体内的显式 with。

__exit__ 应该吞掉异常吗

除非异常本身就是业务协议的一部分，否则返回 False，让调用方决定重试、记录或终止，清理逻辑与异常策略分开。

把共享对象限制在稳定配置，把调用态交给新建的上下文实例，再用正常退出和异常退出各跑一遍，基本就能消除 ContextDecorator 重复调用时的状态串线。