Python中的上下文管理器(contextlib)模块

Python中的contextlib模块提供了一些用于创建和管理上下文管理器(context managers)的工具。上下文管理器是实现了__enter__()和__exit__()方法的对象,它们通常用于确保在代码块执行前后执行某些操作,比如资源获取与释放、设置和重置环境、错误处理等。with语句与上下文管理器一起使用,可以自动处理这些操作,使代码更加简洁和健壮。

contextlib模块提供了几个有用的函数和类,用于创建和管理上下文管理器,主要包括:

  1. contextlib.contextmanager :这是一个装饰器,用于将生成器函数转化为上下文管理器。在生成器函数中,使用yield语句分隔__enter__()和__exit__()方法的逻辑。

示例:

复制代码

python复制代码

|---|---------------------------------------------------|
| | from contextlib import contextmanager |
| | |
| | @contextmanager |
| | def managed_resource(*args, **kwds): |
| | # __enter__ 逻辑 |
| | resource = acquire_resource(*args, **kwds) |
| | try: |
| | yield resource |
| | finally: |
| | # __exit__ 逻辑 |
| | release_resource(resource) |
| | |
| | # 使用 with 语句 |
| | with managed_resource(arg1, arg2) as resource: |
| | # 使用 resource |
| | pass |

  1. contextlib.ExitStack :这是一个上下文管理器,可以处理多个上下文管理器的__enter__()和__exit__()调用。这使得可以方便地管理一组需要同时设置和清理的资源。

示例:

复制代码

python复制代码

|---|---------------------------------------------------------------------|
| | from contextlib import ExitStack |
| | |
| | with ExitStack() as stack: |
| | files = [stack.enter_context(open(fname)) for fname in filelist] |
| | # 使用 files 列表中的文件 |

  1. contextlib.nullcontext :这是一个不做任何操作的上下文管理器,通常用于在不需要实际上下文管理的情况下与with语句一起使用。

示例:

复制代码

python复制代码

|---|---------------------------------------|
| | from contextlib import nullcontext |
| | |
| | with nullcontext(obj) as result: |
| | # 这里 result 就是 obj,但没有任何额外的操作 |
| | pass |

  1. contextlib.suppress :这是一个上下文管理器,用于抑制指定的异常。在with语句块中,指定的异常会被捕获并忽略,不会传播到外部。

示例:

复制代码

python复制代码

|---|-------------------------------------------------|
| | from contextlib import suppress |
| | |
| | with suppress(FileNotFoundError): |
| | os.remove('somefile.tmp') |
| | # 如果 somefile.tmp 不存在,则不会抛出 FileNotFoundError |

通过使用contextlib模块,你可以更方便地创建和使用上下文管理器,以简化资源管理和错误处理代码。

相关推荐
2601_966949651 小时前
如何根据研究的起止时间截取分钟级行情?Python 实战与 QuantDash 数据处理指南
开发语言·python·pandas·量化交易·1024程序员节·股票数据·quantdash
在世修行5 小时前
智学文枢:页面切换与跨页数据预填
python·桌面应用·signal·1024程序员节·pyside·跨页跳转·状态预填
2601_962885726 小时前
如何用 Python 把 K 线重采样成任意周期?
开发语言·python
打工仔折腾 AI6 小时前
在 Ubuntu ARM64 部署 Q1 V3.0 MuJoCo 模型:加载、单步推理与执行器响应解析
linux·运维·人工智能·python·深度学习·ubuntu·性能优化
信誓旦旦的程序猿7 小时前
【Python量化实战宝典 #21】北向资金跟随手册:聪明钱择时信号(Python 可跑)
python·1024程序员节·股票数据api·股票数据api接口·股票api数据接口·股票量化数据api·股票量化数据接口
严谨的麻辣烫7 小时前
静态 ISP 代理请求超时怎么排查?Python 与 curl 实战
python·http·requests·1024程序员节·网络排障
KeepRobots8 小时前
11-01 温哥华不回拨:用 60 行 Python 查你的工单系统会不会差一小时(KeepRobots 工程笔记)
python
xiao77mwq8 小时前
你以为 OCR 慢?实测 81.3 s 里全是故障代价:如何把「定格态」消掉,让耗时从 6.5 s 到 13 倍差距显出原形
python
深圳市益普科技有限公司9 小时前
半导体良率管理YMS:良率与设备参数的相关性,先拉哪几组数据
人工智能·python·算法
“AI国潮设计-小江”9 小时前
【AIGC实战】Python+SDXL打造“潮汕英歌舞IP”万能立牌插件,批量生成“发财元宝蛋糕”视觉营销方案(附源码)
开发语言·人工智能·python·prompt·aigc