Python上下文管理器原理_with语句实现机制

6次阅读

Python上下文管理器原理_with语句实现机制

pythonwith 语句背后依赖的是上下文管理器协议,其核心是对象必须实现 __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 的底层逻辑是:

  • 调用 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 接收的对象。

text=ZqhQzanResources