17370845950

上下文管理器的核心机制是对象实现__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(...) 完成。

用 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 个文件并在任意一个失败时确保其余全部关闭。