Python上下文管理器进阶教程_资源管理与自定义实现

上下文管理器是支持__enter__和__exit__方法的对象，确保资源在with代码块进入前准备、退出后清理，即使发生异常也可靠执行释放逻辑。

什么是上下文管理器：不只是 with 语句

上下文管理器本质是一个支持 __enter__ 和 __exit__ 方法的对象。当你用 with 语句时，python 自动调用这两个方法，确保资源在进入代码块前被准备、退出后被清理——哪怕中间发生了异常。它不是语法糖，而是明确的生命周期契约。

为什么必须用上下文管理器管理资源

文件、数据库连接、锁、临时状态等资源，若仅靠手动 close() 或 release()，极易因忘记调用、提前 return、或异常跳过而泄漏。上下文管理器把“获取-使用-释放”三步绑定为原子行为，__exit__ 在任何退出路径下都会执行（包括 raise、return、break），这是 try/finally 的可靠封装，但更简洁、更可复用。

手写一个实用的上下文管理器

以临时修改环境变量为例，说明如何自定义：

• 在 __enter__ 中保存原值并设置新值
• 在 __exit__ 中无条件恢复原值（即使发生异常也要还原）
• __exit__ 接收三个参数：exc_type, exc_value, traceback；返回 True 可抑制异常，一般不建议，除非你明确要吞掉错误

示例代码：

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

class temp_env:     def __init__(self, **env_vars):         self.env_vars = env_vars         self.old_values = {}      def __enter__(self):         for key in self.env_vars:             self.old_values[key] = os.environ.get(key)         os.environ.update(self.env_vars)         return self      def __exit__(self, exc_type, exc_val, exc_tb):         for key, old_val in self.old_values.items():             if old_val is None:                 os.environ.pop(key, None)             else:                 os.environ[key] = old_val
用法：with temp_env(PATH="/tmp/bin"): do_something() ——退出后 PATH 自动回滚。
用 contextlib 简化开发
不需要每次都写类。常用方式有两类：

 @contextmanager 装饰器：用生成器函数定义上下文逻辑，yield 前是 __enter__，之后是 __exit__（自动处理异常传播）
 contextlib.closing()：包装已有 close() 方法的对象，如 with closing(urlopen(url)) as f: 
 contextlib.nullcontext()：占位上下文，用于条件分支中统一 with 结构（比如某些情况无需资源管理）

例如用 @contextmanager 写日志开关：
@contextmanager def log_level(level):     old = logger.level     logger.setLevel(level)     try:         yield     finally:         logger.setLevel(old)
调用：with log_level(Logging.DEBUG): run_test()。