详解Python中的上下文管理器原理

 更新时间:2023年03月11日 14:04:26   作者:一根头发学一年  
这篇文章主要为大家详细介绍了Python中的上下文管理器的原理与使用,文中的示例代码讲解详细,感兴趣的小伙伴可以跟随小编一起学习一下

with语句

在我们日常使用场景中,经常会操作一些资源,比如文件对象、数据库连接、Socket连接等,资源操作完了之后,不管操作的成功与否,最重要的事情就是关闭该资源,否则资源打开太多而没有关闭,程序会报错,以文件操作为例,通常我们会这样写:

f = open('file.txt', 'w')
try:
    f.write("Hello")
finally:
    f.close()

但既然close方法是必须的操作,那就没必要显式地调用,所以Python给我们提供了一种更优雅的方式,使用with语句:

with open('file.txt', 'w') as f:
    f.write("Hello")

在退出with语句下的代码块之后,f 对象会自动执行自己的close方法,实现资源的释放,简洁优雅。

上下文管理器原理

上下文管理器实际是内部实现了__enter__和__exit__方法的对象。

当我们使用with语法时:

__enter__()方法:返回一个值,可以将它赋值给as后面的对象,例如上面的中的f;

__exit__()方法:with语句退出或者发送异常时会执行这个方法。

1、__enter__方法说明

上下文管理器的__enter__方法是可以带返回值的,默认返回None,这个返回值通过with…as…中的 as 赋给它后面的那个变量,所以 with EXPR as VAR 就是将EXPR对象__enter__方法的返回值赋给 VAR。

当然with...as...并非固定组合,单独使用with...也是可以的,上下文管理器的__enter__方法还是正常执行,只是这个返回值并没有赋给一个变量,with下面的代码块也不能使用这个返回值。

2、__exit__方法说明

上下文管理器的__exit__方法接收3个参数exc_type、exc_val、exc_tb,如果代码块BLOCK发生了异常e并退出,这3个参数分别为type(e)、str(e)、e.__traceback__,否则都为None。

同样__exit__方法也是可以带返回值的,这个返回值应该是一个布尔类型True或False,默认为None(即False)。如果为False,异常会被抛出,用户需要进行异常处理。如果为True,则表示忽略该异常。

一个上下文管理器一般使用如下:

with EXPR as VAR:
    BLOCK

上述代码的执行过程等价于:

ContextManager = EXPR
VAR = ContextManager.__enter__()
try:
    BLOCK
finally:
    ContextManager.__exit__()

f 对象就是把自己的close方法定义在了它的__exit__方法内部,实现了代码块BLOCK执行完之后自动关闭自身。

自定义上下文管理器

下面我们定义一个文件类,内部实现了__enter__和__exit__两个方法:

class File:
    def __init__(self, filename, mode):
        self.filename = filename
        self.mode = mode

    def __enter__(self):
        print("进入")
        self.f = open(self.filename, self.mode)
        return self.f

    def __exit__(self, exc_type=None, exc_val=None, exc_tbs=None):
        print("退出")
        self.f.close()

这时候File类就是一个上下文管理器

我们分别通过 with语句 和 try/finally语句 使用File类对文件进行写入操作

通过with语句执行:

with File('file.txt', 'w') as f:
    print("正在写入...")
    f.write('Hello')

控制台输出:

进入
正在写入...
退出

并得到了一个写了 Hello 的 file.txt 文件

通过try/finally语句执行:

ContextManager = File('file.txt', 'w')
VAR = ContextManager .__enter__()
try:
    print("正在写入...")
    VAR.write('Hello')
finally:
    ContextManager.__exit__()

控制台输出:

进入
正在写入...
退出

并得到了一个写了 Hello 的 file.txt 文件

两者输出一致,所以验证了二中执行过程的等价关系是正确的。

contextmanager 装饰器

Python还提供了一个contextmanager装饰器,允许用户将一个生成器定义为上下文管理器,该装饰器将生成器中的代码通过yield语句分成两部分,yield之前的代码为__enter__方法,yield之后的代码为__exit__方法,yield的返回值即__enter__方法的返回值,用于赋给as后的变量。

