0

0

Python 上下文管理器的自定义实现

冷炫風刃

冷炫風刃

发布时间:2026-01-30 16:24:08

|

475人浏览过

|

来源于php中文网

原创

__enter__和__exit__必须成对出现,因为with语句依赖二者完成资源获取与清理的完整生命周期;缺__exit__会报AttributeError,且无法保证异常路径下资源释放。

python 上下文管理器的自定义实现

为什么 __enter____exit__ 必须成对出现

因为 with 语句的底层机制依赖这两个方法构成完整生命周期:进入时调用 __enter__ 获取资源,退出时无条件触发 __exit__ 做清理。缺一不可,否则会报 AttributeError: __exit__

常见错误是只写了 __enter__(比如想简单返回个值),结果 with 块结束时找不到退出逻辑,资源泄漏或异常被吞掉。

  • __enter__ 应该返回要绑定到 as 变量的对象(可以是 self,也可以是其他实例)
  • __exit__ 必须接收三个参数:exc_typeexc_valuetraceback,即使你什么都不做也得声明它们
  • 如果 __exit__ 返回 True,会抑制异常;返回 NoneFalse 则异常照常抛出

如何让自定义上下文管理器支持异常传播与部分抑制

关键在 __exit__ 的返回值逻辑。不是“捕获异常”,而是“决定是否让异常继续向上冒泡”。比如日志记录后仍要抛出,就不能 return True。

典型场景:打开文件写入,希望 IO 错误暴露给调用方,但临时锁必须释放。

立即学习Python免费学习笔记(深入)”;

def __exit__(self, exc_type, exc_value, traceback):
    self.lock.release()  # 总要释放
    if exc_type is not None and issubclass(exc_type, ValueError):
        print("忽略 ValueError")
        return True  # 抑制这一类
    return False  # 其他异常照常抛出
  • exc_type is not None 判断是否有异常发生
  • issubclass(exc_type, SomeException) 精确匹配类型,避免用字符串比较
  • 不要在 __exit__ 里主动 raise 新异常——这会覆盖原始异常,极难调试

contextlib.contextmanager 装饰器替代手写类的适用边界

当逻辑简单、无状态、不需复用实例时,@contextmanager 更轻量;但一旦涉及属性缓存、多次进入、或需要继承扩展,就必须用类。

多奥淘宝客程序API免费版 F8.0
多奥淘宝客程序API免费版 F8.0

多奥淘宝客程序免费版拥有淘宝客站点的基本功能,手动更新少,管理简单等优点,适合刚接触网站的淘客们,或者是兼职做淘客们。同样拥有VIP版的模板引擎技 术、强大的文件缓存机制,但没有VIP版的伪原创跟自定义URL等多项创新的搜索引擎优化技术,除此之外也是一款高效的API数据系统实现无人值守全自动 化运行的淘宝客网站程序。4月3日淘宝联盟重新开放淘宝API申请,新用户也可使用了

下载

例如临时切换工作目录,用函数式写法很干净:

from contextlib import contextmanager
import os

@contextmanager def cd(path): old = os.getcwd() os.chdir(path) try: yield finally: os.chdir(old)

  • yield 之前的部分相当于 __enter__,之后(finally 块)相当于 __exit__
  • 不能在 @contextmanager 函数里返回值给 as 绑定——除非用 yield value,且该值会在 with 块中可用
  • 装饰器版本无法被继承,也不支持 isinstance(obj, ContextManager) 检查(它只是个 generator function)

测试自定义上下文管理器是否真正“安全退出”

不能只测正常流程,重点验证异常路径下资源是否释放。比如文件句柄、网络连接、数据库事务等,漏掉异常分支就等于埋雷。

最直接的方式是故意抛异常并检查状态:

with MyResource() as r:
    assert r.is_open
    raise KeyError("boom")
# 此处断言 r.is_open == False,否则 __exit__ 没生效
  • pytest.raises 包裹整个 with 块,再检查退出后状态
  • 避免在 __exit__ 里做耗时操作(如网络请求),它可能在异常处理链中被多次调用
  • 如果上下文对象本身有状态(如计数器、缓存),确保 __exit__ 不依赖未初始化的属性——__enter__ 失败时 __exit__ 仍会被调用

实际用起来最难的不是写两个方法,而是预判所有退出路径——正常结束、returnbreak、各种异常、甚至 os._exit()。只要没覆盖全,就存在资源残留风险。

热门AI工具

