​什么是上下文管理器

对于像文件操作、连接数据库等资源管理的操作,我们必须在使用完之后进行释放,不然就容易造成资源泄露。为了解决这个问题,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语句执行这个上下文管理器时,依次会执行如下操作。

  1. 首先方法“__init__()”被调用,程序初始化对象 FileManager,使得文件名(name)是“test.txt”,文件模式 (mode) 是“w”;

  2. 方法“__enter__()”被调用,文件“test.txt”以写入的模式被打开,并且返回 FileManager 对象赋予变量 f;

  3. 字符串“hello world”被写入文件“test.txt”;

  4. 方法“__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 中的关闭文件操作便会执行。

到此,我们已经把两种上下文管理器的实现都介绍完了。这两者在功能上是一致的,只不过,基于类的上下文管理器更加灵活,适用于大型的系统开发。而基于生成器的上下文管理器更加方便、简洁,适用于中小型程序。

欢迎大家留言和我交流。更多有趣的内容,请关注公众号 老韩随笔

Python上下文管理器你学会了吗?的更多相关文章

  1. python 上下文管理器

    作者:Vamei 出处:http://www.cnblogs.com/vamei 欢迎转载,也请保留这段声明.谢谢! 上下文管理器(context manager)是Python2.5开始支持的一种语 ...

  2. Python 上下文管理器和else块

    最终,上下文管理器可能几乎与子程序(subroutine)本身一样重要.目前,我们只了解了上下文管理器的皮毛--Basic 语言有with 语句,而且很多语言都有.但是,在各种语言中 with 语句的 ...

  3. Python上下文管理器

    在Python中让自己创建的函数.类.对象支持with语句,就实现了上线文管理协议.我们经常使用with open(file, "a+") as f:这样的语句,无需手动调用f.c ...

  4. python上下文管理器ContextLib及with语句

    http://blog.csdn.net/pipisorry/article/details/50444736 with语句 with语句是从 Python 2.5 开始引入的一种与异常处理相关的功能 ...

  5. Python上下文管理器 with

    对于系统资源的操作,如:文件操作.数据库操作等,我们往往打开文件.连接数据库后忘了将其close掉,这时就可能会引发异常,因此我们常用的做法是: # coding:utf-8 f = open(&qu ...

  6. Python上下文管理器(Context managers)

    上下文管理器(Context managers) 上下文管理器允许你在有需要的时候,精确地分配和释放资源. 使用上下文管理器最广泛的案例就是with语句了.想象下你有两个需要结对执行的相关操作,然后还 ...

  7. python上下文管理器细读

    test 1 上下文管理器,将生成器转化为上下文管理器 import contextlib @contextlib.contextmanager def a(): print(1) yield pri ...

  8. 吃透Python上下文管理器

    什么是上下文管理器? 我们常见的with open操作文件,就是一个上下文管理器.如: with open(file, 'rb') as f: text = f.read() 那上下文管理器具体的定义 ...

  9. python 上下文管理器contextlib.ContextManager

    1 模块简介 在数年前,Python 2.5 加入了一个非常特殊的关键字,就是with.with语句允许开发者创建上下文管理器.什么是上下文管理器?上下文管理器就是允许你可以自动地开始和结束一些事情. ...

随机推荐

  1. linux rpm包解压

    rpm2cpio xxx.rpm | cpio -div

  2. Centos6.8安装mysql 步骤

    第1步.查看CentOS下是否已安装mysql 输入命令 :yum list installed | grep mysql 第2步.删除已安装mysql 输入命令:yum -y remove mysq ...

  3. linux基础之进阶命令二

    本节内容 基础命令二: 1.      alias:查看和设置别名(alias:别名) alias 'ren=ls -ld' \ren 可不取消别名,执行ren原本的命令 2.      unalia ...

  4. 用PHP爬取知乎的100万用户

    http://blog.jobbole.com/88788/ 突然发现 大数据 Python的爬虫能力很强 爬取到的数据 直接可以用于维修QQ营销 精准营销

  5. 03丨认识文档数据库MongoDB

  6. Python判断身份证是否合法

    利用正则表达式实现对身份证合法程度的判断 1 # !usr/bin/env python3 2 # coding:utf-8 3 """ 4 @ Copyright (c ...

  7. Java核心技术卷阅读随笔--第4章【对象与类】

    对 象 与 类 4.1 面向对象程序设计概述 面向对象程序设计(简称 OOP) 是当今主流的程序设计范型, 它已经取代了 20 世纪 70 年代的" 结构化" 过程化程序设计开发技 ...

  8. .NET6系列:C#10新功能预览

    系列目录     [已更新最新开发文章,点击查看详细] 2021年4月19日微软发布公告称将于今年夏季发布首款64位的 Visual Studio 2022,2021年5月20日又发布了 Visual ...

  9. MyBatis-框架使用和分析

      一.基础知识 MyBatis 是一款优秀的持久层框架,它支持自定义 SQL.存储过程以及高级映射.MyBatis 免除了几乎所有的 JDBC 代码以及设置参数和获取结果集的工作.MyBatis 可 ...

  10. 微服务架构(Microservices) ——Martin Flower

    不知不觉到达了Sring Boot的学习中了,在学习之前,了解微服务架构是很有必要的,对于自己提升今后面试的软实力有很大帮助,在此写下. 让我们接下来看下Martin Flower 如何解释微服务架构 ...