0

0

Python 多资源嵌套管理的优雅写法

冷炫風刃

冷炫風刃

发布时间:2026-02-19 16:58:02

|

526人浏览过

|

来源于php中文网

原创

exitstack比嵌套with更灵活安全,支持运行时动态注册任意数量上下文管理器并逆序清理,避免缩进爆炸和资源泄漏;但不适用于异步场景,需改用asyncexitstack。

python 多资源嵌套管理的优雅写法

contextlib.ExitStack 为什么比多个 with 好用

嵌套多个 with 会缩进爆炸,动态增减资源时根本写不下去。而 ExitStack 允许你在运行时注册任意数量的上下文管理器,统一在退出时按逆序清理。

常见错误现象:手动调用 __enter__/__exit__、用 try/finally 模拟、或硬编码固定层数的 with —— 一旦某个资源初始化失败,前面已打开的资源可能漏关。

  • 必须在 with ExitStack() as stack: 块内注册资源,否则退出时不触发清理
  • stack.enter_context(cm) 返回 cm.__enter__() 的结果,可直接赋值给变量
  • 注册顺序和退出顺序相反,适合“先开后关”的依赖关系(比如文件要等数据库连接关了再关)
  • 若某次 enter_context 抛异常,已成功进入的资源仍会被正确退出
with ExitStack() as stack:
    db = stack.enter_context(get_db_connection())
    log_file = stack.enter_context(open("log.txt", "a"))
    cache = stack.enter_context(RedisPool())

哪些资源不能直接丢给 ExitStack

ExitStack 只负责调用 __exit__,不判断资源是否“真正需要关闭”。有些对象实现了上下文协议但语义特殊,硬塞进去反而出错。

使用场景:你拿到一个第三方库返回的类实例,不确定它是否适合被 ExitStack 管理。

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

橙篇
橙篇

百度文库发布的一款综合性AI创作工具

下载
  • threading.Lock 不该用 —— 它的 __exit__release(),但锁可能根本没 acquire 成功,直接 release 会报 RuntimeError: release unlocked lock
  • io.BytesIOStringIO 无需管理 —— 没有底层资源要释放,__exit__ 是空操作,加了白费事
  • 自定义类若只实现 __exit__ 但没做幂等处理(比如重复 close 导致崩溃),放进 ExitStack 就危险

想中途取消某个资源的自动清理,怎么操作

有时某个资源打开后发现条件不满足,你想让它跳过自动 __exit__ —— 比如预分配了一个临时文件,但后续逻辑决定直接用原文件,这个临时文件就得手动删、不走 close 流程。

参数差异:ExitStack.pop_all() 会把所有待清理项清空并返回新 ExitStack,但更常用的是 pop_all() 配合 callback 注册手动清理逻辑。

  • stack.callback(func, *args, **kwds) 注册一个无参函数,在退出时执行;它不关联任何上下文管理器,纯手动钩子
  • 若要彻底移除某个已注册的管理器,只能靠 pop_all() + 重新注册其余项 —— ExitStack 本身不支持“删中间一个”
  • 注意 callback 执行时机在所有 enter_context 对应的 __exit__ 之后,适合做收尾日志或清理副作用
with ExitStack() as stack:
    tmp = stack.enter_context(TemporaryDirectory())
    if not should_use_tmp():
        # 改为手动清理,跳过自动 rmtree
        stack.callback(shutil.rmtree, tmp.name)

async with 场景下 ExitStack 不管用,得换谁

ExitStack 是同步的,遇到 async with 直接报 TypeError: unhashable type: 'AsyncExitStack'。Python 3.7+ 提供了 contextlib.AsyncExitStack,API 几乎一致但专用于协程。

性能影响:异步栈的 enter_async_context 是 awaitable,每次注册都会挂起,高频注册(比如循环开 100 个 async 文件)会有可观调度开销。

  • 不能混用:enter_contextenter_async_context 必须分开用,不能在一个 AsyncExitStack 里塞同步资源
  • 若需同步+异步混合管理,只能拆成两个栈,各自处理,靠外层逻辑协调生命周期
  • 第三方异步库(如 aiofilesaiomysql)返回的对象,确认其实现了 __aenter__/__aexit__ 再传给 enter_async_context

