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 可抑制异常，一般不建议，除非你明确要吞掉错误

示例代码：

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()。