上下文管理器是支持__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免费学习笔记（深入）”；
<font size="2"><pre class="brush:php;toolbar:false;">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 写日志开关：
<font size="2"><pre class="brush:php;toolbar:false;">@contextmanager
def log_level(level):
old = logger.level
logger.setLevel(level)
try:
yield
finally:
logger.setLevel(old)
调用：with log_level(logging.DEBUG): run_test()。
