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爬虫使用bs4方法实现数据解析

    Python爬虫使用bs4方法实现数据解析

    这篇文章主要介绍了Python爬虫使用bs4方法实现数据解析,文中通过示例代码介绍的非常详细,对大家的学习或者工作具有一定的参考学习价值,需要的朋友可以参考下
    2020-08-08
  • Python自动抢红包教程详解

    Python自动抢红包教程详解

    在本篇文章里小编给大家整理了关于Python自动抢红包的相关知识点内容,有兴趣的朋友们参考学习下。
    2019-06-06
  • OpenCV半小时掌握基本操作之边界填充

    OpenCV半小时掌握基本操作之边界填充

    这篇文章主要介绍了OpenCV基本操作之边界填充,本文给大家介绍的非常详细,对大家的学习或工作具有一定的参考借鉴价值,需要的朋友可以参考下
    2021-09-09
  • python实现四舍五入方式

    python实现四舍五入方式

    这篇文章主要介绍了python实现四舍五入方式,具有很好的参考价值,希望对大家有所帮助。如有错误或未考虑完全的地方,望不吝赐教
    2023-01-01
  • python的Template使用指南

    python的Template使用指南

    本文主要讲解了python中Template使用方法以及使用技巧,非常实用,有需要的朋友可以参考下
    2014-09-09
  • 使用Python实现LLM的模型迁移

    使用Python实现LLM的模型迁移

    在当今的人工智能领域,大型语言模型(LLM)如GPT、BERT等已经成为了研究和应用的热点,但其训练和部署成本高昂,且在不同领域或任务间的迁移能力有限,因此,如何有效地实现LLM的模型迁移,成为了一个重要的研究方向,本文将深入探讨如何使用Python实现LLM的模型迁
    2025-02-02
  • numpy求矩阵的特征值与特征向量(np.linalg.eig函数用法)

    numpy求矩阵的特征值与特征向量(np.linalg.eig函数用法)

    这篇文章主要介绍了numpy求矩阵的特征值与特征向量(np.linalg.eig函数用法),具有很好的参考价值,希望对大家有所帮助。如有错误或未考虑完全的地方,望不吝赐教
    2023-02-02
  • 通过Python快速获取Excel工作表名的方法示例

    通过Python快速获取Excel工作表名的方法示例

    在数据分析与办公自动化领域,通过Python处理Excel文件已成为必备技能,通过获取工作表名称,我们可以:快速了解文件结构、自动化处理多工作表数据、批量执行跨表格操作,本文将通过两个示例详解如何借助一个免费库通过Python读取Excel工作表名称,需要的朋友可以参考下
    2025-04-04
  • python高级特性简介

    python高级特性简介

    这篇文章主要介绍了python高级特性的相关内容,其中包括切片,迭代,列表生成式,生成器,迭代器,感兴趣的朋友可以了解下
    2020-08-08
  • flask route对协议作用及设计思路

    flask route对协议作用及设计思路

    这篇文章主要为大家介绍了flask route对协议作用及设计思路详解,有需要的朋友可以借鉴参考下,希望能够有所帮助,祝大家多多进步,早日升职加薪
    2022-07-07

最新评论