一文带你搞懂Python上下文管理器

 更新时间:2022年06月23日 15:57:44   作者:曲鸟  
这篇文章主要为大家介绍了Python上下文管理器,具有一定的参考价值,感兴趣的小伙伴们可以参考一下,希望能够给你带来帮助

一、什么是上下文管理器

我们在处理文件的时候经常看到下面这样的代码,它即是上下文管理器:

with open('test.txt', encoding='utf-8') as f:
    print(f.readlines())

它的含义是打开当前目录下的test.txt文件并打印它里面的内容,与下面的代码效果是一样的:

f = open('test.txt', encoding='utf-8')
print(f.readlines())
f.close()

对比两种写法能够发现,使用with自动执行了f.close()(关闭文件)的这步操作,能够少写一点代码。

那这样的上下文管理器是怎么实现的,下面为你讲解。

二、如何实现上下文管理器

1. 通过类实现

如果要实现上面open的上下文管理器功能,我们可以通过创建一个类,并添加__enter____exit__方法即可,如下面的代码所示:

class DiyOpen(object):

    def __init__(self, filename, **kwargs):
        self.f = open(filename, **kwargs)

    def __enter__(self):
        return self.f

    def __exit__(self, exc_type, exc_val, exc_tb):
        print('关闭文件')
        self.f.close()


with DiyOpen('test.txt', encoding='utf-8') as f:
    print(f.readlines())

输出结果

['第一行\n', '第二行\n', '第三行']
关闭文件

可以看到在我们打印出文件内容后,自动执行了关闭文件的操作。

__enter____exit__的含义是什么,__exit__后面的exc_type, exc_val, exc_tb又是什么意思呢?

1)_enter_

__enter__相对来说好理解的多,当出现with语句时,它就会被触发,有返回值时,会把返回值赋值给as声明的变量,也就是我们上面的as f中的f。

2)_exit_

__exit__是在with执行完成后自动执行的,他后面的参数含义如下:

  • exc_type:异常类型
  • exc_val:异常原因e
  • xc_tb:堆栈追踪信息

当with中执行的代码报错时,除了不继续执行with包含的代码外,还会将报错信息放入上面的三个参数中,例如下面的代码:

class DiyOpen(object):

    def __init__(self, filename, **kwargs):
        self.f = open(filename, **kwargs)

    def __enter__(self):
        return self.f

    def __exit__(self, exc_type, exc_val, exc_tb):
        print(exc_type)
        print(exc_val)
        print(exc_tb)
        self.f.close()


with DiyOpen('test.txt', encoding='utf-8') as f:
    print(f.no())

输出结果

<class 'AttributeError'>
'_io.TextIOWrapper' object has no attribute 'no'
<traceback object at 0x000002A34B834900>

需要注意的是:

  • 我们可以手动指定__exit__的返回值为True让它不报错。
  • 没有异常信息时,上面的三个参数值都会为None

2. 通过contextlib实现

Python内置了contextlib这个模块用于实现上下文管理器,它是通过生成器yield实现的,这个模块让我们不必再创建类和__enter__和__exit__了。

通过contextlib实现open功能的代码如下:

from contextlib import contextmanager

@contextmanager
def diy_open(filename, **kwargs):
    f = open(filename, **kwargs)  # __init__
    try:
        yield f  # __enter__
    finally:  # __exit__
        f.close()

with diy_open('test.txt', encoding='utf-8') as f:
    print(f.readlines())

总结

本篇文章就到这里了,希望能够给你带来帮助,也希望您能够多多关注脚本之家的更多内容!

相关文章

  • 浅要分析Python程序与C程序的结合使用

    浅要分析Python程序与C程序的结合使用

    这篇文章主要介绍了Python程序与C程序的结合使用,包括Python程序如何利用C程序的dll外链等等,来自IBM官网的技术文档,需要的朋友可以参考下
    2015-04-04
  • Python ArcPy实现批量拼接长时间序列栅格图像

    Python ArcPy实现批量拼接长时间序列栅格图像

    这篇文章主要介绍了如何基于Python中ArcPy模块,对大量不同时相的栅格遥感影像按照其成像时间依次执行批量拼接的方法,感兴趣的可以了解一下
    2023-03-03
  • python中文编码与json中文输出问题详解

    python中文编码与json中文输出问题详解

    Python的编码问题还是很让人头疼的,做下笔记帮助别人,也帮助自己,下面这篇文章主要给大家介绍了关于python中文编码与json中文输出问题的相关资料,文中通过示例代码介绍的非常详细,需要的朋友可以参考下
    2018-08-08
  • Python实现的HMacMD5加密算法示例

    Python实现的HMacMD5加密算法示例

    这篇文章主要介绍了Python实现的HMacMD5加密算法,简单说明了HMAC-MD5加密算法的概念、原理并结合实例形式分析了Python实现HMAC-MD5加密算法的相关操作技巧,,末尾还附带了Java实现HMAC-MD5加密算法的示例,需要的朋友可以参考下
    2018-04-04
  • Python中的heapq模块源码详析

    Python中的heapq模块源码详析

    这篇文章主要给大家介绍了关于Python中heapq模块的相关资料,文中通过示例代码介绍的非常详细,对大家学习或者使用python具有一定的参考学习价值,需要的朋友们下面随着小编来一起学习学习吧
    2019-01-01
  • 如何使用python-opencv批量生成带噪点噪线的数字验证码

    如何使用python-opencv批量生成带噪点噪线的数字验证码

    这篇文章主要介绍了如何使用python-opencv批量生成带噪点噪线的数字验证码,文中通过示例代码介绍的非常详细,对大家的学习或者工作具有一定的参考学习价值,需要的朋友们下面随着小编来一起学习学习吧
    2020-12-12
  • Python使用PyYAML库读写yaml文件的方法

    Python使用PyYAML库读写yaml文件的方法

    这篇文章主要介绍了Python使用PyYAML库读写yaml文件的方法,包括YAML基本概念介绍及YAML语法知识,结合示例代码给大家介绍的非常详细,需要的朋友可以参考下
    2022-04-04
  • 利用python绘制CDF图的示例代码

    利用python绘制CDF图的示例代码

    这篇文章主要为大家详细介绍了如何利用python实现绘制CDF图,文中的示例代码简洁易懂,具有一定的借鉴价值,感兴趣的小伙伴可以学习一下
    2023-08-08
  • 详解pandas库pd.read_excel操作读取excel文件参数整理与实例

    详解pandas库pd.read_excel操作读取excel文件参数整理与实例

    这篇文章主要介绍了pandas库pd.read_excel操作读取excel文件参数整理与实例,小编觉得挺不错的,现在分享给大家,也给大家做个参考。一起跟随小编过来看看吧
    2019-02-02
  • Python任意字符串转16, 32, 64进制的方法

    Python任意字符串转16, 32, 64进制的方法

    今天小编就为大家分享一篇Python任意字符串转16, 32, 64进制的方法,具有很好的参考价值,希望对大家有所帮助。一起跟随小编过来看看吧
    2019-06-06

最新评论