Python如何给类增加上下文装饰器_实现同时支持with和@的类-Python教程-PHP中文网

@contextmanager不能装饰类，因其仅接受生成器函数；类需同时实现__enter__/__exit__（支持with）和__call__（支持@），并通过延迟绑定区分使用场景。

python如何给类增加上下文装饰器_实现同时支持with和@的类

为什么不能直接用 @contextmanager 装饰类

@contextmanager 只接受生成器函数，传入类会直接报 TypeError: contextmanager expected a generator function。类本身不是可调用的生成器，更不是函数——这是最常卡住的第一步。

想让一个类同时支持 with 和 @ 两种用法，本质是让它具备双重身份：既是上下文管理器（实现 __enter__/__exit__），又是可调用对象（实现 __call__），且能接收被装饰函数并返回包装后的新可调用对象。

不实现 __call__ → 无法当装饰器用（@MyContext 会报 TypeError: 'MyContext' object is not callable）
不实现 __enter__/__exit__ → with MyContext(): 会报 AttributeError: __enter__
把逻辑写死在 __init__ 里 → 无法区分“创建实例用于 with”和“创建实例用于装饰”，导致装饰时提前执行了本该在 __enter__ 里做的事

如何设计类的初始化与调用分流

关键在于：**延迟绑定行为**。类实例化时不做实际操作，只存参数；真正触发动作的时机由使用方式决定——with 走 __enter__，@ 走 __call__。

典型结构：

class Timer:
    def __init__(self, label="block"):
        self.label = label
        self.start_time = None  # 不在这里 time.time()
<pre class="brush:php;toolbar:false;">def __enter__(self):
    import time
    self.start_time = time.time()
    return self

def __exit__(self, *exc):
    import time
    print(f"{self.label}: {time.time() - self.start_time:.3f}s")

def __call__(self, func):
    import functools
    @functools.wraps(func)
    def wrapper(*args, **kwargs):
        with self:  # 复用 __enter__/__exit__
            return func(*args, **kwargs)
    return wrapper

macOS UI 自动化技能包 —— 用 Python 控制你的 Mac

让 macOS 自动化像说话一样简单 —— 用代码控制你的 Mac，无需学习 AppleScript 系统要求 macOS 14+ (Sonoma) Python 3.9+ 需要开启：辅助功能 + 屏幕录制权限

下载

__call__ 内部复用 with self:，避免逻辑重复
装饰时传入的 func 在 wrapper 中才真正执行，此时 self 才进入上下文
如果需要支持带参装饰器（如 @Timer("slow")），得再包一层工厂函数，但那已超出“同一类”的范畴——此时应返回新实例，而非在原实例上改状态

with 和 @ 共享状态时的坑：实例复用问题

下面这种写法看似简洁，实则危险：

ctx = Timer("shared")
@ctx
def f(): ...
<p>with ctx:  # ❌ 同一个实例被反复 enter/exit
...
</p>

问题在于：同一个 Timer 实例被多次 __enter__，而 __exit__ 并不重置 self.start_time，第二次 with 会用第一次的起始时间计算耗时。

解决办法：要么禁止复用实例（文档注明“每个 with/@ 应新建实例”），要么在 __enter__ 中强制重新初始化关键状态
更稳妥的做法是让类本身无状态，把状态交给上下文内部（例如用 threading.local() 或闭包），但会增加复杂度
装饰器场景下，__call__ 返回的 wrapper 每次调用都新建上下文，天然隔离；但显式 with 就得靠使用者注意

兼容性与调试建议

Python 3.7+ 支持 contextlib.AbstractContextManager，可用于类型提示和静态检查，但非必需。重点是运行时行为是否符合预期。

测试 with：确保 __exit__ 正确处理异常（比如返回 True 吞掉异常需显式声明）
测试 @：验证被装饰函数的 __name__、__doc__ 是否保留（依赖 functools.wraps）
调试时打印 id(self) 可快速确认是否意外复用了实例
如果类需要支持异步（async with / @asynccontextmanager），必须额外实现 __aenter__/__aexit__，且不能和同步版本混用