3、上下文管理器：with语句的原理与最佳实践
说实话，刚学Python那会儿，我总觉得with open() as f就是个“打开文件的固定写法”。直到有一次，我在一个数据处理脚本里忘了关文件句柄，跑了一整天后程序直接崩了——打开的文件太多，系统资源耗尽。嗯，从那以后，我才真正理解了with语句的价值。
3.1 为什么需要上下文管理器？
你想想看，日常开发中我们经常需要这样的操作：
- 打开一个文件，用完要关闭
- 获取一个锁，用完要释放
- 连接数据库，用完要断开
- 开启一个网络连接，用完要回收
这些操作都有一个共同点：有借有还。如果只借不还，资源就会泄漏。我见过最夸张的一次，一个同事的爬虫跑了三天，打开了两万多个文件描述符没关，最后服务器直接卡死。
传统的做法是try...finally：
f = open('data.txt', 'r')
try:
data = f.read()
finally:
f.close()
这样写没问题，但每次都要写try...finally，代码显得很啰嗦。上下文管理器就是来解决这个痛点的。
3.2 with语句的工作原理
说白了，with语句就是try...finally的语法糖。但它背后有一套完整的协议——上下文管理协议。
一个对象要能用在with语句里，必须实现两个魔术方法：
| 方法 | 触发时机 | 作用 |
|---|---|---|
__enter__(self) |
进入with块时 |
获取资源，返回值赋给as后的变量 |
__exit__(self, exc_type, exc_val, exc_tb) |
离开with块时 |
释放资源，处理异常 |
执行流程是这样的：
- 调用
__enter__，拿到资源 - 执行
with块内的代码 - 无论是否发生异常，都会调用
__exit__ - 如果
__exit__返回True，异常会被吞掉
我画了一张图，帮你理清这个流程：
3.3 自定义上下文管理器
除了文件、锁这些内置对象，我们也可以自己写上下文管理器。我个人习惯用两种方式：
方式一：基于类实现
class Timer:
"""一个简单的计时器上下文管理器"""
def __enter__(self):
import time
self.start = time.time()
return self # 返回给 as 后面的变量
def __exit__(self, exc_type, exc_val, exc_tb):
import time
self.elapsed = time.time() - self.start
print(f"耗时: {self.elapsed:.3f}秒")
# 返回 False 表示不吞掉异常
return False
# 使用
with Timer() as t:
sum(range(1000000))
print(f"代码块耗时: {t.elapsed:.3f}秒")
方式二：基于 contextlib 模块
Python标准库提供了contextlib，用起来更简洁。我特别喜欢@contextmanager装饰器：
from contextlib import contextmanager
@contextmanager
def timer():
import time
start = time.time()
try:
yield # 这里就是 with 块执行的地方
finally:
elapsed = time.time() - start
print(f"耗时: {elapsed:.3f}秒")
# 使用
with timer():
sum(range(1000000))
💡 我的建议：如果逻辑简单，用
@contextmanager装饰器更省事。如果需要在__enter__里做复杂的资源初始化，用类实现更清晰。
3.4 最佳实践与避坑指南
这些年用下来，我总结了几个要点：
- 能用 with 就别手动 close —— 这是最基本的原则。手动关闭容易遗漏，尤其是在异常分支里。
- 多个资源可以写在一个 with 里 —— Python 3.1+ 支持这样写：
with open('a') as f1, open('b') as f2:，代码更紧凑。 - 注意 __exit__ 的返回值 —— 返回
True会吞掉异常。我曾经在代码里不小心返回了True，结果异常被静默吞掉，排查了半天才发现。
⚠️ 我曾经踩过的坑：在
__exit__里又抛出了新的异常。比如关闭文件时又出错了，这时候新异常会覆盖原始异常。建议在__exit__里做好异常处理，不要轻易往外抛。
3.5 实际项目中的应用
我在做数据库操作时，经常这样封装：
from contextlib import contextmanager
import sqlite3
@contextmanager
def db_connection(db_path):
"""数据库连接上下文管理器"""
conn = sqlite3.connect(db_path)
try:
yield conn
conn.commit() # 正常退出时提交
except:
conn.rollback() # 异常时回滚
raise
finally:
conn.close() # 无论如何都关闭连接
# 使用
with db_connection('test.db') as conn:
cursor = conn.execute('SELECT * FROM users')
for row in cursor:
print(row)
你看，这样写的好处是：提交、回滚、关闭全自动处理，调用方只需要关注业务逻辑。这就是上下文管理器的精髓——把“借还”的脏活累活封装起来，让代码更干净。
核心要点回顾：
with语句 =__enter__+__exit__的语法糖- 无论是否异常，
__exit__一定会被调用 - 用类实现或
@contextmanager都可以，看场景选 - 资源管理优先用
with，别手动close
嗯，上下文管理器就聊到这儿。说白了它就是Python帮你管理资源的“管家”，你只管用，它帮你收拾。下次写文件操作的时候，记得用with，别让自己再掉进资源泄漏的坑里。