用上下文管理器，让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()
为什么使用上下文管理器？
- 代码更简洁：避免了繁琐的try-finally语句
- 更安全：确保资源总是被正确释放
- 更可读：清晰地显示了资源的生命周期
上下文管理器体现了Python“优雅胜于复杂”的设计哲学。在日常开发中合理使用它，能让你的代码更加健壮和易于维护。
