Python 中文件打开之外的上下文管理器
内容提要
本文介绍Python上下文管理器在文件处理之外的实用场景。with语句确保代码块无论成功或异常都会执行清理。可用类实现带状态的计时器，用@contextmanager简化单一设置与清理流程，用TemporaryDirectory自动删除临时目录，用ExitStack管理运行时数量不定的资源，用patch.dict()安全还原环境变量等状态。关键原则是__exit__()默认返回False，避免静默吞掉异常。
延伸解读
异常路径才是清理逻辑的真正考验
文章开头的评估脚本在成功时能正确清理，但一旦 evaluate() 抛出异常，清理代码就被跳过，导致缓存路径、日志级别和临时文件残留。这揭示了手动清理的脆弱性：它只对正常路径有效。上下文管理器通过 __exit__() 保证无论代码块如何退出都会执行清理，将“成功才清理”变为“无论如何都清理”，尤其适合错误频发的生产环境。
选择上下文管理器形态的实用准则
文章给出了清晰的选择规则：需要读取管理器状态（如计时器的 elapsed）时用类实现；单一设置-清理流程用 @contextmanager 简化；标准库已管理生命周期（如 TemporaryDirectory、patch.dict）时直接使用；资源数量运行时才确定时用 ExitStack。关键原则是 __exit__() 默认返回 False，避免静默吞掉异常，只有明确需要继续执行时才用 contextlib.suppress()。
全局状态修改的并发风险
patch.dict() 能安全还原环境变量等字典状态，但文章提醒：os.environ 是进程级全局状态，修改对同一进程内所有线程和协程同时可见。在并发场景下，这种临时状态变更可能导致难以调试的问题。标准库的 contextlib.chdir() 也带有类似警告。因此，文章建议在并发环境中将路径或配置作为参数传递，而非依赖全局状态修改。
Q&A
Python 的 with 语句中，__exit__() 的返回值有什么作用？
__exit__() 的返回值决定异常是否被抑制：返回真值会吞掉异常，返回 False 或 None 则异常在清理后继续传播。默认应返回 False，避免静默吞掉异常。
如何用类实现一个带状态的计时器上下文管理器？
定义一个类，在 __enter__() 中记录开始时间并返回自身，在 __exit__() 中计算耗时并打印，同时返回 False 让异常传播。这样即使代码块抛出异常，也能输出耗时和失败状态。
@contextmanager 装饰器适合什么场景？使用时要注意什么？
适合单一设置、执行、清理的流程，如临时修改日志级别。生成器函数中 yield 之前做设置，之后做清理，清理应放在 finally 中。必须恰好 yield 一次，且若捕获异常需重新抛出，否则会变成异常抑制器。
TemporaryDirectory 如何自动清理临时目录？有什么注意事项？
TemporaryDirectory 在 with 块结束时自动删除目录及其内容，无论块内是否异常。它创建唯一命名的目录，避免冲突。在 Windows 上若文件未关闭可能导致 PermissionError，可传 ignore_cleanup_errors=True 降级为尽力清理，但更好的做法是嵌套 with 确保文件先关闭。
ExitStack 适用于什么情况？它如何管理资源？
ExitStack 适用于运行时数量不定的资源管理，如打开未知数量的分片文件。通过 enter_context() 注册上下文管理器，callback() 注册普通清理函数，块结束时按后进先出顺序展开清理。注意它只在块结束或调用 close() 时展开，不会依赖垃圾回收。
如何安全地临时修改环境变量并在之后恢复？
可以使用 unittest.mock.patch.dict() 作为上下文管理器，在块内修改字典（如 os.environ），块结束后自动恢复原状：已存在的键恢复旧值，不存在的键被删除。注意环境变量是进程全局的，在多线程或异步代码中不安全，应改为传递参数。
