python内置模块之上下文管理contextlib


Posted in Python onJune 14, 2022

Python中当我们们打开文本时,通常会是用with语句,with语句允许我们非常方便的使用资源,而不必担心资源没有关闭。

with open('/path/filename', 'r') as f:
    f.read()

然而,并不是只有open()函数返回fp对象才能使用 with 语句。实际上,任何对象,只要正确实现上下文管理,就可以使用with语句。

实现上下文管理是通过 __enter__ 和 __exit__ 这两个方法实现的。例如,下面的class实现了这两个方法:

class Query(object):
 
    def __init__(self, name):
        self.name = name
 
    def __enter__(self):
        print('Begin')
        return self
 
    def __exit__(self, exc_type, exc_value, traceback):
        if exc_type:
            print('Error')
        else:
            print('End')
 
    def query(self):
        print('Query info about %s...' % self.name)

这样我们可以把自己写的资源对象用于 with 语句。

with Query('Bob') as q:
    q.query()

一、@contextmanager

编写 __enter__ 和 __exit__ 仍然很繁琐,因此Python的标准库 contextlib 提供了更简单的写法,上面的代码可以改写为:

from contextlib import contextmanager
 
class Query(object):
 
    def __init__(self, name):
        self.name = name
 
    def query(self):
        print('Query info about %s...' % self.name)
 
@contextmanager
def create_query(name):
    print('Begin')
    q = Query(name)
    yield q
    print('End')

@contextmanager 这个装饰器接受一个 generator,用 yield 语句把 with ... as var 把变量输出去,然后,with 语句就可以正常的工作了:

with create_query('Bob') as q:
    q.query()

很多时候,我们希望在某段代码执行前后自动执行特定代码,也可以用 @contextmanager实现。

@contextmanager
def tag(name):
    print("<%s>" % name)
    yield
    print("" % name)
 
with tag("h1"):
    print("hello")
    print("world")

上述代码执行结果:

hello
world
</h1>

代码的执行顺序是:

  • with 语句 首先执行 yield 之前的语句,因此打印出.
  • yield 调用会执行 with 语句内部的所有语句,因此打印出 hello 和 world.
  • 最后执行yield之后的语句,打印出.

二、@closing

如果一个对象没有实现上下文,就不能使用 with 语句,但是可以用 closing() 来把对象变为上下文对象。

from contextlib import closing
from urllib.request import urlopen
 
with closing(urlopen('https://www.python.org')) as page:
    for line in page:
        print(line)

closing 也是一个经过 @contextmanager 装饰的generator

@contextmanager
def closing(thing):
    try:
        yield thing
    finally:
        thing.close()

它的作用就是把任意对象变为上下文对象,并支持 with语句。

到此这篇关于python内置模块之上下文管理contextlib的文章就介绍到这了。希望对大家的学习有所帮助,也希望大家多多支持三水点靠木。

Python 相关文章推荐
Python输出各行命令详解
Feb 01 Python
详解Python使用tensorflow入门指南
Feb 09 Python
Python学习小技巧总结
Jun 10 Python
python实现简易数码时钟
Feb 19 Python
python+opencv实现霍夫变换检测直线
Oct 23 Python
python图形用户接口实例详解
Dec 16 Python
Python callable内置函数原理解析
Mar 05 Python
python新手学习可变和不可变对象
Jun 11 Python
python如何求圆的面积
Jul 01 Python
基于Python+QT的gui程序开发实现
Jul 03 Python
Python 执行矩阵与线性代数运算
Aug 01 Python
PyQt 如何创建自定义QWidget
Mar 24 Python
Python时间操作之pytz模块使用详解
Django框架之路由用法
Jun 10 #Python
深入理解pytorch库的dockerfile
Jun 10 #Python
如何利用python实现列表嵌套字典取值
Jun 10 #Python
python中filter,map,reduce的作用
Jun 10 #Python
Django框架中模型的用法
Jun 10 #Python
Django框架中视图的用法
Jun 10 #Python
You might like
40年前的这部特摄片恐龙特级克塞号80后的共同回忆
2020/03/08 日漫
比较好用的PHP防注入漏洞过滤函数代码
2012/04/11 PHP
PHP根据图片色界在不同位置加水印的方法
2015/07/01 PHP
使用Composer安装Yii框架的方法
2016/03/15 PHP
分享php多功能图片处理类
2016/05/15 PHP
基于ThinkPHP5框架使用QueryList爬取并存入mysql数据库操作示例
2019/05/25 PHP
Javascript下的keyCode键码值表
2007/04/10 Javascript
js实现缓冲运动效果的方法
2015/04/10 Javascript
jQuery实现页面顶部显示的进度条效果完整实例
2015/12/09 Javascript
JS本地刷新返回上一页代码
2016/07/25 Javascript
bootstrap fileinput实现文件上传功能
2017/08/23 Javascript
Vue中使用 setTimeout() setInterval()函数的问题
2018/09/13 Javascript
解决vue2 在mounted函数无法获取prop中的变量问题
2018/11/15 Javascript
微信小程序图表插件wx-charts用法实例详解
2019/05/20 Javascript
Layui实现数据表格默认全部显示(不要分页)
2019/10/26 Javascript
js get和post请求实现代码解析
2020/02/06 Javascript
Vue自动构建发布脚本的方法示例
2020/07/24 Javascript
详解Python的Django框架中Manager方法的使用
2015/07/21 Python
Python实现返回数组中第i小元素的方法示例
2017/12/04 Python
python程序快速缩进多行代码方法总结
2019/06/23 Python
python区分不同数据类型的方法
2019/10/14 Python
Python绘图实现显示中文
2019/12/04 Python
Python 项目转化为so文件实例
2019/12/23 Python
python自动下载图片的方法示例
2020/03/25 Python
python 字典item与iteritems的区别详解
2020/04/25 Python
Pygame的程序开始示例代码
2020/05/07 Python
python使用多线程+socket实现端口扫描
2020/05/28 Python
Python爬取股票信息,并可视化数据的示例
2020/09/26 Python
在canvas上实现元素图片镜像翻转动画效果的方法
2018/03/20 HTML / CSS
Canvas在超级玛丽游戏中的应用详解
2021/02/06 HTML / CSS
NBA欧洲商店(法国):NBA Europe Store FR
2016/10/19 全球购物
美国知名珠宝首饰品牌:Gemvara
2017/10/06 全球购物
HashMap和Hashtable的区别
2013/05/18 面试题
学校火灾防控方案
2014/06/09 职场文书
英语系毕业生求职信
2014/07/13 职场文书
Redis中key的过期删除策略和内存淘汰机制
2022/04/12 Redis