0

0

Python上下文管理实现_自定义with说明【教程】

冷漠man

冷漠man

发布时间:2026-01-02 20:07:13

|

928人浏览过

|

来源于php中文网

原创

Python中with语句依赖__enter__和__exit__方法实现上下文管理,支持类定义或contextlib.contextmanager装饰器两种方式,用于自动管理资源生命周期,避免手动清理。

python上下文管理实现_自定义with说明【教程】

Python 中的 with 语句依赖上下文管理器协议,核心是实现 __enter____exit__ 两个特殊方法。自定义上下文管理器能让资源(如文件、数据库连接、锁等)在进入和退出时自动初始化与清理,避免手动 try/finally 的冗余代码。

用类实现上下文管理器

最直接的方式是定义一个类,显式实现两个方法:

  • __enter__:在 with 语句块开始时调用,通常返回需要被管理的对象(如打开的文件),也可返回 self 或其他值供 as 绑定
  • __exit__(exc_type, exc_value, traceback):在 with 块结束时(无论是否异常)调用;三个参数分别表示异常类型、异常值、回溯信息;若该方法返回 True,则会抑制异常传播

示例:一个简单的计时上下文管理器

import time
<p>class Timer:
def <strong>enter</strong>(self):
self.start = time.time()
return self</p><pre class="brush:php;toolbar:false;">def __exit__(self, exc_type, exc_value, traceback):
    self.end = time.time()
    print(f"耗时: {self.end - self.start:.3f} 秒")

使用

with Timer() as t: time.sleep(1)

输出:耗时: 1.001 秒

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

用 contextlib.contextmanager 装饰器(推荐)

更简洁的方式是使用 contextlib.contextmanager 装饰生成器函数。函数中 yield 之前的代码相当于 __enter__,之后的代码(在 finally 块中执行)相当于 __exit__

  • yield 的值即为 as 接收的对象
  • 所有清理逻辑写在 yield 后,天然支持异常安全(类似 try/finally
  • 无需手动处理异常参数,也无需返回布尔值来抑制异常(如需抑制,仍需加 try/except

示例:临时修改环境变量

DodeCMS成创企业网站管理系统英文版1.1
DodeCMS成创企业网站管理系统英文版1.1

DodeCmsV1.1英文版企业版说明 免费版 (ASP+ACCESS) DodeCMS成创网站内容管理系统由辽宁成创网络科技有限公司基于微软ASP、通用ACCESS/MSSQL数据库开发完成; 访问模式采用动态模式,基本上实现了系统的自定义功能,代码简洁高效,易于修改维护,可扩展性强。 ======================================================

下载
from contextlib import contextmanager
import os
<p>@contextmanager
def temp_env(**kwargs):
old_values = {}
for key, value in kwargs.items():
old_values[key] = os.environ.get(key)
os.environ[key] = str(value)
try:
yield
finally:
for key, old_value in old_values.items():
if old_value is None:
os.environ.pop(key, None)
else:
os.environ[key] = old_value</p><h1>使用</h1><p>with temp_env(DEBUG="1", ENV="test"):
print(os.environ["DEBUG"])  # 1
print(os.environ["ENV"])    # test</p><h1>退出后自动恢复原环境变量</h1><p></p><p><span>立即学习</span>“<a href="https://pan.quark.cn/s/00968c3c2c15" style="text-decoration: underline !important; color: blue; font-weight: bolder;" rel="nofollow" target="_blank">Python免费学习笔记(深入)</a>”;</p>

常见注意事项与技巧

写自定义上下文管理器时容易忽略的关键点:

  • __exit__ 方法必须接受三个参数(即使不用也要保留),否则会报错
  • 若上下文管理器本身可能抛出异常,应在 __exit__ 中谨慎处理,避免掩盖原始错误
  • 多个上下文管理器可写在同一行:with open("a.txt") as f1, open("b.txt") as f2:
  • 想复用已有对象的上下文行为?确认它实现了协议(比如 threading.Lockpathlib.Path.open() 返回的文件对象都支持 with

何时该自己写,而不是用现成的?

标准库和第三方包已覆盖大部分场景(文件、数据库连接池、锁、临时目录等)。只有当业务有明确的“进入-执行-退出”生命周期,且涉及资源独占、状态切换或副作用清理时,才值得封装:

  • 临时切换日志级别
  • 测试中模拟网络超时或异常响应
  • GPU 显存分配与释放(PyTorch/TensorFlow 中常见)
  • 数据库事务的 begin/commit/rollback 封装

不复杂但容易忽略细节,掌握协议本质比死记语法更重要。

热门AI工具

更多
DeepSeek
DeepSeek

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

豆包大模型
豆包大模型

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

通义千问
通义千问

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

腾讯元宝
腾讯元宝

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

文心一言
文心一言

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

讯飞写作
讯飞写作

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

即梦AI
即梦AI

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

ChatGPT
ChatGPT

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

相关专题

更多
数据库三范式
数据库三范式

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

384

2023.06.29

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

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

2110

2023.08.14

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

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

357

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、尝试更新驱动程序,然后再次尝试创建或使用对象。

420

2023.10.16

vb连接access数据库的方法
vb连接access数据库的方法

vb连接access数据库方法:1、使用ADO连接,首先导入System.Data.OleDb模块,然后定义一个连接字符串,接着创建一个OleDbConnection对象并使用Open() 方法打开连接;2、使用DAO连接,首先导入 Microsoft.Jet.OLEDB模块,然后定义一个连接字符串,接着创建一个JetConnection对象并使用Open()方法打开连接即可。

475

2023.10.16

vb连接数据库的方法
vb连接数据库的方法

vb连接数据库的方法有使用ADO对象库、使用OLEDB数据提供程序、使用ODBC数据源等。详细介绍:1、使用ADO对象库方法,ADO是一种用于访问数据库的COM组件,可以通过ADO连接数据库并执行SQL语句。可以使用ADODB.Connection对象来建立与数据库的连接,然后使用ADODB.Recordset对象来执行查询和操作数据;2、使用OLEDB数据提供程序方法等等。

231

2023.10.19

Go高并发任务调度与Goroutine池化实践
Go高并发任务调度与Goroutine池化实践

本专题围绕 Go 语言在高并发任务处理场景中的实践展开,系统讲解 Goroutine 调度模型、Channel 通信机制以及并发控制策略。内容包括任务队列设计、Goroutine 池化管理、资源限制控制以及并发任务的性能优化方法。通过实际案例演示,帮助开发者构建稳定高效的 Go 并发任务处理系统,提高系统在高负载环境下的处理能力与稳定性。

4

2026.03.10

热门下载

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

精品课程

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

共4课时 | 22.5万人学习

Django 教程
Django 教程

共28课时 | 4.9万人学习

SciPy 教程
SciPy 教程

共10课时 | 1.9万人学习

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

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