Python的with语句与上下文管理器

① with语句就是简洁版的try/finally语句

在我们日常使用场景中,经常会操作一些资源,比如文件对象、数据库连接、Socket连接等,资源操作完了之后,不管操作的成功与否,最重要的事情就是关闭该资源,否则资源打开太多而没有关闭,程序会报错,以文件操作为例,通常我们会这样写:

f = open('file.txt', 'w')
try:
    f.write("Hello")
finally:
    f.close()

但既然close方法是必须的操作,那就没必要显式地调用,所以Python给我们提供了一种更优雅的方式,使用with语句:

with open('file.txt', 'w') as f:
    f.write("Hello")

在退出with语句下的代码块之后,f 对象会自动执行自己的close方法,实现资源的释放,简洁优雅。

② 上下文管理器原理

f 对象之所以会自动执行自己的close方法,是因为它是一个上下文管理器,所以我们要先说说什么是上下文管理器。
上下文管理器是内部实现了enterexit方法的对象(比如下面EXPR表达式获取到的对象)
它的一般使用方法是:

with EXPR as VAR:
    BLOCK

上述代码的执行过程等价于:

ContextManager = EXPR
VAR = ContextManager.__enter__()
try:
    BLOCK
finally:
    ContextManager.__exit__()

f 对象就是把自己的close方法定义在了它的exit方法内部,实现了代码块BLOCK执行完之后自动关闭自身。

③ 自定义上下文管理器

为了验证上述的执行过程,我们定义一个文件类,内部实现了enterexit两个方法:

class File:
    def __init__(self, filename, mode):
        self.filename = filename
        self.mode = mode

    def __enter__(self):
        print("进入")
        self.f = open(self.filename, self.mode)
        return self.f

    def __exit__(self, exc_type=None, exc_val=None, exc_tbs=None):
        print("退出")
        self.f.close()

这时候File类就是一个上下文管理器
我们分别通过 with语句 和 try/finally语句 使用File类对文件进行写入操作
通过with语句执行:

with File('file.txt', 'w') as f:
    print("正在写入...")
    f.write('Hello')

控制台输出:

进入
正在写入...
退出

并得到了一个写了 Hello 的 file.txt 文件

通过try/finally语句执行:

ContextManager = File('file.txt', 'w')
VAR = ContextManager .__enter__()
try:
    print("正在写入...")
    VAR.write('Hello')
finally:
    ContextManager.__exit__()

控制台输出:

进入
正在写入...
退出

并得到了一个写了 Hello 的 file.txt 文件

两者输出一致,所以验证了②中执行过程的等价关系是正确的

enterexit方法说明

1. enter方法说明

上下文管理器的enter方法是可以带返回值的,默认返回None,这个返回值通过with...as...中的 as 赋给它后面的那个变量,所以 with EXPR as VAR 就是将EXPR对象enter方法的返回值赋给 VAR

当然with...as...并非固定组合,单独使用with...也是可以的,上下文管理器的enter方法还是正常执行,只是这个返回值并没有赋给一个变量,with下面的代码块也不能使用这个返回值。

2. exit方法说明

上下文管理器的exit方法接收3个参数exc_type、exc_val、exc_tb,如果代码块BLOCK发生了异常e并退出,这3个参数分别为type(e)、str(e)、e.__traceback__,否则都为None。

同样exit方法也是可以带返回值的,这个返回值应该是一个布尔类型True或False,默认为None(即False)。如果为False,异常会被抛出,用户需要进行异常处理。如果为True,则表示忽略该异常。

⑤ contextmanager 装饰器

Python还提供了一个contextmanager装饰器,允许用户将一个生成器定义为上下文管理器,该装饰器将生成器中的代码通过yield语句分成两部分,yield之前的代码为enter方法,yield之后的代码为exit方法,yield的返回值即enter方法的返回值,用于赋给as后的变量。
下面我们通过contextmanager装饰器也实现一个关于文件的上下文管理器:

from contextlib import contextmanager

@contextmanager
def open_file(filename, mode):
    print('进入')
    f = open(filename, mode)
    try:
        yield f
    finally:
        print('退出')
        f.close()

说明:这里使用 try/finally 是确保yield的过程中就算出现异常,文件也能正常关闭,当然这里也能处理异常,使用 try/except/finally 即可。

通过with语句执行:

with open_file('file.txt', 'w') as f:
    print("正在写入...")
    f.write('Hello')

执行结果跟之前的上下文管理器执行结果一致,说明contextmanager装饰器也能定义一个上下文管理器。

之前写过一篇文章阐述了装饰器的原理,现在也知道了上下文管理器就是内部实现了enterexit方法的对象,所以有兴趣的同学就可以去看看contextmanager装饰器的源码了,看看它是怎么把一个生成器变成上下文管理器的,其实稍微想想就知道,肯定是contextmanager装饰器把生成器加工成一个内部实现了enterexit方法的对象,有时间的话另写一篇文章带大家看看contextmanager的源码。

⑥ 总结

1. with语句是try/finally语句的替代,简化了资源调用之后的清理工作

2. with语句的操作对象是上下文管理器,它是内部实现了enterexit方法的对象

3. enter方法可以带返回值,它通过with...as...语句中的as赋给它后面的变量;exit可返回布尔值,如果为False,异常会被抛出,用户需要进行异常处理,如果为True,则表示忽略该异常

4. Python还提供了contextmanager装饰器,进一步简化了上下文管理器的定义方法,它通过把生成器加工成一个内部实现了enterexit方法的对象,从而实现生成器变上下文管理器

最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 161,513评论 4 369
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 68,312评论 1 305
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 111,124评论 0 254
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 44,529评论 0 217
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 52,937评论 3 295
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 40,913评论 1 224
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 32,084评论 2 317
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 30,816评论 0 205
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 34,593评论 1 249
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 30,788评论 2 253
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 32,267评论 1 265
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 28,601评论 3 261
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 33,265评论 3 241
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 26,158评论 0 8
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 26,953评论 0 201
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 36,066评论 2 285
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 35,852评论 2 277

推荐阅读更多精彩内容

  • 转载自:http://mp.weixin.qq.com/s/LO1yyFeUA6pR_YPyfDoSig 姓名:梅...
    虐先森阅读 1,385评论 0 1
  • contextlib — Context Manager Utilities contextlib - 上下文管理...
    英武阅读 2,639评论 0 52
  • 本文转自浅谈Python的with语句 引言 with 语句是从 Python 2.5 开始引入的一种与异常处理相...
    Syfun阅读 3,995评论 0 50
  • 引言 with 语句是从 Python 2.5 开始引入的一种与异常处理相关的功能(2.5 版本中要通过 from...
    氨基钠阅读 323评论 0 2
  • 一天的时间过得很快,这两天气温下降的确够冷,公司一待一天就过了。上午还是和几位新伙伴一起学用钉钉,然后看了公司领导...
    JiajiaYU阅读 112评论 0 0