在Python开发中，上下文管理器是管理资源（如文件、数据库连接）的利器。传统的实现方法需要创建一个类并定义__enter__和__exit__两个方法，但Python的contextlib模块提供了更简洁的解决方案——@contextmanager装饰器。

传统上下文管理器的实现痛点

传统方式需要完整定义一个类，包含__enter__和__exit__方法：

class TraditionalManager:

    def __enter__(self):

    # 资源初始化

        return resource 

    def __exit__(self, exc_type, exc_val, exc_tb):

# 资源清理 

这种方式虽然可行，但代码冗余度高，尤其对简单资源管理来说显得笨重。

@contextmanager的革命性解决方案

使用@contextmanager装饰器，可用生成器函数替代完整类：

from contextlib import contextmanager

    @contextmanager

    def simple_manager():

    # 相当于__enter__部分的代码

    resource = setup_resource()

    try:

        yield resource  # 返回给as变量 

    finally:

    # 相当于__exit__部分的代码

        cleanup_resource()

核心机制解析

yield前代码：在进入with块时执行（类似__enter__）

yield值：作为as子句的目标变量值

yield后代码：在退出with块时执行（类似__exit__）

实战案例：输出反向器

创建一个反转所有输出文本的上下文管理器：

from contextlib import contextmanager

import sys 

@contextmanager 

def looking_glass():

    original_write = sys.stdout.write 

# 定义反转函数

def reverse_write(text):

        original_write(text[::-1])

# 替换标准输出 

    sys.stdout.write  = reverse_write 

try:

    yield'JABBERWOCKY'# 返回给as变量

except Exception as e:

        print(f"Error occurred: {e}")

raise

finally:

# 确保恢复原始输出

        sys.stdout.write  = original_write

使用效果：

with looking_glass() as word:

    print("Hello, Python!")

    print(word)

# 输出：

# !nohtyP ,olleH

# YKCOWREBBAJ

异常处理：关键注意事项

使用@contextmanager时必须正确处理异常：

@contextmanager 

def safe_manager():

    resource = acquire()

try:

    yield resource 

except CustomError as e:

        handle_error(e)

finally:

        release(resource)  # 必须确保资源释放

重要原则：

必须使用try/finally保证资源清理

默认情况下会压制异常（与常规上下文管理器行为相反）

如需传播异常，需在函数内重新raise

高级应用：原地文件重写

Martijn Pieters实现的原地文件重写上下文管理器是绝佳实践：

@contextmanager

def inplace(filename, mode='r', **kwargs):

# 创建备份

    backup = filename + '.bak'

    os.rename(filename,  backup)

try:

    with open(backup, mode, **kwargs) as infh:

    with open(filename, 'w' + mode.replace('r',  ''), **kwargs) as outfh:

    yield (infh, outfh)  # 同时提供读写句柄 

except:

# 出错时恢复备份

        os.rename(backup,  filename)

    raise

    finally: