Python实现上下文协议,自定义with对象
本文深入讲解了Python中上下文管理协议的核心机制——通过实现`__enter__`和`__exit__`方法,让自定义对象无缝支持`with`语句,实现资源的自动获取与安全释放;不仅剖析了协议执行流程(如返回值绑定、异常抑制逻辑),还通过计时器、安全文件操作等实用案例演示了手动实现类与`@contextmanager`装饰器两种高效方式,帮助开发者写出更简洁、健壮、符合Python惯用法的资源管理代码。

Python 中的上下文管理协议(Context Management Protocol)通过 __enter__ 和 __exit__ 两个特殊方法实现,让对象能用于 with 语句中,自动完成资源获取与释放,避免手动调用清理逻辑。
什么是上下文管理协议
只要一个类实现了 __enter__ 和 __exit__ 方法,它就支持上下文管理协议。执行 with obj as x: 时:
__enter__在进入with块前被调用,返回值赋给as后的变量(若无as,返回值被忽略);__exit__(exc_type, exc_value, traceback)在离开with块时被调用,无论是否发生异常都会执行,用于清理工作;- 若
__exit__返回True,会抑制异常(不向上抛出);返回None或False则正常传播异常。
写一个简单的自定义 with 对象
比如封装一个计时器,在进入时记录开始时间,退出时打印耗时:
import timeclass Timer: def init(self, name="block"): self.name = name self.start_time = None self.end_time = None
def __enter__(self): self.start_time = time.time() return self # 可选:返回自身或任意值供 as 使用 def __exit__(self, exc_type, exc_value, traceback): self.end_time = time.time() elapsed = self.end_time - self.start_time print(f"[{self.name}] 执行耗时: {elapsed:.4f} 秒") # 不返回 True,不抑制异常
使用方式:
with Timer("数据处理"):
time.sleep(1.2)
输出类似:[数据处理] 执行耗时: 1.2005 秒
带异常处理的上下文管理器
如果希望在发生特定异常时静默处理(如忽略 FileNotFoundError),可在 __exit__ 中判断并返回 True:
class SafeFileOpener:
def __init__(self, filename, mode="r"):
self.filename = filename
self.mode = mode
self.file = None
def __enter__(self):
self.file = open(self.filename, self.mode)
return self.file
def __exit__(self, exc_type, exc_value, traceback):
if self.file:
self.file.close()
if exc_type is FileNotFoundError:
print(f"警告:文件 {self.filename} 不存在,已跳过")
return True # 抑制 FileNotFoundError
return False # 其他异常照常抛出
这样写就不会因文件不存在而中断程序:
with SafeFileOpener("missing.txt") as f:
content = f.read() # 若文件不存在,只打印警告,不报错
更简洁的方式:用 contextlib.contextmanager
对于简单逻辑,无需写完整类,可用装饰器 @contextmanager 将生成器函数转为上下文管理器:
from contextlib import contextmanager@contextmanager def timer(name="block"): start = time.time() try: yield # 进入 with 块的位置 finally: end = time.time() print(f"[{name}] 耗时: {end - start:.4f} 秒")
使用方式相同
with timer("循环计算"): sum(i * i for i in range(10**6))
yield 之前是 __enter__,之后(含 finally)是 __exit__ 的等价逻辑,更轻量易读。
文中关于的知识介绍,希望对你的学习有所帮助!若是受益匪浅,那就动动鼠标收藏这篇《Python实现上下文协议,自定义with对象》文章吧,也可关注golang学习网公众号了解相关技术文章。
大学搜题酱错题本使用教程
- 上一篇
- 大学搜题酱错题本使用教程
- 下一篇
- B站弹幕发送者怎么查?方法全解析
-
- 文章 · python教程 | 1小时前 |
- Python zip strict=True 在第几次迭代发现长度不一致
- 437浏览 收藏
-
- 文章 · python教程 | 6小时前 | python · Python 类型提示 typing get_protocol_members Protocol
- Python get_protocol_members 怎么读取 Protocol 成员集合
- 487浏览 收藏
-
- 文章 · python教程 | 17小时前 | 文件处理 · python · Python TempFile 临时文件 SpooledTemporaryFile rollover
- Python SpooledTemporaryFile 怎么手动触发写入磁盘
- 307浏览 收藏
-
- 文章 · python教程 | 20小时前 |
- Python Path.info 缓存的文件类型信息什么时候会过期
- 438浏览 收藏
-
- 文章 · python教程 | 23小时前 |
- Python asyncio.Queue shutdown 后等待者会收到什么
- 124浏览 收藏
-
- 文章 · python教程 | 1天前 |
- Python asyncio.Barrier 等待任务被取消后会怎样
- 166浏览 收藏
-
- 文章 · python教程 | 1天前 |
- Python TaskGroup 怎么主动终止整组任务
- 242浏览 收藏
-
- 文章 · python教程 | 1天前 | SQLite · 数据一致性 · Python教程 · Python SQLite 数据库备份 sqlite3.Connection.backup
- Python sqlite3.Connection.backup 怎么在线复制数据库
- 264浏览 收藏
-
- 文章 · python教程 | 1天前 | python · Python zip zipfile zipfile.Path
- Python zipfile.Path 怎么像目录一样遍历压缩包
- 370浏览 收藏
-
- 文章 · python教程 | 1天前 | 内存优化 · Python教程 · Python 大数组 PickleBuffer pickle协议5
- Python PickleBuffer 怎么减少大数组复制
- 207浏览 收藏
-
- 文章 · python教程 | 1天前 | 标准库 · Python教程 · Python Traversable importlib.resources zipimport
- Python importlib.resources.files 怎么访问压缩包内资源
- 143浏览 收藏
-
- 前端进阶之JavaScript设计模式
- 设计模式是开发人员在软件开发过程中面临一般问题时的解决方案,代表了最佳的实践。本课程的主打内容包括JS常见设计模式以及具体应用场景,打造一站式知识长龙服务,适合有JS基础的同学学习。
- 543次学习
-
- GO语言核心编程课程
- 本课程采用真实案例,全面具体可落地,从理论到实践,一步一步将GO核心编程技术、编程思想、底层实现融会贯通,使学习者贴近时代脉搏,做IT互联网时代的弄潮儿。
- 516次学习
-
- 简单聊聊mysql8与网络通信
- 如有问题加微信:Le-studyg;在课程中,我们将首先介绍MySQL8的新特性,包括性能优化、安全增强、新数据类型等,帮助学生快速熟悉MySQL8的最新功能。接着,我们将深入解析MySQL的网络通信机制,包括协议、连接管理、数据传输等,让
- 500次学习
-
- JavaScript正则表达式基础与实战
- 在任何一门编程语言中,正则表达式,都是一项重要的知识,它提供了高效的字符串匹配与捕获机制,可以极大的简化程序设计。
- 487次学习
-
- 从零制作响应式网站—Grid布局
- 本系列教程将展示从零制作一个假想的网络科技公司官网,分为导航,轮播,关于我们,成功案例,服务流程,团队介绍,数据部分,公司动态,底部信息等内容区块。网站整体采用CSSGrid布局,支持响应式,有流畅过渡和展现动画。
- 485次学习
-
- PubMedQA
- 深入了解PubMedQA生物医学问答数据集,涵盖其核心功能、使用方法及在临床决策、药物研发等场景的应用,助力提升NLP模型性能。
- 350次使用
-
- H2O EvalGPT
- H2O EvalGPT是H2O.ai推出的开源LLM评估平台,提供详细的大模型性能排行榜、行业特定基准测试及A/B测试功能,助您快速选择最适合项目的高性能大语言模型。
- 412次使用
-
- LMArena
- LMArena是加州大学伯克利分校推出的AI模型匿名评测平台。通过盲测投票机制,用户可对比不同大模型回答并生成实时排行榜,助力开发者优化模型及用户选择最佳AI工具。
- 417次使用
-
- HELM
- 深入了解斯坦福推出的HELM(Holistic Evaluation of Language Models)大模型评测体系。本文解析其核心功能、安装配置步骤及应用场景,涵盖准确性、公平性、鲁棒性等多维度指标,助力开发者全面优化语言模型性能。
- 373次使用
-
- MMBench
- MMBench是由上海人工智能实验室等机构联合推出的多模态基准测试平台,提供细粒度能力评估、大规模数据集及VLMEvalKit工具。本文详细介绍其核心功能、安装使用方法及应用场景,助力开发者全面评估多模态模型性能。
- 197次使用
-
- Python sqlite3 Connection serialize 怎么导出数据库快照:备份窗口、内存占用与恢复校验
- 2026-08-26 501浏览
-
- Python监控网页状态:requests异常处理实战
- 2026-05-29 501浏览
-
- TensorFlow模型部署为API的TF Serving方法
- 2026-05-26 501浏览
-
- Python字符串编码转换:encode与decode详解
- 2026-05-16 501浏览
-
- TensorFlow裁剪无用算子方法详解
- 2026-05-15 501浏览

