Pythoncontextmanager使用详解

**Python contextmanager 装饰器用法深度解析：简化资源管理，提升代码效率** 想要更优雅地管理 Python 中的资源吗？`contextmanager` 装饰器正是你的利器！本文将深入解析 `contextmanager` 的用法，它是 `contextlib` 模块提供的强大工具，可以将生成器函数轻松转化为上下文管理器，配合 `with` 语句，自动处理资源的获取与释放。通过简洁的代码示例，我们将展示 `contextmanager` 如何简化文件操作、数据库连接、临时配置修改等场景下的资源管理流程，告别繁琐的 `__enter__` 和 `__exit__` 方法。同时，我们还将探讨其异常处理机制，确保代码的健壮性。掌握 `contextmanager`，让你的 Python 代码更简洁、更高效！

答案：contextmanager是contextlib提供的装饰器，将生成器函数变为上下文管理器，yield前为__enter__、后为__exit__，用于简化资源管理。

在 Python 中，contextmanager 是 contextlib 模块提供的一个装饰器，用于将一个生成器函数变成一个上下文管理器。这意味着你可以用 with 语句来使用这个函数，自动管理资源的获取与释放。

基本用法

被 @contextmanager 装饰的函数需要是一个生成器，且只包含一个 yield 语句。yield 之前的部分相当于 __enter__ 方法，之后的部分相当于 __exit__ 方法。

from contextlib import contextmanager
<p>@contextmanager
def my_context():
print("进入上下文")
try:
yield "资源"
finally:
print("退出上下文")</p><h1>使用</h1><p>with my_context() as res:
print(f"使用 {res}")</p>

输出：

进入上下文
使用 资源
退出上下文

实际应用场景

常见用途包括文件操作、数据库连接、临时修改配置等需要“准备-使用-清理”流程的场景。

• 临时修改全局状态：比如切换工作目录
• 异常安全的资源管理：确保即使出错也能正确释放资源
• 简化重复代码：把 setup 和 teardown 封装在一起

@contextmanager
def change_dir(path):
    import os
    old_dir = os.getcwd()
    os.chdir(path)
    try:
        yield
    finally:
        os.chdir(old_dir)
<h1>使用</h1><p>with change_dir("/tmp"):
print(os.listdir("."))</p>

异常处理机制

yield 语句可能抛出异常（如被 with 块中的代码引发），可以用 try-except-finally 结构捕获并处理。

@contextmanager
def safe_context():
    print("准备资源")
    try:
        yield
    except ValueError as e:
        print(f"捕获到 ValueError: {e}")
        # 可选择是否继续传播异常
    finally:
        print("清理资源")

如果不在 except 中重新 raise，异常会被吞掉。若需传递异常，可在处理后加 raise。

基本上就这些。用 @contextmanager 写上下文管理器比定义类更简洁，适合逻辑简单的场景。

到这里，我们也就讲完了《Pythoncontextmanager使用详解》的内容了。个人认为，基础知识的学习和巩固，是为了更好的将其运用到项目中，欢迎关注golang学习网公众号，带你了解更多关于的知识点！