如何在Python中使用contextlib模块简化上下文管理

千枫同学_1001

千枫同学_1001

2026-11-07

900人浏览

原创

@contextmanager可简化上下文管理器实现,但需用try/finally保证清理;closing()包装含close()的对象;suppress()忽略指定异常;ExitStack用于动态嵌套上下文。

如何在python中使用contextlib模块简化上下文管理

用 @contextmanager 替代手动写 __enter__ 和 __exit__

手动实现上下文管理器要写两个方法,容易漏掉异常处理逻辑或资源清理顺序。用 @contextmanager 装饰器,只需一个生成器函数,yield 之前的部分相当于 __enter__,之后相当于 __exit__。

常见错误是忘了加 try/finally —— @contextmanager 不自动保证清理,必须显式包裹。

  • yield 前做资源获取(如打开文件、连接数据库)
  • yield 后必须用 finally 或直接放在 yield 同级缩进下做释放(否则异常时不会执行)
  • 如果想捕获并吞掉异常,可在 except 块里返回 True,和 __exit__ 行为一致
from contextlib import contextmanager
<p>@contextmanager
def open_file(name):
f = open(name, 'w')
try:
yield f
finally:
f.close()  # 即使上面抛异常,这里也执行
</p>

用 contextlib.closing() 包裹已有 close 方法的对象

有些对象有 close() 但没实现上下文协议(比如某些老库的 socket、response 对象),直接 with 会报 AttributeError: __exit__。

closing() 是最轻量的包装方式,它只调用一次 close(),不干涉其他行为。

  • 仅适用于有 close() 方法且无参数的对象
  • 不能替代需要复杂清理逻辑的上下文管理器(比如要先 flush 再 close)
  • 注意:它不会抑制异常,异常仍会向上冒泡
from contextlib import closing
import urllib.request
<p>with closing(urllib.request.urlopen('<a href="https://www.php.cn/link/563dc6cc0586f6fe22c71fac9b8783ae">https://www.php.cn/link/563dc6cc0586f6fe22c71fac9b8783ae</a>')) as resp:
data = resp.read()  # resp 自动关闭,无需 resp.close()
</p>

用 contextlib.suppress() 忽略特定异常

不是所有“忽略异常”都要写 try/except: pass。当目标明确是“如果某异常发生就当没发生”,suppress() 更语义清晰、代码更短。

典型场景:删除临时文件、关闭可能已断开的连接、清理可能不存在的资源。

Python Hardener
Python Hardener

通过完整的安全审查、代码整洁重构、健壮的错误处理、正确的日志记录以及完整的文档字符串覆盖,强化 Python 脚本。

下载
  • 传入多个异常类可同时抑制(如 suppress(FileNotFoundError, PermissionError))
  • 它只抑制进入 with 块后发生的匹配异常,不捕获 with 表达式本身(如 open() 失败)的异常
  • 被抑制的异常不会留下痕迹,无法记录或重抛 —— 如果需要日志,还是得手写 try/except
from contextlib import suppress
import os
<p>with suppress(FileNotFoundError):
os.remove('temp.log')  # 文件不存在也不报错
</p>

嵌套多个上下文时优先用 contextlib.ExitStack

当需要动态决定进几个上下文(比如根据配置开关日志、监控、事务),或上下文数量不确定时,硬写嵌套 with 既难读又难维护。

ExitStack 允许运行时注册任意多个上下文,并统一管理退出逻辑,还支持在中途取消已进入的上下文。

  • enter_context() 返回值就是上下文管理器的 __enter__ 返回值
  • push() 可注册普通回调(类似 __exit__ 的清理函数),不依赖上下文协议
  • 若某个 enter_context() 失败,已成功进入的会按逆序自动退出 —— 这点比手写 try/finally 链可靠得多
from contextlib import ExitStack
<p>with ExitStack() as stack:
file1 = stack.enter_context(open('a.txt', 'w'))
file2 = stack.enter_context(open('b.txt', 'w'))
if need_log:
log_file = stack.enter_context(open('log.txt', 'a'))</p><h1>所有文件在退出时自动关闭</h1><p></p>

真正容易被忽略的是异常传播路径:@contextmanager 里 yield 后的代码若出错,会覆盖原始异常;ExitStack 中某个 enter_context() 报错,其余已进的上下文虽能清理,但原始异常仍是那个最先失败的——这些细节不看源码或不测异常分支根本意识不到。

Python免费学习笔记(深入):立即使用
在学习笔记中,你将探索 Python 的核心概念和高级技巧!

相关文章

PHP速学视频免费教程(入门到精通)
PHP速学视频免费教程(入门到精通)

PHP怎么学习?PHP怎么入门?PHP在哪学?PHP怎么学才快?不用担心,这里为大家提供了PHP速学教程(入门到精通),有需要的小伙伴保存下载就能学习啦!

下载

相关标签:

python

本站声明:本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系admin@php.cn

相关专题

更多
python打包成可执行文件
python打包成可执行文件

本专题为大家带来python打包成可执行文件相关的文章,大家可以免费的下载体验。

2023.07.20

1891

4

python能做什么
python能做什么

python能做的有:可用于开发基于控制台的应用程序、多媒体部分开发、用于开发基于Web的应用程序、使用python处理数据、系统编程等等。本专题为大家提供python相关的各种文章、以及下载和课程。

2023.07.25

5164

7

format在python中的用法
format在python中的用法

Python中的format是一种字符串格式化方法,用于将变量或值插入到字符串中的占位符位置。通过format方法,我们可以动态地构建字符串,使其包含不同值。php中文网给大家带来了相关的教程以及文章,欢迎大家前来阅读学习。

2023.07.31

1829

3

python教程
python教程

Python已成为一门网红语言,即使是在非编程开发者当中,也掀起了一股学习的热潮。本专题为大家带来python教程的相关文章,大家可以免费体验学习。

2023.08.03

31177

23

python环境变量的配置
python环境变量的配置

Python是一种流行的编程语言,被广泛用于软件开发、数据分析和科学计算等领域。在安装Python之后,我们需要配置环境变量,以便在任何位置都能够访问Python的可执行文件。php中文网给大家带来了相关的教程以及文章,欢迎大家前来学习阅读。

2023.08.04

3687

5

python eval
python eval

eval函数是Python中一个非常强大的函数,它可以将字符串作为Python代码进行执行,实现动态编程的效果。然而,由于其潜在的安全风险和性能问题,需要谨慎使用。php中文网给大家带来了相关的教程以及文章,欢迎大家前来学习阅读。

2023.08.04

3707

5

scratch和python区别
scratch和python区别

scratch和python的区别:1、scratch是一种专为初学者设计的图形化编程语言,python是一种文本编程语言;2、scratch使用的是基于积木的编程语法,python采用更加传统的文本编程语法等等。本专题为大家提供scratch和python相关的文章、下载、课程内容,供大家免费下载体验。

2023.08.11

1243

5

python合并两个列表
python合并两个列表

Python是一种强大的编程语言,具有许多方便的功能和工具。在Python中,有多种方法可以合并两个列表。php中文网给大家带来了相关的教程以及文章,欢迎大家前来学习阅读。

2023.08.10

616

4

python是前端还是后端
python是前端还是后端

Python属于前端也属于后端,其灵活性和丰富的生态系统使得开发人员能够在不同的领域中灵活运用。本专题为大家提供python相关的文章、下载、课程内容,供大家免费下载体验。

2023.08.11

2863

5

热门下载

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

精品课程

更多
相关推荐
/
热门推荐
/
最新课程