在Python中，资源管理（如文件操作、数据库连接、线程锁等）是一个常见的需求。传统的 try-finally 写法虽然可行，但代码冗长且不够直观。而 contextmanager 装饰器可以让你用更简洁、更优雅的方式实现上下文管理！
1. 什么是contextmanager?contextmanager 是 Python contextlib 模块提供的一个装饰器，它能将一个生成器函数转换为上下文管理器。通过 with 语句，你可以轻松管理资源的获取和释放。
传统写法 vs contextmanager
传统方式（try-finally）
file = open("example.txt", "w")try: file.write("Hello, World!")finally: file.close()使用 contextmanager
from contextlib import contextmanager@contextmanagerdef open_file(filename, mode): file = open(filename, mode) try: yield file # 返回资源，供 with 块使用 finally: file.close() # 确保资源释放# 使用方式with open_file("example.txt", "w") as f: f.write("Hello, ContextManager!")优势：代码更简洁，逻辑更清晰，资源管理更安全！
2. contextmanager的工作原理@contextmanager 装饰器将生成器函数分为三部分：
yield 之前：相当于 __enter__，负责资源分配（如打开文件）。yield：返回资源，供 with 块使用。yield 之后：相当于 __exit__，负责清理资源（如关闭文件）。如果发生异常，contextmanager 会自动确保资源被正确释放！
3. 实际应用场景（1）文件操作
@contextmanagerdef open_json(file_path): import json with open(file_path, "r") as f: yield json.load(f) # 自动解析 JSON，并确保文件关闭with open_json("data.json") as data: print(data["key"])（2）数据库连接管理
@contextmanagerdef db_connection(db_url): conn = connect_to_db(db_url) try: yield conn finally: conn.close() # 确保连接关闭with db_connection("postgres://user:pass@localhost/db") as conn: cursor = conn.cursor() cursor.execute("SELECT * FROM users")（3）临时环境切换
@contextmanagerdef switch_dir(new_dir): import os old_dir = os.getcwd() os.chdir(new_dir) try: yield finally: os.chdir(old_dir) # 切换回原目录with switch_dir("/tmp"): print("当前目录:", os.getcwd()) # 在 /tmp 下操作4. 进阶用法：异常处理contextmanager 也可以捕获和处理异常：
@contextmanagerdef handle_errors(): try: yield except ValueError as e: print(f"捕获到异常: {e}") else: print("没有异常发生！")with handle_errors(): x = int("123") # 正常执行with handle_errors(): x = int("abc") # 触发 ValueError5. 总结✅ contextmanager 的优势：
代码更简洁，避免重复的 try-finally。资源管理更安全，确保 __exit__ 逻辑执行。适用于文件、数据库、锁、临时环境切换等场景。赶快用 contextmanager 优化你的代码吧！
from contextlib import contextmanager# 定义你的自定义上下文管理器@contextmanagerdef my_resource_manager(): # 在这里管理资源！ yield resource # 在这里释放资源！希望这篇文章能帮助你掌握 contextmanager，写出更优雅的 Python 代码！
