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

为什么不能直接用 @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__。
典型结构:
立即学习“Python免费学习笔记(深入)”;
让 macOS 自动化像说话一样简单 —— 用代码控制你的 Mac,无需学习 AppleScript 系统要求 macOS 14+ (Sonoma) Python 3.9+ 需要开启:辅助功能 + 屏幕录制权限
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
-
__call__内部复用with self:,避免逻辑重复 - 装饰时传入的
func在wrapper中才真正执行,此时self才进入上下文 - 如果需要支持带参装饰器(如
@Timer("slow")),得再包一层工厂函数,但那已超出“同一类”的范畴——此时应返回新实例,而非在原实例上改状态
with 和 @ 共享状态时的坑:实例复用问题
下面这种写法看似简洁,实则危险:
ctx = Timer("shared")
@ctx
def f(): ...
<p>with ctx: # ❌ 同一个实例被反复 enter/exit
...
问题在于:同一个 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__,且不能和同步版本混用
真正麻烦的从来不是语法,而是状态生命周期——什么时候初始化、什么时候销毁、能不能跨上下文共享。写完先跑两遍:一次 with,一次 @,再交叉跑一次,多数问题就浮出来了。

