下面我们通过contextmanager装饰器也实现一个关于文件的上下文管理器:

from contextlib import contextmanager

@contextmanager
def open_file(filename, mode):
    print('进入')
    f = open(filename, mode)
    try:
        yield f
    finally:
        print('退出')
        f.close()

说明:这里使用 try/finally 是确保yield的过程中就算出现异常,文件也能正常关闭,当然这里也能处理异常,使用 try/except/finally 即可。

通过with语句执行:

with open_file('file.txt', 'w') as f:
    print("正在写入...")
    f.write('Hello')

执行结果跟之前的上下文管理器执行结果一致,说明contextmanager装饰器也能定义一个上下文管理器。

到此这篇关于详解Python中的上下文管理器原理的文章就介绍到这了,更多相关Python上下文管理器内容请搜索脚本之家以前的文章或继续浏览下面的相关文章希望大家以后多多支持脚本之家!

相关文章

  • 解决pycharm安装第三方库失败的问题

    解决pycharm安装第三方库失败的问题

    这篇文章主要介绍了pycharm安装第三方库失败的解决方法,本文通过图文实例相结合给大家介绍的非常详细,对大家的学习或工作具有一定的参考借鉴价值,需要的朋友可以参考下
    2020-05-05
  • python实现列表中最大最小值输出的示例

    python实现列表中最大最小值输出的示例

    今天小编就为大家分享一篇python实现列表中最大最小值输出的示例,具有很好的参考价值,希望对大家有所帮助。一起跟随小编过来看看吧
    2019-07-07
  • Python实现获取乱序列表排序后的新下标的示例

    Python实现获取乱序列表排序后的新下标的示例

    本文主要介绍了Python实现获取乱序列表排序后的新下标的示例,文中通过示例代码介绍的非常详细,对大家的学习或者工作具有一定的参考学习价值,需要的朋友们下面随着小编来一起学习学习吧
    2023-04-04
  • Python练习之操作SQLite数据库

    Python练习之操作SQLite数据库

    这篇文章主要介绍了Python练习之操作SQLite数据库,主要通过三个问题如何创建SQLite数据库?如何向SQLite表中插入数据?如何查询SQLite表中的数据?展开文章主题详情,需要的朋友可以参考一下
    2022-06-06
  • python中的% 是什么意思,起到什么作用呢

    python中的% 是什么意思,起到什么作用呢

    %Y会被无世纪的年份所替代,%m会被01到12之间的一个十进制月份数替代,其他依次类推,%在python的格式化输出,有转换字符的作用,本文给大家讲解python中的% 是什么意思,起到什么作用,感兴趣的朋友一起看看吧
    2022-12-12
  • python入门教程 python入门神图一张

    python入门教程 python入门神图一张

    这篇文章主要介绍了Python 10分钟入门教程,分享一张Python入门神图一张,具有一定的参考价值,感兴趣的小伙伴们可以参考一下
    2018-03-03
  • Python使用CMD模块更优雅的运行脚本

    Python使用CMD模块更优雅的运行脚本

    这篇文章主要介绍了Python使用CMD模块更优雅的运行脚本的方法,实例分析了Python中cmd模块的相关使用技巧,具有一定参考借鉴价值,需要的朋友可以参考下
    2015-05-05
  • Python中拆包的实现

    Python中拆包的实现

    拆包是一个非常实用且常见的操作,它能够简化代码,文中通过示例代码介绍的非常详细,对大家的学习或者工作具有一定的参考学习价值,需要的朋友们下面随着小编来一起学习学习吧
    2024-07-07
  • baselines示例程序train_cartpole.py的ImportError

    baselines示例程序train_cartpole.py的ImportError

    这篇文章主要为大家介绍了baselines示例程序train_cartpole.py的ImportError引入错误详解,有需要的朋友可以借鉴参考下,希望能够有所帮助,祝大家多多进步,早日升职加薪
    2022-05-05
  • python django下载大的csv文件实现方法分析

    python django下载大的csv文件实现方法分析

    这篇文章主要介绍了python django下载大的csv文件实现方法,结合实例形式分析了Django框架下载csv大文件的相关操作技巧与注意事项,需要的朋友可以参考下
    2019-07-07

最新评论