让Python代码更优雅：深入理解上下文管理器

你是否曾为频繁打开关闭文件而烦恼？是否在异常处理时总担心资源没释放？Python的上下文管理器正是解决这些痛点的优雅方案。

基础用法：with语句

最常见的莫过于文件操作：

with open('data.txt', 'r') as f:
    content = f.read()
# 文件自动关闭，无需显式调用f.close()

with语句确保了资源被正确释放，即使发生异常也不例外。

自定义上下文管理器

实现上下文管理器有两种方式。第一种是定义__enter__和__exit__方法：

class DatabaseConnection:
    def __enter__(self):
        print("连接数据库")
        return self
    def __exit__(self, exc_type, exc_val, exc_tb):
        print("关闭连接")

with DatabaseConnection() as conn:
    # 执行数据库操作
    pass

更简洁的是使用contextlib模块的@contextmanager装饰器：

from contextlib import contextmanager

@contextmanager
def timed_block(name):
    import time
    start = time.time()
    try:
        yield
    finally:
        elapsed = time.time() - start
        print(f"{name} 耗时: {elapsed:.2f}秒")

with timed_block("数据处理"):
    # 需要计时的代码块
    time.sleep(1)

实际应用场景

上下文管理器的应用远不止资源管理。你可以用它实现：

• 临时修改环境变量
• 代码块计时统计
• 数据库事务自动提交/回滚
• 临时切换日志级别

掌握上下文管理器，让代码更简洁、更安全、更Pythonic。下次遇到需要“先做某事，最后清理”的场景，不妨用with语句封装起来！