Python自学成才之路 with到上下文管理器

标签: python

在做文件IO操作时,有一个with语句,他能够自动的对文件进行打开和关闭操作。


with open('d:\\abc.txt', 'w') as f:
    f.write('hello world')

With有这个特效,其实背后实际上是基于__enter__和__exit__这两个魔术方法来实现的。一个对象实现了__enter__和__exit__这两个魔术方法后,也能使用with语法。举个栗子,把大象放进冰箱有三步操作,第一步打开冰箱,第二步把大象放进冰箱,第三步关上冰箱,如果使用with语法可以这样实现:


class Elephant(object):

    def __init__(self):
        pass

    def __enter__(self):
        print('open ice box')

    def __exit__(self, exc_type, exc_val, exc_tb):
        print('close ice box')


with Elephant():
    print('put elephant into ice box')
输出:
open ice box
put elephant into ice box
close ice box

上下文管理器(context manager)
前面说with的IO操作特效是基于__enter__和__exit__这两个魔术方法来实现的,其实更专业的说法,这叫做上下文管理器。

上下文管理器是指在一段代码执行之前执行一段代码,用于一些预处理工作;执行之后再执行一段代码,用于一些清理工作。比如打开文件进行读写,读写完之后需要将文件关闭。在上下文管理协议中,有两个方法__enter__和__exit__,分别实现上述两个功能。

使用with语法格式是这样的:


with EXPR as VAR:
    BLOCK

整个执行流程:

  1. 执行expr语句,获得上下文管理器
  2. 调用__enter__方法,做一些预处理操作
  3. as var 是接收__enter__执行后返回的结果,如果没有返回结果,as var可以省略掉
  4. 执行block
  5. 执行__exit__方法,__exit__方法有三个参数,这三个参数是用来处理异常信息的,exc_type表示异常类型,exc_val表示异常信息,exc_tb表示异常堆栈信息。__exit__方法也可以有返回值,如果为True,那么表示异常被忽视,相当于进行了try-except操作;如果为False,则该异常会被重新raise。

为了演示效果,在上面那个把大象放入冰箱的例子中人为加一个异常:


class Elephant(object):

    def __init__(self, name):
        self.name = name

    def __enter__(self):
        print('open ice box')
        return self.name

    def __exit__(self, exc_type, exc_val, exc_tb):
        print('close ice box')
        print(exc_type)
        print(exc_val)
        print(exc_tb)
        return True


with Elephant('peter') as name:
    print('elephant name: %s'%name)
    print('put elephant into ice box')
    1/0
输出:
open ice box
elephant name: peter
put elephant into ice box
close ice box
<class 'ZeroDivisionError'>
division by zero
<traceback object at 0x000001B1E179C148>

__exit__方法默认返回false,如果返回true异常会在__exit__内部被消化掉,如果返回false异常会重新抛出来。

contextlib实现上下文管理器的功能
通过with实现上下文管理器的功能需要实现两个方法,contextlib库中有一个contextmanage装饰器,可以通过这个装饰器以一种更简单的方式来实现上下文管理器的功能。

contextmanage中yield之前的代码类似于__enter__的作用,yield的返回值赋值给as之后的变量,yield之后的代码类似于_exit__的作用。之前有讲过yield放在函数中表示的是生成器,但是如果函数加了contextmanager装饰器,则该函数就变成了了上下文管理器了。


import contextlib

@contextlib.contextmanager
def elephant(name):
    print('open ice box')

    yield name

    print('close ice box')


with elephant('peter') as name:
    print('elephant name: %s'%name)
    print('put elephant into ice box')
输出:
open ice box
elephant name: peter
put elephant into ice box
close ice box

和with不同的是,contextmanage实现的上下文管理器没有对异常做捕获,需要自己处理。

上下文管理器的应用
自定义一个文件io操作的上下文管理器。


class FileOpener(object):

    def __init__(self, filename, filemode):
        self.filename = filename
        self.filemode = filemode

    def __enter__(self):
        self.fp = open(self.filename, self.filemode)
        return self.fp

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


with FileOpener('test.txt', 'w') as fp:
    fp.write('hello')




本人是做大数据开发的,在微信上开了个个人号,会经常在上面分享一些学习心得,原创文章都会首发到公众号上,感兴趣的盆友可以关注下哦!
在这里插入图片描述
备注:微信公众号搜索‘大数据入坑指南

