网络编程
位置:首页>> 网络编程>> Python编程>> python中@contextmanager实例用法

python中@contextmanager实例用法

作者:小妮浅浅  发布时间:2021-07-18 18:19:11 

标签: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()

来源:https://www.py.cn/jishu/jichu/23704.html

0
投稿

猜你喜欢

手机版 网络编程 asp之家 www.aspxhome.com