Python 怎么用 contextmanager 装饰器简化上下文管理器？

Question

仓乃文 · Accepted Answer

自定义上下文管理器需要写 __enter__ 和 __exit__ 方法，代码量大。contextmanager 装饰器把生成器函数转为上下文管理器，yield 之前是 __enter__，yield 之后是 __exit__，几行代码实现资源自动管理。

基本用法：
```python
from contextlib import contextmanager

@contextmanager
def timer(label):
    import time
    start = time.time()
    yield
    elapsed = time.time() - start
    print(f"{label}: {elapsed:.3f}s")

with timer("数据处理"):
    process_data()
# 输出：数据处理: 1.234s
```

数据库事务管理：
```python
@contextmanager
def transaction(conn):
    cursor = conn.cursor()
    try:
        yield cursor
        conn.commit()
    except Exception:
        conn.rollback()
        raise

with transaction(conn) as cur:
    cur.execute("INSERT INTO users VALUES (?, ?)", ("Tom", 18))
```

临时修改环境：
```python
import os

@contextmanager
def temp_env(key, value):
    old = os.environ.get(key)
    os.environ[key] = value
    yield
    if old is None:
        os.environ.pop(key, None)
    else:
        os.environ[key] = old

with temp_env("DEBUG", "1"):
    run_tests()  # 临时设置环境变量
```

补充说明：
- yield 的值就是 as 后面的变量
- yield 之前的代码在 __enter__ 时执行，之后的在 __exit__ 时执行
- 异常会从 yield 处抛出，用 try/except 包裹 yield 处理异常
- contextlib 还提供 closing、suppress、redirect_stdout 等实用工具
- 适合计时、锁、事务、临时配置等资源管理场景