0

0

with 语句和上下文管理器(Context Manager)的原理

幻影之瞳

幻影之瞳

发布时间:2025-09-03 14:43:01

|

330人浏览过

|

来源于php中文网

原创

with语句通过上下文管理器协议确保资源在进入和退出代码块时被正确初始化和清理,即使发生异常也能自动释放资源,从而避免资源泄漏;它通过__enter__和__exit__方法或contextlib的@contextmanager装饰器实现,使文件、数据库连接等资源管理更安全、简洁。

with 语句和上下文管理器(context manager)的原理

with
语句在 Python 里,说白了,就是一种优雅地处理资源(比如文件、网络连接、锁)的方式,它能确保这些资源在使用完毕后,不管过程中有没有出错,都能被妥善地清理掉。它的魔法源自“上下文管理器”协议,这个协议定义了对象如何进入和退出一个特定的代码块。

在编程实践中,我们经常需要处理一些有“生命周期”的资源:打开文件后要关闭,连接数据库后要断开,获取锁后要释放。如果手动去管理这些步骤,尤其是在代码路径复杂或者出现异常的情况下,很容易遗漏清理操作,导致资源泄漏或者程序行为异常。

with
语句就是为了解决这个痛点而生的。它通过与实现了特定方法的对象(也就是上下文管理器)协作,在进入
with
代码块时执行资源准备工作,在退出代码块时(无论正常退出还是异常退出)执行资源清理工作。这不仅让代码更简洁,也大大提高了程序的健壮性。

为什么我们要用
with
语句来处理文件或数据库连接?

我记得刚开始写Python的时候,处理文件都是这样:

f = open('my_file.txt', 'r'); data = f.read(); f.close()
。后来发现,如果
f.read()
出了问题,比如文件太大内存溢出,那
f.close()
就根本执行不到,文件句柄就一直开着,时间长了,系统资源就耗尽了。或者说,如果一个函数里有多个
return
语句,你还得在每个
return
之前都加上
f.close()
,想想都头大。

这就是

with
语句的价值所在。它最核心的优势在于“异常安全”和“自动化清理”。

拿文件操作来说,当你写

with open('my_file.txt', 'r') as f:
,Python 会在文件打开后,把文件对象赋值给
f
。当
with
块内的代码执行完毕,或者在执行过程中抛出了任何异常,Python 都会自动调用文件对象的清理方法(也就是关闭文件)。你完全不用操心
f.close()
,也不用写
try...finally
这样的结构。这不仅仅是代码量减少那么简单,更重要的是,它极大地降低了因疏忽而导致资源泄漏的风险。

数据库连接也是一个典型的例子。连接池中的连接是宝贵的资源,一个应用如果不断地获取连接而不释放,很快就会把连接池耗尽,导致其他请求无法获取连接。使用

with
语句来管理数据库连接,可以确保连接在事务完成后(或发生错误时)被自动归还给连接池,或者直接关闭,避免了资源浪费。这种模式让开发者可以更专注于业务逻辑,而不是繁琐的资源管理细节。

自己动手实现一个上下文管理器需要注意什么?

实现一个自定义的上下文管理器,其实就是让你的类遵循上下文管理器协议,也就是实现两个特殊方法:

__enter__
__exit__

__enter__(self)
方法会在
with
语句块被执行前调用。它的主要职责是进行资源的初始化和准备工作,并且它必须返回一个对象。这个对象就是
as
关键字后面跟着的变量所绑定的值。比如,
with MyResource() as res:
res
就会是
MyResource
实例的
__enter__
方法的返回值。如果你的资源对象本身就是你想要操作的对象,那么通常
__enter__
直接返回
self
就可以了。

__exit__(self, exc_type, exc_val, exc_tb)
方法则是在
with
语句块执行完毕后(无论是正常结束还是因为异常而中断)调用。这个方法负责资源的清理工作。它的三个参数非常关键:

Draft&Goal-Detector
Draft&Goal-Detector

检测文本是由 AI 还是人类编写的

下载
  • exc_type
    : 如果
    with
    块内发生了异常,这里就是异常的类型(比如
    ValueError
    )。如果没有异常,则是
    None
  • exc_val
    : 如果有异常,这里是异常实例本身。没有异常时是
    None
  • exc_tb
    : 如果有异常,这里是异常的追踪信息(traceback)。没有异常时是
    None

__exit__
方法里,你可以根据
exc_type
是否为
None
来判断
with
块内是否发生了异常。如果你希望在
__exit__
方法中处理掉这个异常(比如记录日志后不再向上层抛出),那么
__exit__
方法需要返回
True
。返回
True
会告诉 Python 解释器,这个异常已经被妥善处理了,不要再继续向上抛出。如果返回
False
(或者不返回任何值,因为默认就是
None
,相当于
False
),那么异常会继续传播。

举个例子,假设我们要实现一个简单的计时器上下文管理器:

import time

class MyTimer:
    def __enter__(self):
        self.start_time = time.time()
        print("计时开始...")
        return self # 返回自身实例,这样 with MyTimer() as timer: 就可以用 timer 了

    def __exit__(self, exc_type, exc_val, exc_tb):
        end_time = time.time()
        duration = end_time - self.start_time
        print(f"计时结束,总耗时:{duration:.4f} 秒")

        if exc_type: # 如果有异常发生
            print(f"with 块内发生异常:{exc_type.__name__}: {exc_val}")
            # 如果我们不想让异常继续传播,可以在这里返回 True
            # return True
        # 默认返回 None,即 False,异常会继续传播
        print("资源清理完成。")

