Python上下文管理器你学会了吗?
什么是上下文管理器
对于像文件操作、连接数据库等资源管理的操作,我们必须在使用完之后进行释放,不然就容易造成资源泄露。为了解决这个问题,Python的解决方式便是上下文管理器。上下文管理器能够帮助你自动分配并且释放资源,其中最典型的应用便是with语句。我们来看一下打开文件的例子。
for x in range(10000): f = open('test.txt', 'w') f.write('hello world')
这段代码表示我们打开了1万个文件,但是用完之后没有进行关闭,这是一个典型的资源泄露的例子,这是一个错误的示例。我们应该这么写。
for x in range(10000): with open('test.txt', 'w') as f: f.write('hello world')
这样我们每次打开文件“test.txt”,并写入“hello world”之后,这个文件便会自动关闭,相应的资源也会释放,防止资源泄露。with语句的代码,我们还可以用下面的形式代替。
f = open('test.txt', 'w') try: f.write('hello world') finally: f.close()
要注意的是,最后的 finally语句尤其重要,哪怕在写入文件时发生错误异常,它也可以保证该文件最终被关闭。不过与 with 语句相比,这样的代码就显得冗余了,并且还容易漏写,因此我们一般更倾向于使用 with 语句。
上下文管理器的实现
接下来我们详细分析一下它的内部原理和实现。这里我们定义了一个上下管理器类FileManager,模拟Python打开、关闭文件的操作。
class FileManager: def __init__(self, name, mode): print('calling __init__ method') self.name = name self.mode = mode self.file = None def __enter__(self): print('calling __enter__ method') self.file = open(self.name, self.mode) return self.file def __exit__(self, exc_type, exc_val, exc_tb): print('calling __exit__ method') if self.file: self.file.close() with FileManager('test.txt', 'w') as f: f.write('hello world') print('write success') ####输出#### calling __init__ method calling __enter__ method write success calling __exit__ method
这个是基于类的上下文管理器的实现,当我们使用类来创建上下文管理器时,我们需要包含“__enter__”和“__exit__”方法。其中,“__enter__”方法返回需要被管理的资源,方法“__exit__”里通常会做一些释放、清理资源的操作。当我们用with语句执行这个上下文管理器时,依次会执行如下操作。
-
首先方法“__init__()”被调用,程序初始化对象 FileManager,使得文件名(name)是“test.txt”,文件模式 (mode) 是“w”;
-
方法“__enter__()”被调用,文件“test.txt”以写入的模式被打开,并且返回 FileManager 对象赋予变量 f;
-
字符串“hello world”被写入文件“test.txt”;
-
方法“__exit__()”被调用,负责关闭之前打开的文件流。
值得一提的是,方法“__exit__()”中的参数“exc_type, exc_val, exc_tb”,分别表示 exception_type、exception_value 和 traceback。当我们执行含有上下文管理器的 with 语句时,如果有异常抛出,异常的信息就会包含在这三个变量中,传入方法“__exit__()”。因此,如果你需要处理可能发生的异常,可以在“__exit__()”添加相应的处理异常的代码。
python中的上下文管理器除了基于类的实现还可以基于生成器的实现,我们接着来看下面这个例子。你可以使用装饰器 contextlib.contextmanager,来定义自己所需的基于生成器的上下文管理器,用以支持 with 语句。还是拿前面的类上下文管理器 FileManager 来说,我们也可以用下面形式来表示:
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() 是一个生成器,当我们执行 with 语句时,便会打开文件,并返回文件对象 f;当 with 语句执行完后,finally block 中的关闭文件操作便会执行。
到此,我们已经把两种上下文管理器的实现都介绍完了。这两者在功能上是一致的,只不过,基于类的上下文管理器更加灵活,适用于大型的系统开发。而基于生成器的上下文管理器更加方便、简洁,适用于中小型程序。
欢迎大家留言和我交流。更多有趣的内容,请关注公众号 老韩随笔
原文地址:https://www.cnblogs.com/laohanshuibi/p/15003307.html
- phalcon-入门篇3(优美的URL与Config)
- 数据库中间件 Sharding-JDBC 源码分析 —— 事务(一)之BED
- 熔断器 Hystrix 源码解析 —— 命令执行(二)之执行隔离策略
- phalapi-入门篇4(国际化高可用和自动生成文档)
- 用JavaScript动态输出的JS脚本不能执行
- Dubbo源码解析 —— 服务暴露原理
- [Golang软件推荐] RSA公私钥加解密(解决Golang私钥加密公钥解密问题)
- [喵咪大数据]Hive+Hbase关联
- 再战子域共享Cookie问题
- [喵咪大数据]Presto查询引擎
- 如何在5分钟内做出你的第一个开源贡献
- [喵咪大数据]HUE大数据管理工具
- Dubbo源码解析 —— Zookeeper 订阅
- 注册中心 Eureka 源码解析 —— 项目结构简介
- JavaScript 教程
- JavaScript 编辑工具
- JavaScript 与HTML
- JavaScript 与Java
- JavaScript 数据结构
- JavaScript 基本数据类型
- JavaScript 特殊数据类型
- JavaScript 运算符
- JavaScript typeof 运算符
- JavaScript 表达式
- JavaScript 类型转换
- JavaScript 基本语法
- JavaScript 注释
- Javascript 基本处理流程
- Javascript 选择结构
- Javascript if 语句
- Javascript if 语句的嵌套
- Javascript switch 语句
- Javascript 循环结构
- Javascript 循环结构实例
- Javascript 跳转语句
- Javascript 控制语句总结
- Javascript 函数介绍
- Javascript 函数的定义
- Javascript 函数调用
- Javascript 几种特殊的函数
- JavaScript 内置函数简介
- Javascript eval() 函数
- Javascript isFinite() 函数
- Javascript isNaN() 函数
- parseInt() 与 parseFloat()
- escape() 与 unescape()
- Javascript 字符串介绍
- Javascript length属性
- javascript 字符串函数
- Javascript 日期对象简介
- Javascript 日期对象用途
- Date 对象属性和方法
- Javascript 数组是什么
- Javascript 创建数组
- Javascript 数组赋值与取值
- Javascript 数组属性和方法