`contextmanager` 是 Python 中 `contextlib` 模块提供的一个装饰器，用于**简化上下文管理器（context manager）的创建**。它允许你使用一个生成器函数来定义 `__enter__` 和 `__exit__` 方法，而无需手动编写一个类。 ### 上下文管理器的作用： 上下文管理器用于管理资源的获取与释放，例如： - 打开和关闭文件 - 获取和释放锁 - 连接和断开数据库 使用 `with` 语句可以自动管理这些资源，确保即使在发生异常的情况下也能正确释放。 --- ### 使用 `@contextmanager` 的方式： 你只需定义一个生成器函数，函数中使用 `yield` 分隔 `__enter__` 和 `__exit__` 的逻辑： - `yield` 之前的代码会在 `with` 块开始前执行（相当于 `__enter__`） - `yield` 后的代码会在 `with` 块结束后执行（相当于 `__exit__`） --- ### 示例代码： ```python from contextlib import contextmanager @contextmanager def simple_context_manager(name): print(f"Entering the context: {name}") try: yield name.upper() # 可以传递给 with 块中的变量 finally: print(f"Exiting the context: {name}") # 使用示例 with simple_context_manager("example") as value: print("Inside the context:", value) ``` ### 输出结果： ``` Entering the context: example Inside the context: EXAMPLE Exiting the context: example ``` --- ### 异常处理： `@contextmanager` 中的 `yield` 后面的代码会自动捕获异常，并确保资源被释放。如果你希望在 `with` 块中发生的异常继续抛出，可以在 `try...finally` 中处理。 --- ### 优点： | 优点 | 说明 | |------|------| | 简洁 | 不需要手动实现 `__enter__` 和 `__exit__` | | 易维护 | 逻辑清晰，资源获取和释放集中在一个函数中 | | 安全 | 配合 `try...finally` 能确保资源释放 | --- ### 与自定义上下文管理器的对比： | 方式 | 实现方式 | 优点 | 缺点 | |------|----------|------|------| | 类实现（`__enter__`/`__exit__`） | 更灵活，适合复杂场景 | 控制精细 | 代码量多 | | `@contextmanager` | 生成器函数 + 装饰器 | 简洁、易用 | 适合简单场景 | --- ### 示例：模拟文件操作 ```python @contextmanager def open_file(filename, mode): f = open(filename, mode) try: yield f finally: f.close() with open_file('test.txt', 'w') as f: f.write('Hello, world!') ``` ---