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
语句块执行完毕后(无论是正常结束还是因为异常而中断)调用。这个方法负责资源的清理工作。它的三个参数非常关键:

Pi智能演示文档
Pi智能演示文档

领先的AI PPT生成工具

下载
  • 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
块则确保了清理代码无论如何都会执行。这种方式对于快速实现功能性上下文管理器,简直是神来之笔。

相关专题

更多
python开发工具
python开发工具

php中文网为大家提供各种python开发工具,好的开发工具,可帮助开发者攻克编程学习中的基础障碍,理解每一行源代码在程序执行时在计算机中的过程。php中文网还为大家带来python相关课程以及相关文章等内容,供大家免费下载使用。

716

2023.06.15

python打包成可执行文件
python打包成可执行文件

本专题为大家带来python打包成可执行文件相关的文章,大家可以免费的下载体验。

626

2023.07.20

python能做什么
python能做什么

python能做的有:可用于开发基于控制台的应用程序、多媒体部分开发、用于开发基于Web的应用程序、使用python处理数据、系统编程等等。本专题为大家提供python相关的各种文章、以及下载和课程。

739

2023.07.25

format在python中的用法
format在python中的用法

Python中的format是一种字符串格式化方法,用于将变量或值插入到字符串中的占位符位置。通过format方法,我们可以动态地构建字符串,使其包含不同值。php中文网给大家带来了相关的教程以及文章,欢迎大家前来阅读学习。

617

2023.07.31

python教程
python教程

Python已成为一门网红语言,即使是在非编程开发者当中,也掀起了一股学习的热潮。本专题为大家带来python教程的相关文章,大家可以免费体验学习。

1236

2023.08.03

python环境变量的配置
python环境变量的配置

Python是一种流行的编程语言,被广泛用于软件开发、数据分析和科学计算等领域。在安装Python之后,我们需要配置环境变量,以便在任何位置都能够访问Python的可执行文件。php中文网给大家带来了相关的教程以及文章,欢迎大家前来学习阅读。

547

2023.08.04

python eval
python eval

eval函数是Python中一个非常强大的函数,它可以将字符串作为Python代码进行执行,实现动态编程的效果。然而,由于其潜在的安全风险和性能问题,需要谨慎使用。php中文网给大家带来了相关的教程以及文章,欢迎大家前来学习阅读。

575

2023.08.04

scratch和python区别
scratch和python区别

scratch和python的区别:1、scratch是一种专为初学者设计的图形化编程语言,python是一种文本编程语言;2、scratch使用的是基于积木的编程语法,python采用更加传统的文本编程语法等等。本专题为大家提供scratch和python相关的文章、下载、课程内容,供大家免费下载体验。

699

2023.08.11

php源码安装教程大全
php源码安装教程大全

本专题整合了php源码安装教程,阅读专题下面的文章了解更多详细内容。

7

2025.12.31

热门下载

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

精品课程

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

共4课时 | 0.6万人学习

Django 教程
Django 教程

共28课时 | 2.6万人学习

SciPy 教程
SciPy 教程

共10课时 | 1.0万人学习

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

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