Python 上下文管理器：从入门到实战
在 Python 编程中，资源管理是一个永恒的话题。无论是打开文件、连接数据库，还是获取网络资源，我们都需要确保在使用完毕后正确释放这些资源。传统的 try-finally 模式虽然有效，但代码冗长且容易出错。Python 的上下文管理器（Context Manager）提供了一种更优雅的解决方案。
你可能已经熟悉这样的代码模式：
with open('data.txt', 'r') as f:
content = f.read()
这段简洁的代码背后，是 Python 上下文管理器协议在起作用。但你知道吗？你可以创建自己的上下文管理器来解决各种实际问题？
一、上下文管理器的工作原理
上下文管理器通过实现两个特殊方法来工作:__enter__ 和 __exit__。当进入 with 块时调用 __enter__，退出时调用 __exit__。
class FileManager:
def __enter__(self):
return open(self.filename, self.mode)
def __exit__(self, exc_type, exc_val, exc_tb):
pass
二、使用 contextlib 简化代码
Python 的 contextlib 模块提供了更简洁的方式来创建上下文管理器。使用 @contextmanager 装饰器，你可以用生成器函数代替类。
from contextlib import contextmanager
@contextmanager
def timer(description):
start = time.perf_counter()
try:
yield
finally:
end = time.perf_counter()
print(f"{description} 耗时: {end - start}")
三、实战:数据库连接管理
在实际项目中，上下文管理器最常见的使用场景是资源管理。创建一个数据库连接的上下文管理器可以确保连接正确关闭，事务正确处理。
@contextmanager
def db_connection(db_path):
conn = sqlite3.connect(db_path)
try:
yield conn
conn.commit()
except:
conn.rollback()
raise
finally:
conn.close()
四、实战:临时目录管理
临时文件管理是另一个实用场景。使用上下文管理器可以自动创建和清理临时目录。
@contextmanager
def temporary_directory():
temp_dir = tempfile.mkdtemp()
try:
yield Path(temp_dir)
finally:
shutil.rmtree(temp_dir)
五、嵌套上下文管理器
你可以同时使用多个上下文管理器，Python 会按照顺序进入和退出。
with step("初始化"), step("加载配置"), step("处理数据"):
config = {"debug": True}
data = [1, 2, 3, 4, 5]
六、进阶:异步上下文管理器
在异步编程中，你也可以创建异步上下文管理器，使用 async with 语法。
class AsyncResource:
async def __aenter__(self):
await asyncio.sleep(0.1)
return self
async def __aexit__(self, exc_type, exc_val, exc_tb):
await asyncio.sleep(0.1)
总结
上下文管理器是 Python 中强大而优雅的工具。通过本文，你学习了上下文管理器的基本原理、使用 contextlib 简化代码、实际应用场景包括数据库连接和临时目录管理，以及嵌套和异步上下文管理器。掌握这些技巧后，你可以编写更健壮、更易维护的代码。
