__enter__和__exit__必须成对出现,因为with语句依赖二者完成资源获取与清理的完整生命周期;缺__exit__会报AttributeError,且无法保证异常路径下资源释放。

为什么 __enter__ 和 __exit__ 必须成对出现
因为 with 语句的底层机制依赖这两个方法构成完整生命周期:进入时调用 __enter__ 获取资源,退出时无条件触发 __exit__ 做清理。缺一不可,否则会报 AttributeError: __exit__。
常见错误是只写了 __enter__(比如想简单返回个值),结果 with 块结束时找不到退出逻辑,资源泄漏或异常被吞掉。
-
__enter__应该返回要绑定到as变量的对象(可以是self,也可以是其他实例) -
__exit__必须接收三个参数:exc_type、exc_value、traceback,即使你什么都不做也得声明它们 - 如果
__exit__返回True,会抑制异常;返回None或False则异常照常抛出
如何让自定义上下文管理器支持异常传播与部分抑制
关键在 __exit__ 的返回值逻辑。不是“捕获异常”,而是“决定是否让异常继续向上冒泡”。比如日志记录后仍要抛出,就不能 return True。
典型场景:打开文件写入,希望 IO 错误暴露给调用方,但临时锁必须释放。
立即学习“Python免费学习笔记(深入)”;
def __exit__(self, exc_type, exc_value, traceback):
self.lock.release() # 总要释放
if exc_type is not None and issubclass(exc_type, ValueError):
print("忽略 ValueError")
return True # 抑制这一类
return False # 其他异常照常抛出
- 用
exc_type is not None判断是否有异常发生 - 用
issubclass(exc_type, SomeException)精确匹配类型,避免用字符串比较 - 不要在
__exit__里主动raise新异常——这会覆盖原始异常,极难调试
用 contextlib.contextmanager 装饰器替代手写类的适用边界
当逻辑简单、无状态、不需复用实例时,@contextmanager 更轻量;但一旦涉及属性缓存、多次进入、或需要继承扩展,就必须用类。
多奥淘宝客程序免费版拥有淘宝客站点的基本功能,手动更新少,管理简单等优点,适合刚接触网站的淘客们,或者是兼职做淘客们。同样拥有VIP版的模板引擎技 术、强大的文件缓存机制,但没有VIP版的伪原创跟自定义URL等多项创新的搜索引擎优化技术,除此之外也是一款高效的API数据系统实现无人值守全自动 化运行的淘宝客网站程序。4月3日淘宝联盟重新开放淘宝API申请,新用户也可使用了
例如临时切换工作目录,用函数式写法很干净:
from contextlib import contextmanager import os@contextmanager def cd(path): old = os.getcwd() os.chdir(path) try: yield finally: os.chdir(old)
-
yield之前的部分相当于__enter__,之后(finally块)相当于__exit__ - 不能在
@contextmanager函数里返回值给as绑定——除非用yield value,且该值会在with块中可用 - 装饰器版本无法被继承,也不支持
isinstance(obj, ContextManager)检查(它只是个 generator function)
测试自定义上下文管理器是否真正“安全退出”
不能只测正常流程,重点验证异常路径下资源是否释放。比如文件句柄、网络连接、数据库事务等,漏掉异常分支就等于埋雷。
最直接的方式是故意抛异常并检查状态:
with MyResource() as r:
assert r.is_open
raise KeyError("boom")
# 此处断言 r.is_open == False,否则 __exit__ 没生效
- 用
pytest.raises包裹整个with块,再检查退出后状态 - 避免在
__exit__里做耗时操作(如网络请求),它可能在异常处理链中被多次调用 - 如果上下文对象本身有状态(如计数器、缓存),确保
__exit__不依赖未初始化的属性——__enter__失败时__exit__仍会被调用
实际用起来最难的不是写两个方法,而是预判所有退出路径——正常结束、return、break、各种异常、甚至 os._exit()。只要没覆盖全,就存在资源残留风险。









