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

相关推荐
零基础1231 分钟前
Agent 的 Memory 怎么做科研:以中医诊断场景为例
人工智能·经验分享·python·语言模型
databook43 分钟前
面向数据工程师的正则表达式:从日志清洗到字段提取
python·正则表达式·数据分析
ss2731 小时前
AI全栈实战 | 3.2-01 Python 基础:四大数据容器怎么选,推导式为什么是 Pythonic 的灵魂
开发语言·人工智能·python
计算机毕业编程指导师1 小时前
【大数据毕设选题推荐】基于Spark的WTA职业网球赛事演变与竞技格局分析系统源码 毕业设计 选题推荐 毕设选题 数据分析 机器学习 深度学习
大数据·python·计算机·spark·毕业设计·课程设计·wta网球
言乐62 小时前
Python区分广度优先深度优先宽度优先的区别
python·算法·深度优先·广度优先·宽度优先
打工仔折腾 AI2 小时前
LLaMA 1 到 LLaMA 3 架构演进拆解:从 RoPE、GQA 到词表扩张
人工智能·后端·python·深度学习·langchain·llama
旋生万物2 小时前
螺旋角分布的功率谱分析:多窗谱 + FDR 校正
开发语言·python
专业程序开发源3 小时前
SSM校园拍摄交流服务平台36936-计算机课程设计、毕业设计
java·spring boot·后端·python·elasticsearch·php·课程设计
wuyk5553 小时前
Python 网络爬虫入门到实战 第 04 章:请求头、UA 伪装、超时、异常处理、基础反爬绕过
开发语言·爬虫·python
kaiyou20263 小时前
用户研究岗秋招,统计分析能力怎么学习和证明?
数据库·python·学习