python中@contextmanager装饰器的用法详解

 更新时间:2023年07月21日 09:24:45   作者:发条宇  
这篇文章主要介绍了python中@contextmanager装饰器的用法详解,@contextmanager 的作用就是我们可以把一个非自定义类改成一个上下文管理器,需要的朋友可以参考下

@contextmanager 的作用就是我们可以把一个非自定义类改成一个上下文管理器。

这个会经常遇到,所以有必要理解一下。

上下文管理器 是有 enter__和__exit ,那么我们可以为需要改写的普通类创建一个子类,在其中添加上__enter__和__exit__方法通过继承父类的方式实现上下文管理器。

但是通过contextlib模块下提供的@contextmanager装饰器,我们能够更方便的将一个普通类变成上下文管理器。

@contextmanager通过将一个函数变成生成器的方式来为普通类添加进入和退出时的处理代码,从而实现了将普通类变成一个上下文管理器。

还是拿文件来举个例子 # 结果 打开文件 查询文件 关闭文件

from contextlib import contextmanager
class File():
    def query(self):
        print('查询文件')
@contextmanager
def open():
    print('打开文件')
    yield File()
    print('关闭文件')
with open() as f:
    f.query()

执行的顺序:

①with语句调用open函数=>②执行open中yield之前的代码(打开文件)=>③执行yield语句中的代码(File())=>④执行with语句中的代码(f.query)=>⑤执行yiled语句后的代码(关闭文件)

注意:

  • 此时的with语句后是调用被修饰的函数,而不是实例化上下文管理器对象
  • with open() as f 中的f是yield语句中的内容,上述示例中即所创建的File对象
  • yield前、中、后的代码都可以省略不写

栗子2:

from contextlib import contextmanager
class DataBase():
    def query(self):
        print('写入操作')
    @contextmanager
    def open(self):
        try:
            yield
            self.commit()
        except Exception as e:
            self.rollback()
            raise e
db = DataBase()
with db.open():
    db.query()

重点理解执行顺序 ①with语句调用open函数=>②执行try异常判断=>③执行with语句中的写入操作=>④执行数据库的提交操作=>⑤如果出现异常执行数据库回滚操作

小结:

1.在思考过程中我们可以假设with语句中的内容整体移到了yield语句后,从而将with语句的执行简化为@contextmanager所修饰的函数的执行

2.当我们需要在一段代码的执行前后加上其他的处理代码时都可以使用@contextmanager来完成,并不需要一定是对上下文管理器的操作

到此这篇关于python中@contextmanager装饰器的用法详解的文章就介绍到这了,更多相关python的@contextmanager装饰器内容请搜索脚本之家以前的文章或继续浏览下面的相关文章希望大家以后多多支持脚本之家!

相关文章

  • Python入门之基础语法详解

    Python入门之基础语法详解

    这篇文章主要介绍了Python入门之基础语法详解,文中有非常详细的代码示例,对正在学习python的小伙伴们有很大的帮助哟,需要的朋友可以参考下
    2021-05-05
  • python+requests接口压力测试500次,查看响应时间的实例

    python+requests接口压力测试500次,查看响应时间的实例

    这篇文章主要介绍了python+requests接口压力测试500次,查看响应时间的实例,具有很好的参考价值,希望对大家有所帮助。一起跟随小编过来看看吧
    2020-04-04
  • Django rest framework工具包简单用法示例

    Django rest framework工具包简单用法示例

    这篇文章主要介绍了Django rest framework工具包简单用法,结合匿名访问控制的具体实例分析了Django rest framework工具包的注册、路由设置、视图、权限控制、配置等相关操作技巧,需要的朋友可以参考下
    2018-07-07
  • Python获取秒级时间戳与毫秒级时间戳的示例代码

    Python获取秒级时间戳与毫秒级时间戳的示例代码

    这篇文章主要介绍了Python获取秒级时间戳与毫秒级时间戳的示例代码,文中通过示例代码介绍的非常详细,对大家的学习或者工作具有一定的参考学习价值,需要的朋友们下面随着小编来一起学习学习吧
    2021-04-04
  • Python批量上传文件信息到服务器的实现示例

    Python批量上传文件信息到服务器的实现示例

    在进行软件测试的过程中,经常会需要准备一批数据,本文主要介绍了Python批量上传文件信息到服务器的实现示例,具有一定的参考价值,感兴趣的可以了解一下
    2023-12-12
  • 打包PyQt5应用时的注意事项

    打包PyQt5应用时的注意事项

    这篇文章主要介绍了打包PyQt5应用时的注意事项的相关资料,需要的朋友可以参考下
    2020-02-02
  • Python定义二叉树及4种遍历方法实例详解

    Python定义二叉树及4种遍历方法实例详解

    这篇文章主要介绍了Python定义二叉树及4种遍历方法,结合实例形式较为详细的分析了二叉树的概念、原理,以及Python定义与遍历二叉树相关操作技巧,需要的朋友可以参考下
    2018-07-07
  • Python中ArcPy栅格裁剪栅格(批量对齐栅格图像范围并统一行数与列数)

    Python中ArcPy栅格裁剪栅格(批量对齐栅格图像范围并统一行数与列数)

    本文介绍基于Python中ArcPy模块,实现基于栅格图像批量裁剪栅格图像,同时对齐各个栅格图像的空间范围,统一其各自行数与列数的方法,本文通过实例代码给大家介绍的非常详细,需要的朋友可以参考下
    2023-02-02
  • Python3指定路径寻找符合匹配模式文件

    Python3指定路径寻找符合匹配模式文件

    这篇文章主要介绍了Python3指定路径寻找符合匹配模式文件,涉及Python文件查找与模式匹配的相关技巧,需要的朋友可以参考下
    2015-05-05
  • Pytorch从0实现Transformer的实践

    Pytorch从0实现Transformer的实践

    本文主要介绍了Pytorch从0实现Transformer的实践,文中通过示例代码介绍的非常详细,对大家的学习或者工作具有一定的参考学习价值,需要的朋友们下面随着小编来一起学习学习吧
    2022-05-05

最新评论