用上下文管理器，让Python资源管理更优雅

用上下文管理器，让Python资源管理更优雅

在Python编程中，资源管理是个常见且重要的话题。无论是文件操作、数据库连接还是线程锁，我们都需要确保使用后正确释放资源。今天介绍一个让代码更简洁、更安全的特性——上下文管理器（Context Manager）。

with语句的魔法

你可能已经见过这样的代码：

with open('file.txt', 'r') as f:
    content = f.read()

这背后的魔法就是上下文管理器。使用with语句可以确保文件在使用后自动关闭，即使发生异常也不例外。

创建自己的上下文管理器

除了内置支持，我们也可以自定义上下文管理器。最简单的方式是使用contextlib模块：

from contextlib import contextmanager

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

# 使用
with timer():
    # 执行一些操作
    time.sleep(1)

更面向对象的方式

也可以通过实现__enter__和__exit__方法来创建上下文管理器：

class DatabaseConnection:
    def __init__(self, db_name):
        self.db_name = db_name

    def __enter__(self):
        self.connection = connect(self.db_name)
        return self.connection

    def __exit__(self, exc_type, exc_val, exc_tb):
        self.connection.close()

为什么使用上下文管理器？

1. 代码更简洁：避免了繁琐的try-finally语句
2. 更安全：确保资源总是被正确释放
3. 更可读：清晰地显示了资源的生命周期

上下文管理器体现了Python“优雅胜于复杂”的设计哲学。在日常开发中合理使用它，能让你的代码更加健壮和易于维护。