python中@contextmanager的使用示例

这篇文章给大家分享的是有关python中@contextmanager的使用示例的内容。小编觉得挺实用的,因此分享给大家做个参考,一起跟随小编过来看看吧。

在对于python中的装饰器,我们一般会使用它辅助方法。在我们学习的上下文管理器中,有一个@contextmanager装饰器,它能够帮助我们在函数中定义上下文管理器方法,实现代码方面的简化。下面我们就python中@contextmanager的概念、使用注意、实例进行分享。

1、概念

是contextlib模块提供的一个装饰器,用于将一个函数声明上下文管理,无需创建一个类或者单独的__enter__()方法和__exit__()方法,就可以实现上下文管理。

2、注意

装饰器@contextmanager真正的用法并不是简化上下文管理器,而是去定义上下文管理器的方法,是对原来不是上下文管理器的类变成了一个上下文管理器。

3、实例from contextlib import contextmanager import time @contextmanager def demo(label):   start = time.time()   try:     yield   finally:     end = time.time()     print('{}: {}'.format(label, end - start)) with demo('counting'):   n = 10000000   while n > 0:     n -= 1 # counting: 1.32399988174

看上面这个例子,函数中yield之前的所有代码都类似于上下文管理器中__enter__方法的内容。而yield之后的所有代码都如__exit__方法的内容。如果执行过程中发生了异常,则会在yield语句触发。

实例扩展:

实例一:

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

实例二:

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()

感谢各位的阅读!关于“python中@contextmanager的使用示例”这篇文章就分享到这里了,希望以上内容可以对大家有一定的帮助,让大家可以学到更多知识,如果觉得文章不错,可以把它分享出去让更多的人看到吧!

郑重声明:本文版权归原作者所有,转载文章仅为传播更多信息之目的,如作者信息标记有误,请第一时间联系我们修改或删除,多谢。