必须实现__enter__和__exit__方法,因为Python解释器在执行with语句时会强制调用这两个方法,缺一不可;contextlib装饰器只是自动生成它们,并非绕过。
做不到。Python 中让一个类支持 with 语句的唯一合法方式,就是实现 __enter__ 和 __exit__ 方法。
__enter__ 和 __exit__
Python 的 with 语句在底层会显式调用这两个特殊方法:
with 块时,解释器自动调用 obj.__enter__()
with 块(无论正常还是异常)时,自动调用 obj.__exit__(exc_type, exc_value, traceback)
AttributeError: __enter__(或 __exit__)
@contextlib.contextmanager 看似不写 __enter__/__exit__,但它只是帮你自动生成——本质仍是实现了这两个方法:
from contextlib import contextmanager
@contextmanager
def my_resource():
print("setup")
yield "resource"
print("teardown")
这个装饰器返回的是一个 _GeneratorContextManager 实例,它内部已完整实现了 __enter__ 和 __exit__,并接管了生成器的启停逻辑。
@contextmanager 包装后的对象才是__enter__ 启动生成器并执行到 yield
__exit__ 处理异常并继续执行 yield 后代码(或关闭生成器)with,但模拟资源管理行为如果你只是想“看起来像 with”,又不想写两个魔术方法,那只能放弃 with 语法本身:
obj.acquire(); ...; obj.release()
try/finally 显式包裹清理逻辑contextlib.AbstractContextManager —— 它仍要求你实现 __enter__ 和 __exit__,只是提供类型提示和抽象约束没有魔法路径能跳过这两个方法;Python 解释器强制检查它们的存在性,这是语言层面的硬性约定。
真正容易被忽略的点是:哪怕只写空实现(return None 和 pass),也得有这两个方法。少一个,with 就直接报错,不给任何商量余地。