容易被忽略的地方:很多开发者以为 ExitStack 是通用解法,直到第一次在 async def 里用它报错才意识到——异步上下文管理是另一套机制,没有银弹。

热门AI工具

更多
DeepSeek
DeepSeek

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

豆包大模型
豆包大模型

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

通义千问
通义千问

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

腾讯元宝
腾讯元宝

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

文心一言
文心一言

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

讯飞写作
讯飞写作

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

即梦AI
即梦AI

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

ChatGPT
ChatGPT

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

相关专题

更多
堆和栈的区别
堆和栈的区别

堆和栈的区别:1、内存分配方式不同;2、大小不同;3、数据访问方式不同;4、数据的生命周期。本专题为大家提供堆和栈的区别的相关的文章、下载、课程内容,供大家免费下载体验。

419

2023.07.18

堆和栈区别
堆和栈区别

堆(Heap)和栈(Stack)是计算机中两种常见的内存分配机制。它们在内存管理的方式、分配方式以及使用场景上有很大的区别。本文将详细介绍堆和栈的特点、区别以及各自的使用场景。php中文网给大家带来了相关的教程以及文章欢迎大家前来学习阅读。

594

2023.08.10

数据库三范式
数据库三范式

数据库三范式是一种设计规范,用于规范化关系型数据库中的数据结构,它通过消除冗余数据、提高数据库性能和数据一致性,提供了一种有效的数据库设计方法。本专题提供数据库三范式相关的文章、下载和课程。

374

2023.06.29

如何删除数据库
如何删除数据库

删除数据库是指在MySQL中完全移除一个数据库及其所包含的所有数据和结构,作用包括:1、释放存储空间;2、确保数据的安全性;3、提高数据库的整体性能,加速查询和操作的执行速度。尽管删除数据库具有一些好处,但在执行任何删除操作之前,务必谨慎操作,并备份重要的数据。删除数据库将永久性地删除所有相关数据和结构,无法回滚。

2093

2023.08.14

vb怎么连接数据库
vb怎么连接数据库

在VB中,连接数据库通常使用ADO(ActiveX 数据对象)或 DAO(Data Access Objects)这两个技术来实现:1、引入ADO库;2、创建ADO连接对象;3、配置连接字符串;4、打开连接;5、执行SQL语句;6、处理查询结果;7、关闭连接即可。

355

2023.08.31

MySQL恢复数据库
MySQL恢复数据库

MySQL恢复数据库的方法有使用物理备份恢复、使用逻辑备份恢复、使用二进制日志恢复和使用数据库复制进行恢复等。本专题为大家提供MySQL数据库相关的文章、下载、课程内容,供大家免费下载体验。

259

2023.09.05

vb中怎么连接access数据库
vb中怎么连接access数据库

vb中连接access数据库的步骤包括引用必要的命名空间、创建连接字符串、创建连接对象、打开连接、执行SQL语句和关闭连接。本专题为大家提供连接access数据库相关的文章、下载、课程内容,供大家免费下载体验。

329

2023.10.09

数据库对象名无效怎么解决
数据库对象名无效怎么解决

数据库对象名无效解决办法:1、检查使用的对象名是否正确,确保没有拼写错误;2、检查数据库中是否已存在具有相同名称的对象,如果是,请更改对象名为一个不同的名称,然后重新创建;3、确保在连接数据库时使用了正确的用户名、密码和数据库名称;4、尝试重启数据库服务,然后再次尝试创建或使用对象;5、尝试更新驱动程序,然后再次尝试创建或使用对象。

418

2023.10.16

pixiv网页版官网登录与阅读指南_pixiv官网直达入口与在线访问方法
pixiv网页版官网登录与阅读指南_pixiv官网直达入口与在线访问方法

本专题系统整理pixiv网页版官网入口及登录访问方式,涵盖官网登录页面直达路径、在线阅读入口及快速进入方法说明,帮助用户高效找到pixiv官方网站,实现便捷、安全的网页端浏览与账号登录体验。

660

2026.02.13

热门下载

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

精品课程

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

共4课时 | 22.4万人学习

Django 教程
Django 教程

共28课时 | 4.4万人学习

SciPy 教程
SciPy 教程

共10课时 | 1.6万人学习

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

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