Python 上下文管理器的 5 个实用技巧，让你的代码更优雅
在 Python 编程中，上下文管理器（Context Manager）是一个优雅的资源管理工具。你可能已经熟悉最常见的用法——使用 with 语句打开文件，但上下文管理器的能力远不止于此。今天，我将分享 5 个实用技巧，帮助你更好地掌握这个强大的特性。
技巧一：理解 with 语句的本质
with 语句的核心是协议：任何实现了 __enter__ 和 __exit__ 方法的对象都可以作为上下文管理器。__enter__ 方法在进入上下文时执行，返回值赋给 as 后的变量；__exit__ 方法在退出上下文时执行，负责清理资源。
来看一个自定义上下文管理器的例子：
class Timer:
def __init__(self, name="代码块"):
self.name = name
self.start_time = None
def __enter__(self):
import time
self.start_time = time.time()
print(f"开始执行：{self.name}")
return self
def __exit__(self, exc_type, exc_val, exc_tb):
import time
elapsed = time.time() - self.start_time
print(f"执行完成：{self.name}，耗时 {elapsed:.4f} 秒")
return False
with Timer("数据处理") as timer:
result = sum(i * i for i in range(1000000))
这个简单的计时器展示了上下文管理器的基本结构。注意 __exit__ 方法接收三个参数：异常类型、异常值和 traceback 对象。如果 __exit__ 返回 True，异常将被抑制；返回 False 则正常传播。
技巧二：使用 contextlib 简化实现
Python 标准库的 contextlib 模块提供了更简洁的实现方式。使用 @contextmanager 装饰器，你可以用生成器函数代替类：
from contextlib import contextmanager
import time
@contextmanager
def timer(name="代码块"):
start = time.time()
print(f"开始：{name}")
try:
yield
finally:
elapsed = time.time() - start
print(f"结束：{name}，耗时 {elapsed:.4f} 秒")
with timer("数据库查询"):
time.sleep(0.5)
生成器方式的代码量更少，适合简单的上下文管理场景。yield 关键字是关键：它之前的代码相当于 __enter__，之后的代码相当于 __exit__。
技巧三：处理多个上下文管理器
当需要同时管理多个资源时，可以使用多个 with 语句嵌套，或者在 Python 3.10+ 中使用括号将多个上下文管理器组合：
from contextlib import contextmanager
@contextmanager
def database_connection():
print("连接数据库...")
conn = {"connected": True}
try:
yield conn
finally:
print("关闭数据库连接")
@contextmanager
def file_handler(filename, mode):
print(f"打开文件：{filename}")
f = {"filename": filename, "mode": mode}
try:
yield f
finally:
print(f"关闭文件：{filename}")
with (
database_connection() as db,
file_handler("data.txt", "r") as f
):
print(f"数据库：{db}, 文件：{f}")
这种写法不仅代码更整洁，而且所有资源都会在退出时按正确顺序清理，即使发生异常也能保证资源释放。
技巧四：创建可重用的上下文管理器库
在实际项目中，你可以创建自己的上下文管理器工具库。以下是一些实用场景：
from contextlib import contextmanager
import os
import tempfile
import shutil
@contextmanager
def temporary_directory(prefix="tmp"):
temp_dir = tempfile.mkdtemp(prefix=prefix)
try:
yield temp_dir
finally:
shutil.rmtree(temp_dir)
print(f"已清理临时目录：{temp_dir}")
@contextmanager
def change_directory(path):
original_dir = os.getcwd()
try:
os.chdir(path)
yield
finally:
os.chdir(original_dir)
with temporary_directory() as tmpdir:
with open(os.path.join(tmpdir, "test.txt"), "w") as f:
f.write("临时数据")
with change_directory("/tmp"):
print(f"当前目录：{os.getcwd()}")
这些可重用的上下文管理器可以大幅减少样板代码，让业务逻辑更清晰。
技巧五：异常处理与资源清理的最佳实践
上下文管理器的核心价值在于确保资源正确清理，即使发生异常。以下是几个关键原则：
from contextlib import suppress
import os
with suppress(FileNotFoundError, KeyError):
os.remove("不存在的文件.txt")
def process_data_file(input_path, output_path):
with (
open(input_path, "r") as infile,
open(output_path, "w") as outfile
):
data = infile.read()
processed = data.upper()
outfile.write(processed)
总结
上下文管理器是 Python 中实现 RAII 模式的核心工具。通过掌握这 5 个技巧，你可以理解 with 语句的工作原理、使用 contextlib 简化代码、优雅地管理多个资源、创建可重用的工具库，以及确保异常安全与资源清理。
在实际项目中，良好的资源管理不仅能避免内存泄漏和文件句柄耗尽等问题，还能让代码更加简洁易读。下次当你需要处理文件、数据库连接、网络请求或任何需要清理的资源时，考虑使用上下文管理器吧！
