深人了解Python上下文管理器
下面先来介绍一下with关键字在文件读写中的应用，简单了解上下文管理器的功能。
with语句
在Python文件及目录处理方法中介绍了读写大文件建议使用with语句，with语句会进行资源的自动管理。文件很多的情况下也会导致资源泄露，下面来打开100000个文件，不进行文件关闭操作：
for x in range(100000):
file = open('test.txt', 'w')
file_descriptors.append(file)
执行会报如下错误：
OSError: [Errno 24] Too many open files: 'test.txt'
原因就是打开了太多文件而没有及时关闭导致了资源泄露，造成系统崩溃。完成处理后需要对文件进行关闭操作：
file_descriptors = []
for x in range(10000):
file = open('test.txt', 'w')
try:
file_descriptors.append(file)
finally:
file.close()
使用 with 语句可以完成自动分配并且释放资源，比上面的写法更加简洁：
file_descriptors = []
for x in range(10000):
with open('test.txt', 'w') as file:
file_descriptors.append(file)
上下文管理器创建
基于类的上下文管理器
可以使用类来创建上下文管理器，需要保证这个类包括两个方法：__enter__() 和__exit__()。其中，方法 __enter__() 返回需要被管理的资源，方法 __exit__() 进行资源释放、清理操作。
下面来模拟 Python 的打开、关闭文件操作：
class FileManager:
def __init__(self, name, mode):
print('__init__ method called')
self.name = name
self.mode = mode
self.file = None
def __enter__(self):
print('__enter__ method called')
self.file = open(self.name, self.mode)
return self.file
def __exit__(self, exc_type, exc_value, exc_traceback):
print('__exit__ method called')
if self.file:
self.file.close()
if exc_type:
print(f'exc_type: {exc_type}')
print(f'exc_value: {exc_value}')
print(f'exc_traceback: {exc_traceback}')
return True
with FileManager('test.txt', 'w') as f:
print('开始写操作')
f.write('hello world !')
print(f.closed)
执行结果：
__init__ method called
__enter__ method called
开始写操作
__exit__ method called
exc_type: <class 'Exception'>
exc_value: exception raised
exc_traceback: <traceback object at 0x000001B43C2444C8>
True
可以看到执行顺序为：
__init__()：初始化对象 FileManager__enter__()：打开文件，返回 FileManager 对象
with中的代码
__exit__()：关闭打开的文件流
__exit__()方法中的参数exc_type, exc_value, 和 exc_traceback 用于管理异常。
@contextmanager 装饰器
可以使用 contextlib.contextmanager 装饰器而不使用类的方式来实现上下文管理器，它是基于生成器的上下文管理器，用以支持 with 语句。
仍以打开、关闭文件为例：
from contextlib import contextmanager
@contextmanager
def file_manager(name, mode):
try:
f = open(name, mode)
yield f
finally:
f.close()
with file_manager('test.txt', 'w') as f:
f.write('hello world !')
其中 file_manager() 函数是一个生成器，yield 之前可以看成是__enter__方法中的内容，yield 后面的是 __exit__() 内容。加上@contextmanager装饰器，使用基于生成器的上下文管理器时，不需要定义__enter__()和__exit__()方法。
总结
上下文管理器可确保用过的资源得到迅速释放，通常和 with 语句一起使用，大大提高了程序的简洁度。另外需要注意的是，编写基于类或者生成器的上下文管理器时，记住不要忘记释放资源。--THE END--
本篇文章就到这里了，希望能够给你带来帮助，也希望您能够多多关注脚本之家的更多内容!
相关文章
这篇文章主要介绍了如何使用Python 绘制瀑布图，我们一起了解瀑布图的重要性，以及如何使用不同的绘图库绘制瀑布图。瀑布图是一种二维图表，专门用于了解随着时间或多个步骤或变量的增量正负变化的影响，下文更多详细内容需要的小伙伴可以参考一下2022-05-05
算法分析的主要目标是从运行时间和内存空间消耗等方面比较算法。本文将为大家详细介绍Python数据结构与算法中的算法分析，需要的可以参考一下2021-12-12
这篇文章主要介绍了Pytorch中的数据集划分&正则化方法，具有很好的参考价值，希望对大家有所帮助。如有错误或未考虑完全的地方，望不吝赐教2021-05-05
今天小编就为大家分享一篇postman模拟访问具有Session的post请求方法，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧2019-07-07
今天小编就为大家分享一篇在Python中将函数作为另一个函数的参数传入并调用的方法，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧2019-01-01
这篇文章主要介绍了python协程与 asyncio 库详情，文章基于python展开详细内容，具有一定的参考价值。需要的小伙伴可以参考一下2022-05-05
这篇文章主要介绍了python自然语言编码转换模块codecs介绍,codecs专门用作编码转换,通过它的接口是可以扩展到其他关于代码方面的转换,需要的朋友可以参考下2015-04-04
这篇文章主要介绍了Python查询IP地址归属完整代码，具有一定的参考价值，感兴趣的小伙伴们可以参考一下2017-06-06
这篇文章主要介绍了解决安装Python的第三方库pandas报错问题,具有很好的参考价值,希望对大家有所帮助,如有错误或未考虑完全的地方,望不吝赐教2023-08-08
这篇文章主要为大家详细介绍了Python实现渐变色的水平堆叠图，文中示例代码介绍的非常详细，具有一定的参考价值，感兴趣的小伙伴们可以参考一下2022-04-04
最新评论