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