本文是对python中@contextmanager的简单总结。
在对于python中的装饰器，我们一般会使用它辅助方法。在我们学习的上下文管理器中，有一个@contextmanager装饰器，它能够帮助我们在函数中定义上下文管理器方法，实现代码方面的简化。下面我们就python中@contextmanager的概念、使用注意、实例进行分享。
1、概念
是contextlib模块提供的一个装饰器，用于将一个函数声明上下文管理，无需创建一个类或者单独的__enter__()方法和__exit__()方法，就可以实现上下文管理。
2、注意
装饰器@contextmanager真正的用法并不是简化上下文管理器，而是去定义上下文管理器的方法，是对原来不是上下文管理器的类变成了一个上下文管理器。
3、实例
| 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 | 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语句触发。
实例扩展：
实例一：
| 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 | from contextlib import contextmanager
class File():
def query(self):
print('查询文件')
@contextmanager
def open():
print('打开文件')
yield File()
print('关闭文件')
with open() as f:
f.query()
# 结果 打开文件 查询文件 关闭文件
|
实例二：
| 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 | 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实例用法的文章就介绍到这了，如有问题可以在评论区讨论。
本文内容由网友自发贡献，版权归原作者所有，本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容，请联系:hwhale#tublm.com(使用前将#替换为@)
