上下文装饰器是既支持函数装饰又支持with语句的复合型装饰器，需同时实现__call__和__enter__/__exit__；与普通装饰器仅控制执行前后逻辑不同，它确保被装饰函数运行在特定上下文环境中，如自动管理资源或事务。

如何在python中实现自定义的上下文装饰器

什么是上下文装饰器，和普通装饰器有什么区别

普通装饰器只控制函数执行前后的逻辑，而上下文装饰器需要让被装饰函数“运行在某个上下文环境里”——比如自动打开/关闭文件、切换数据库事务、临时修改线程局部变量。关键在于：它得支持 with 语句的进入（enter）和退出（exit）行为，同时还能作为函数装饰器使用。

这要求类同时实现：

可调用接口（__call__），用于装饰函数
上下文管理接口（__enter__/__exit__），用于 with 块

但注意：不能直接把类实例当装饰器又当上下文管理器混用，否则会出错——比如 @MyContext() 装饰后，再写 with MyContext(): 就是两个独立实例，状态不共享。

用类实现带状态的上下文装饰器

最稳妥的方式是让装饰器返回一个可调用对象，该对象本身也支持上下文协议。典型结构是定义一个类，在 init 接收装饰器参数，在 call 返回一个包装函数；同时这个包装函数（或其内部对象）要能响应 with。

常见错误现象：TypeError: object does not support the context manager protocol，往往是因为返回的包装函数没实现 enter/exit。

正确做法是让包装函数本身是一个支持上下文的类实例：

class timer:
    def __init__(self, label=""):
        self.label = label
<pre class="brush:php;toolbar:false;">def __call__(self, func):
    def wrapped(*args, **kwargs):
        # 这里不能直接写 with self: —— self 是装饰器实例，不是上下文实例
        # 所以要构造一个新的上下文对象
        ctx = _TimerContext(self.label)
        with ctx:
            return func(*args, **kwargs)
    return wrapped

class _TimerContext: def init(self, label): self.label = label

def __enter__(self):
    self.start = time.time()
    print(f"[{self.label}] start")

def __exit__(self, *exc):
    elapsed = time.time() - self.start
    print(f"[{self.label}] done in {elapsed:.3f}s")

这样 @timer("db-query") 就能正常工作，且内部用 with _TimerContext(...) 确保上下文隔离。

用 `contextlib.contextmanager` 简化实现

如果不需要复杂状态或多次重入，用生成器 + @contextmanager 更轻量。但它不能直接当装饰器——得再包一层。

容易踩的坑：@contextmanager 函数返回的是一个上下文管理器工厂，不是管理器本身；直接 @my_cm 会报 TypeError: 'generator' object is not callable。

提示词大师-python版

图片提示词生成器？不止如此。马甲系统 —— 把脑海中的画面，翻译成AI能理解的专业表达。用得越多，它越懂你：首次需要多问几句确认方向，用久了几乎一说就懂。用得越多，它越快：缓存机制让后续对话越来越省。 RAG进化：成功案例持续入库，越跑越聪明。输入「新手指南」查看完整功能介绍

下载

正确组合方式：

定义一个 @contextmanager 函数（如 log_execution）
再写一个装饰器函数，内部用 with log_execution(): 包裹原函数调用

from contextlib import contextmanager
<p>@contextmanager
def log_execution(name):
print(f"→ entering {name}")
yield
print(f"← leaving {name}")</p><p>def with_log(name):
def decorator(func):
def wrapped(*args, *<em>kwargs):
with log_execution(name):
return func(</em>args, **kwargs)
return wrapped
return decorator</p><h1>使用</h1><p>@with_log("cache-fetch")
def get_user(uid):
return {"id": uid, "name": "alice"}
</p>

这种写法清晰、易测，且 log_execution 本身还可单独用于 with 块，复用性高。

装饰器参数和上下文参数不一致怎么办

常见场景：想让装饰器接收配置（如超时秒数），但 with 块里还想动态传参（如具体重试次数）。这时不能只靠类初始化参数。

解决方案是让上下文对象接受运行时参数，而装饰器保留默认值：

class retry:
    def __init__(self, max_attempts=3, delay=1):
        self.default_max = max_attempts
        self.default_delay = delay
<pre class="brush:php;toolbar:false;">def __call__(self, func):
    def wrapped(*args, **kwargs):
        # 允许调用时覆盖默认值：get_user(123, _retry=5)
        attempts = kwargs.pop("_retry", self.default_max)
        delay = kwargs.pop("_delay", self.default_delay)
        ctx = _RetryContext(attempts, delay)
        with ctx:
            return func(*args, **kwargs)
    return wrapped

class _RetryContext: def init(self, max_attempts, delay): self.max_attempts = max_attempts self.delay = delay