
Python 的 with 语句背后依赖的是上下文管理器协议,其核心是对象必须实现 __enter__ 和 __exit__ 两个特殊方法。
上下文管理器协议的两个关键方法
任何对象只要定义了以下两个方法,就能被 with 语句使用:
-
__enter__(self):在进入with块时自动调用,通常用于资源获取(如打开文件、连接数据库),返回值会绑定到as后的变量(若无as子句,返回值被忽略) -
__exit__(self, exc_type, exc_value, traceback):在离开with块时**必定执行**(无论是否发生异常),用于清理资源(如关闭文件、回滚事务)。三个参数分别表示异常类型、异常值、追踪信息;若正常退出,三者均为None
with 语句的等价展开形式
下面这段代码:
with open('data.txt') as f:
data = f.read()
实际等价于:
立即学习“Python免费学习笔记(深入)”;
f = open('data.txt')
try:
data = f.read()
finally:
f.close()
但更准确地说,with 的底层逻辑是:
【极品模板】出品的一款功能强大、安全性高、调用简单、扩展灵活的响应式多语言企业网站管理系统。 产品主要功能如下: 01、支持多语言扩展(独立内容表,可一键复制中文版数据) 02、支持一键修改后台路径; 03、杜绝常见弱口令,内置多种参数过滤、有效防范常见XSS; 04、支持文件分片上传功能,实现大文件轻松上传; 05、支持一键获取微信公众号文章(保存文章的图片到本地服务器); 06、支持一键
- 调用
f.__enter__(),将返回值赋给f - 执行
with块内语句 - 无论是否异常,都调用
f.__exit__(...);若该方法返回真值(True),则抑制异常(不向上抛出);否则异常继续传播
手动实现一个上下文管理器
比如封装一个简单的计时器:
import time
<p>class Timer:
def <strong>enter</strong>(self):
self.start = time.time()
return self</p><pre class="brush:php;toolbar:false;">def __exit__(self, exc_type, exc_val, exc_tb):
self.end = time.time()
print(f'耗时: {self.end - self.start:.2f} 秒')使用
with Timer() as t: time.sleep(1)
输出:耗时: 1.00 秒。注意 __exit__ 在 sleep 结束后立即执行,哪怕中间抛出异常也照常运行。
@contextmanager 装饰器简化写法
如果不想写完整类,可用标准库 contextlib 提供的 @contextmanager:
from contextlib import contextmanager
<p>@contextmanager
def timer():
start = time.time()
try:
yield # 暂停,把控制权交给 with 块
finally:
end = time.time()
print(f'耗时: {end - start:.2f} 秒')
</p>原理上,它把函数体拆成 __enter__(yield 前)和 __exit__(finally 块),yield 的值即为 as 接收的对象。









