Python with 上下文管理器原理解析

python的with语句依赖上下文管理器协议，即实现__enter__和__exit__方法：前者在进入时获取资源并返回对象，后者在退出时清理资源并可选择抑制异常。

Python 的 with 语句背后依赖的是上下文管理器协议，其核心是对象实现 __enter__ 和 __exit__ 两个特殊方法。只要一个对象支持这个协议，就能用于 with 语句中，自动完成“进入”和“退出”时的资源管理逻辑。

上下文管理器协议的关键方法

__enter__ 在 with 语句开始时被调用，通常用于获取资源（如打开文件、连接数据库、加锁等），它的返回值会绑定到 as 后的变量上；__exit__ 在 with 块结束时（无论是否发生异常）被调用，负责清理工作（如关闭文件、回滚事务、释放锁等）。

该方法接收三个参数：exc_type（异常类）、exc_value（异常实例）、traceback（回溯对象）。若 __exit__ 返回 True，表示已处理异常，不会向上抛出；返回 None 或 False 则异常继续传播。

手动实现一个上下文管理器

你可以通过定义类并实现两个方法来创建自定义上下文管理器：

立即学习“Python免费学习笔记（深入）”；

• 在 __enter__ 中初始化资源并返回有用对象（如打开的文件句柄）
• 在 __exit__ 中确保资源被释放，即使中间发生异常
• 注意：如果 __exit__ 不想抑制异常，不要显式返回 True

例如，一个简单的计时上下文管理器可记录代码块执行时间，__enter__ 记录起始时间，__exit__ 计算并打印耗时。

@contextmanager 装饰器简化写法

标准库 contextlib 提供了 @contextmanager，允许用生成器函数替代类定义：

• 函数中 yield 之前的代码相当于 __enter__
• yield 的值作为 as 绑定的对象
• yield 之后的代码相当于 __exit__，会在退出时执行（包括异常情况）

这种方式更简洁，适合逻辑不复杂、无需持久状态的场景，比如临时修改环境变量、切换工作目录等。

常见误区与注意事项

不是所有带 close() 方法的对象都能直接用于 with —— 必须显式支持上下文协议。比如老版本的某些自定义类可能只提供 close，但没实现 __exit__，此时需包装或补全协议。

另外，with 并不保证线程安全，也不自动处理嵌套异常的全部细节；若需在 __exit__ 中区分不同异常类型，应检查 exc_type 参数，而非仅靠 try/except 包裹 yield。

不复杂但容易忽略。