上下文管理器的核心机制是对象实现__enter__和__exit__方法:with语句自动调用__enter__获取返回值,并在退出时(无论是否异常)调用__exit__执行清理;后者接收exc_type、exc_value、traceback三参数,返回True可抑制异常。

什么是上下文管理器的核心机制
上下文管理器的本质是对象对 __enter__ 和 __exit__ 两个特殊方法的实现。当使用 with 语句时,Python 自动调用 __enter__ 获取进入上下文的返回值,并在退出时(无论是否发生异常)调用 __exit__ 执行清理逻辑。
关键点在于:__exit__ 接收三个参数 —— exc_type、exc_value、traceback。若该方法返回 True,则表示已处理异常,异常不会向上抛出;返回 None 或 False,异常照常传播。
手写一个带状态感知的文件处理器
相比内置 open(),自定义类可记录打开模式、是否已修改、实际读取字节数等信息,便于调试和审计。
- 在 __enter__ 中打开文件并初始化状态属性(如 self.was_modified = False)
- 提供 write()、read() 等方法,在内部更新状态(如设置 self.was_modified = True)
- 在 __exit__ 中根据状态决定是否刷新缓冲区、是否记录日志、是否校验完整性
示例场景:写入关键配置后,自动计算 SHA256 并写入 .sha256 同名文件 —— 这类逻辑无法用普通 with open(...) 完成。
立即学习“Python免费学习笔记(深入)”;
用 contextlib.contextmanager 装饰器简化开发
无需定义完整类,只需一个生成器函数,用 yield 分隔进入与退出逻辑:
- yield 之前的代码相当于 __enter__,其返回值即 as 后的变量
- yield 之后的代码在退出时执行,天然支持 try/except/finally 结构
- 若生成器中抛出异常,会传递给 __exit__ 的对应参数,仍可选择吞掉或重新抛出
适合快速封装一次性资源(如临时目录创建、数据库连接池借用、线程局部变量设置等),代码更紧凑,语义更清晰。
嵌套与组合多个上下文管理器
Python 3.1+ 支持在单个 with 语句中管理多个资源,逗号分隔即可:
with DatabaseConnection() as db, open('log.txt', 'a') as log_f:
db.execute('UPDATE ...')
log_f.write('done')
所有管理器按顺序进入,逆序退出。若某个 __enter__ 失败,之前成功进入的管理器会正常调用 __exit__ 清理(即“回滚已获取资源”)。也可用 contextlib.ExitStack 动态管理不确定数量的上下文,比如批量打开 N 个文件并在任意一个失败时确保其余全部关闭。