版权声明:本文为u012062455原创文章,遵循 CC 4.0 BY-SA 版权协议,转载请附上原文出处链接和本声明。
本文链接:https://blog.csdn.net/u012062455/article/details/108310339

智能推荐

一致性hash算法

散列(hash)在我看来就是一个数组,而与数组不同的点在于数组是按顺序写入的,而hash是按照一定的hash算法确定元素在数组中的位置的。hash最难的问题在于会有冲突出现,如果两个object根据相应的hash算法得出的值一样便产生了hash冲突。在所有解决hash冲突的方法中,我最欣赏的是链式解决法,即将hash到同一位置的元素用链表连接。当然还有其它几种处理hash冲突的算法,比如建立公共溢...

OpenCV-Python learning-1.安装,图片读取显示

1. OpenCV与OpenGL区别 https://www.zhihu.com/question/20212016 一个是让机器识别东西的,OpenCV是给电脑做眼睛的。 一个是让机器计算出更好画面的,OpenGL用在游戏渲染方面很多。 OpenCV(Open Source Computer Vision Library)是一个基于(开源)发行的跨平台计算机视觉库,OpenGL(全写Open G...

Mycat+Mysql分布式架构改造和性能压力测试

架构实现 Mycat作为数据库高可用中间件具备很多的功能,如负载均衡,分库分表,读写分离,故障迁移等。结合项目的实际情况,分库分表功能对于关联查询有很高的要求,需要从业务角度考虑分库分表后的关联查询SQL的分析,业务代码动作较大,所以在此方案中我们不考虑分库分表。主要应用Mycat的负载均衡及故障迁移的功能即可。 整个架构改造包括两个部分,第一是单例Mysql改为多个Mysql,同时负载均衡,并且...

人脸识别之疲劳检测(二)阈值法、KNN分类和K-means聚类

Table of Contents 1、均值法 2、中值法 3、KNN 4、K-means 结合上一节在获得人眼特征点后需要对睁眼闭眼状态做出判断,方法的选择需要经验结合公平的评价方法,使用大量测试集得到不同方法下的精确度并做出比较: 1、均值法 50帧睁眼数据取均值,得到不同阈值下精确度。 2、中值法 50帧睁眼数据取中值,得到不同阈值下精确度。 3、KNN KNN是一种ML常用分类算法,通过测...

CodeForce Tic-Tac-Toe

Two bears are playing tic-tac-toe via mail. It's boring for them to play usual tic-tac-toe game, so they are a playing modified version of this game. Here are its rules. The game is played on the foll...

猜你喜欢

Python雾里看花-抽象类ABC (abstract base class)

首先认识模块 abc,python中没有提供抽象类与抽象方法,然而提供了内置模块abc来模拟实现抽象类,例如提供泛映射类型的抽象类 abc.MutableMapping 继承abc.MutableMapping构造一个泛映射类型(类似python中的dict) 当然继承abc.Mapping 也可以,毕竟MutableMapping是其子类 dict是python中典型的映射类型数据结构,其接口的...

python 文件操作

2, with open (‘xx.txt’,‘w’,encoding=‘utf-8’) as f: f.write(‘文件内容或对象’)...

【Python基础】使用统计函数绘制简单图形

机器学习算法与自然语言处理出品 @公众号原创专栏作者 冯夏冲 学校 | 哈工大SCIR实验室在读博士生 2.1 函数bar 用于绘制柱状图 2.2 函数barh 用于绘制条形图 2.3 函数hist 用于绘制直方图 直方图与柱状图的区别 函数pie 用于绘制饼图 2.5 函数polor 用于绘制极线图 极线图是在极坐标系上绘出的一种图。在极坐标系中,要确定一个点,需要指明这个点距原点的角...

css:顶部按钮固定,上面内容滑动

这种需求我们平时见到很多的,实现方法也多的参差不齐,下面我说一种简单的。如图: 可以看到只有红线部分滚动,底下按钮是固定的。 代码...

环形公路堵车概率模型(含详细解析)

文章目录 基础理论 代码实现 图形分析 基础理论 路面上有n辆车,以不同的速度向前行驶, 模拟堵车问题。 有以下假设: 假设某辆车的当前速度是v。 若前方可见范围内没车,则它在下一秒的车速提高到v+1,直到达到规定的最高限速。 若前方有车,前车的距离为d,且d < v,则它下 一秒的车速降低到d-1 。 每辆车会以概率p随机减速v-1。、 代码实现 图形分析 图形中颜色越重的地方,说明很多车...