# 使用示例
with MyTimer():
    print("正在执行一些耗时操作...")
    time.sleep(1.5)
    # raise ValueError("哦豁,出错了!") # 尝试解注释这行看看异常处理

print("with 块外部代码继续执行。")

这里

__enter__
返回了
self
,所以
with MyTimer() as timer:
这里的
timer
就是
MyTimer
的实例。
__exit__
负责计算并打印耗时,同时展示了如何捕获异常信息。通过这样的设计,你可以封装任何需要“进入”和“退出”阶段的逻辑。

contextlib
模块如何简化上下文管理器的创建?

说实话,每次都写一个类,然后实现

__enter__
__exit__
,对于一些简单的上下文管理任务来说,确实有点小麻烦。Python 的标准库
contextlib
就是来解决这个问题的,它提供了一些工具,让创建上下文管理器变得更简单,特别是
@contextmanager
装饰器。

@contextmanager
装饰器允许你用一个生成器函数来创建上下文管理器。它的原理是,在生成器函数中,
yield
关键字之前的所有代码,相当于
__enter__
方法的逻辑;
yield
语句本身返回的值,就是
__enter__
方法的返回值;而
yield
语句之后的所有代码,则相当于
__exit__
方法的逻辑。

我们来用

@contextmanager
重新实现上面的计时器:

from contextlib import contextmanager
import time

@contextmanager
def simple_timer():
    start_time = time.time()
    print("计时开始 (通过 contextlib)...")
    try:
        yield # 这里是 with 块的代码执行的地方
    except Exception as e:
        print(f"with 块内发生异常 (通过 contextlib):{type(e).__name__}: {e}")
        # 如果要抑制异常,这里可以不 re-raise,或者捕获后不抛出
        # 也可以再次抛出,或者抛出新的异常
        raise # 默认会重新抛出捕获到的异常
    finally:
        end_time = time.time()
        duration = end_time - start_time
        print(f"计时结束 (通过 contextlib),总耗时:{duration:.4f} 秒")
        print("资源清理完成 (通过 contextlib)。")

# 使用示例
with simple_timer():
    print("正在执行一些耗时操作 (通过 contextlib)...")
    time.sleep(0.8)
    # raise TypeError("又出错了!") # 尝试解注释这行

print("with 块外部代码继续执行 (通过 contextlib)。")

你看,是不是简洁多了?一个函数搞定一切。

yield
语句将生成器函数分割成了两个部分,完美地对应了
__enter__
__exit__
的行为。
try...except...finally
结构在
yield
周围,使得异常处理和清理逻辑变得非常直观。如果
yield
内部的代码抛出异常,这个异常会被
try...except
捕获,你可以在
except
块中处理它。如果
except
块没有重新抛出异常,那么异常就被抑制了。
finally
块则确保了清理代码无论如何都会执行。这种方式对于快速实现功能性上下文管理器,简直是神来之笔。

热门AI工具

更多
DeepSeek
DeepSeek

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

豆包大模型
豆包大模型

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

通义千问
通义千问

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

腾讯元宝
腾讯元宝

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

文心一言
文心一言

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

讯飞写作
讯飞写作

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

即梦AI
即梦AI

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

ChatGPT
ChatGPT

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

相关专题

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

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

358

2023.06.29

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

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

2082

2023.08.14

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

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

349

2023.08.31

MySQL恢复数据库
MySQL恢复数据库

MySQL恢复数据库的方法有使用物理备份恢复、使用逻辑备份恢复、使用二进制日志恢复和使用数据库复制进行恢复等。本专题为大家提供MySQL数据库相关的文章、下载、课程内容,供大家免费下载体验。

256

2023.09.05

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

vb中连接access数据库的步骤包括引用必要的命名空间、创建连接字符串、创建连接对象、打开连接、执行SQL语句和关闭连接。本专题为大家提供连接access数据库相关的文章、下载、课程内容,供大家免费下载体验。

326

2023.10.09

数据库对象名无效怎么解决
数据库对象名无效怎么解决

数据库对象名无效解决办法:1、检查使用的对象名是否正确,确保没有拼写错误;2、检查数据库中是否已存在具有相同名称的对象,如果是,请更改对象名为一个不同的名称,然后重新创建;3、确保在连接数据库时使用了正确的用户名、密码和数据库名称;4、尝试重启数据库服务,然后再次尝试创建或使用对象;5、尝试更新驱动程序,然后再次尝试创建或使用对象。

412

2023.10.16

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

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

411

2023.10.16

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

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

223

2023.10.19

clawdbot ai使用教程 保姆级clawdbot部署安装手册
clawdbot ai使用教程 保姆级clawdbot部署安装手册

Clawdbot是一个“有灵魂”的AI助手,可以帮用户清空收件箱、发送电子邮件、管理日历、办理航班值机等等,并且可以接入用户常用的任何聊天APP,所有的操作均可通过WhatsApp、Telegram等平台完成,用户只需通过对话,就能操控设备自动执行各类任务。

1

2026.01.29

热门下载

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

精品课程

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

共4课时 | 22.4万人学习

Django 教程
Django 教程

共28课时 | 3.6万人学习

SciPy 教程
SciPy 教程

共10课时 | 1.3万人学习

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

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