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

 更新时间:2022年06月14日 08:31:31   作者:springsnow  
这篇文章介绍了python内置模块之上下文管理contextlib,文中通过示例代码介绍的非常详细。对大家的学习或工作具有一定的参考借鉴价值,需要的朋友可以参考下

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中顺序结构和循环结构示例代码

    这篇文章主要介绍了Python中的条件语句和循环语句,条件语句用于根据条件执行不同的代码块,循环语句用于重复执行一段代码,文章还详细说明了range函数的使用和continue与break语句的作用,需要的朋友可以参考下
    2025-02-02
  • 教你使用Python的pygame模块实现拼图游戏

    教你使用Python的pygame模块实现拼图游戏

    pygame模块是一个可以跨平台的模块,其设计目的就是为电子游戏而设计,能够支持图片和声音,下面这篇文章主要给给大家介绍了关于使用Python的pygame模块实现拼图游戏的相关资料,需要的朋友可以参考下
    2022-07-07
  • Python实现随机生成图片验证码详解

    Python实现随机生成图片验证码详解

    这篇文章主要介绍了如何利用Python生成随机的图片验证码 并打印验证码的值,文中的示例代码简洁易懂,感兴趣的小伙伴可以跟随小编一起试试
    2022-01-01
  • Python正则表达式的应用详解

    Python正则表达式的应用详解

    这篇文章主要介绍了Python中正则表达式的详细教程,正则表达式是Python学习进阶当中的重要内容,需要的朋友可以参考下,希望能给你带来帮助
    2021-08-08
  • python项目127.0.0.1:5000访问失败问题解决

    python项目127.0.0.1:5000访问失败问题解决

    Windows环境下启动python项目,接口访问失败,本文给大家分享python项目127.0.0.1:5000访问失败问题解决方法,感兴趣的朋友跟随小编一起看看吧
    2023-09-09
  • 火车票抢票python代码公开揭秘!

    火车票抢票python代码公开揭秘!

    火车票抢票python代码公开揭秘!使用Python模拟抢票程序,给大家揭秘抢票到底是怎么回事,感兴趣的小伙伴们可以参考一下
    2018-03-03
  • python实现在图片上画特定大小角度矩形框

    python实现在图片上画特定大小角度矩形框

    这篇文章主要为大家详细介绍了python实现在图片上画特定大小角度矩形框,以及C++的实现方法,具有一定的参考价值,感兴趣的小伙伴们可以参考一下
    2018-10-10
  • 使用opencv中匹配点对的坐标提取方式

    使用opencv中匹配点对的坐标提取方式

    这篇文章主要介绍了使用opencv中匹配点对的坐标提取方式,具有很好的参考价值,希望对大家有所帮助。一起跟随小编过来看看吧
    2020-06-06
  • python使用PIL缩放网络图片并保存的方法

    python使用PIL缩放网络图片并保存的方法

    这篇文章主要介绍了python使用PIL缩放网络图片并保存的方法,涉及Python操作网络图片的相关技巧,非常具有实用价值,需要的朋友可以参考下
    2015-04-04
  • python opencv读mp4视频的实例

    python opencv读mp4视频的实例

    今天小编就为大家分享一篇python opencv读mp4视频的实例,具有很好的参考价值,希望对大家有所帮助。一起跟随小编过来看看吧
    2018-12-12

最新评论