更多
DeepSeek
DeepSeek

幻方量化公司旗下的开源大模型平台

豆包大模型
豆包大模型

字节跳动自主研发的一系列大型语言模型

通义千问
通义千问

阿里巴巴推出的全能AI助手

腾讯元宝
腾讯元宝

腾讯混元平台推出的AI助手

文心一言
文心一言

文心一言是百度开发的AI聊天机器人,通过对话可以生成各种形式的内容。

讯飞写作
讯飞写作

基于讯飞星火大模型的AI写作工具,可以快速生成新闻稿件、品宣文案、工作总结、心得体会等各种文文稿

即梦AI
即梦AI

一站式AI创作平台,免费AI图片和视频生成。

ChatGPT
ChatGPT

最最强大的AI聊天机器人程序,ChatGPT不单是聊天机器人,还能进行撰写邮件、视频脚本、文案、翻译、代码等任务。

相关专题

更多
java中break的作用
java中break的作用

本专题整合了java中break的用法教程,阅读专题下面的文章了解更多详细内容。

118

2025.10.15

java break和continue
java break和continue

本专题整合了java break和continue的区别相关内容,阅读专题下面的文章了解更多详细内容。

258

2025.10.24

js 字符串转数组
js 字符串转数组

js字符串转数组的方法:1、使用“split()”方法;2、使用“Array.from()”方法;3、使用for循环遍历;4、使用“Array.split()”方法。本专题为大家提供js字符串转数组的相关的文章、下载、课程内容,供大家免费下载体验。

319

2023.08.03

js截取字符串的方法
js截取字符串的方法

js截取字符串的方法有substring()方法、substr()方法、slice()方法、split()方法和slice()方法。本专题为大家提供字符串相关的文章、下载、课程内容,供大家免费下载体验。

212

2023.09.04

java基础知识汇总
java基础知识汇总

java基础知识有Java的历史和特点、Java的开发环境、Java的基本数据类型、变量和常量、运算符和表达式、控制语句、数组和字符串等等知识点。想要知道更多关于java基础知识的朋友,请阅读本专题下面的的有关文章,欢迎大家来php中文网学习。

1502

2023.10.24

字符串介绍
字符串介绍

字符串是一种数据类型,它可以是任何文本,包括字母、数字、符号等。字符串可以由不同的字符组成,例如空格、标点符号、数字等。在编程中,字符串通常用引号括起来,如单引号、双引号或反引号。想了解更多字符串的相关内容,可以阅读本专题下面的文章。

624

2023.11.24

java读取文件转成字符串的方法
java读取文件转成字符串的方法

Java8引入了新的文件I/O API,使用java.nio.file.Files类读取文件内容更加方便。对于较旧版本的Java,可以使用java.io.FileReader和java.io.BufferedReader来读取文件。在这些方法中,你需要将文件路径替换为你的实际文件路径,并且可能需要处理可能的IOException异常。想了解更多java的相关内容,可以阅读本专题下面的文章。

653

2024.03.22

php中定义字符串的方式
php中定义字符串的方式

php中定义字符串的方式:单引号;双引号;heredoc语法等等。想了解更多字符串的相关内容,可以阅读本专题下面的文章。

609

2024.04.29

C++ 设计模式与软件架构
C++ 设计模式与软件架构

本专题深入讲解 C++ 中的常见设计模式与架构优化,包括单例模式、工厂模式、观察者模式、策略模式、命令模式等,结合实际案例展示如何在 C++ 项目中应用这些模式提升代码可维护性与扩展性。通过案例分析,帮助开发者掌握 如何运用设计模式构建高质量的软件架构,提升系统的灵活性与可扩展性。

8

2026.01.30

热门下载

更多
网站特效
/
网站源码
/
网站素材
/
前端模板

精品课程

更多
相关推荐
/
热门推荐
/
最新课程
最新Python教程 从入门到精通
最新Python教程 从入门到精通

共4课时 | 22.4万人学习

Django 教程
Django 教程

共28课时 | 3.6万人学习

SciPy 教程
SciPy 教程

共10课时 | 1.3万人学习

关于我们 免责申明 举报中心 意见反馈 讲师合作 广告合作 最新更新
php中文网:公益在线php培训,帮助PHP学习者快速成长!
关注服务号 技术交流群
PHP中文网订阅号
每天精选资源文章推送

Copyright 2014-2026 https://www.php.cn/ All Rights Reserved | php.cn | 湘ICP备2023035733号