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模块,你可以更方便地创建和使用上下文管理器,以简化资源管理和错误处理代码。

相关推荐
兵慌码乱8 小时前
面向桌面端的资产管理系统分层架构设计与核心模块实现
python·系统架构·sqlite·pyqt5·数据库设计·桌面应用开发·mvc架构
hboot9 小时前
AI工程师第三课 - 机器学习基础
python·scikit-learn·kaggle
顾林海14 小时前
Agent入门阶段-编程基础-Python:流程控制
python·agent·ai编程
呱呱复呱呱17 小时前
Django CBV 源码解读:一个请求是怎么找到你的 get() 方法的
python·django
曲幽21 小时前
刚部署的 LibreTranslate 频频翻车?我掏出了 20 年前的 StarDict 词典,用 FastAPI 搭了个本地词典翻译 API
python·fastapi·web·translate·goldendict·libretranslate·stardict·pystardict
荣码1 天前
用Streamlit给AI应用套个界面,10行代码出Web页面
java·python
兵慌码乱1 天前
基于Python+PyQt5+SQLite的药房管理系统实现:事务一致性与界面解耦全流程解析
python·sqlite·信号与槽·pyqt5·数据库设计·桌面应用开发·事务处理
金銀銅鐵1 天前
[Python] 体验用欧几里得算法计算最大公约数的过程
python·数学
FreakStudio2 天前
W55MH32L-EVB 上手测评:硬件 TCP/IP 加持的以太网单片机,MicroPython 零门槛开发
python·单片机·嵌入式·大学生·面向对象·并行计算·电子diy·电子计算机
用户0332126663672 天前
使用 Python 从零创建 Word 文档
python