在對于python中的裝飾器，我們一般會使用它輔助方法。在我們學(xué)習(xí)的上下文管理器中，有一個@contextmanager裝飾器，它能夠幫助我們在函數(shù)中定義上下文管理器方法，實(shí)現(xiàn)代碼方面的簡化。下面我們就python中@contextmanager的概念、使用注意、實(shí)例進(jìn)行分享。
1、概念
是contextlib模塊提供的一個裝飾器，用于將一個函數(shù)聲明上下文管理，無需創(chuàng)建一個類或者單獨(dú)的__enter__()方法和__exit__()方法，就可以實(shí)現(xiàn)上下文管理。
2、注意
裝飾器@contextmanager真正的用法并不是簡化上下文管理器，而是去定義上下文管理器的方法，是對原來不是上下文管理器的類變成了一個上下文管理器。
3、實(shí)例
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
看上面這個例子，函數(shù)中yield之前的所有代碼都類似于上下文管理器中__enter__方法的內(nèi)容。而yield之后的所有代碼都如__exit__方法的內(nèi)容。如果執(zhí)行過程中發(fā)生了異常，則會在yield語句觸發(fā)。
實(shí)例擴(kuò)展：
實(shí)例一：
from contextlib import contextmanager
class File():
def query(self):
print('查詢文件')
@contextmanager
def open():
print('打開文件')
yield File()
print('關(guān)閉文件')
with open() as f:
f.query()
# 結(jié)果 打開文件 查詢文件 關(guān)閉文件
實(shí)例二：
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()
到此這篇關(guān)于python中@contextmanager實(shí)例用法的文章就介紹到這了,更多相關(guān)python中@contextmanager是什么內(nèi)容請搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家！
您可能感興趣的文章:- Python with關(guān)鍵字，上下文管理器，@contextmanager文件操作示例
