在對于python中的裝飾器，我們一般會使用它輔助方法。在我們學(xué)習(xí)的上下文管理器中，有一個@contextmanager裝飾器，它能夠幫助我們在函數(shù)中定義上下文管理器方法，實(shí)現(xiàn)代碼方面的簡化。下面我們就python中@contextmanager的概念、使用注意、實(shí)例進(jìn)行分享。

是contextlib模塊提供的一個裝飾器，用于將一個函數(shù)聲明上下文管理，無需創(chuàng)建一個類或者單獨(dú)的__enter__()方法和__exit__()方法，就可以實(shí)現(xiàn)上下文管理。

裝飾器@contextmanager真正的用法并不是簡化上下文管理器，而是去定義上下文管理器的方法，是對原來不是上下文管理器的類變成了一個上下文管理器。

from contextlib import contextmanagerimport time@contextmanagerdef 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 contextmanagerclass File(): def query(self): print(’查詢文件’)@contextmanagerdef open(): print(’打開文件’) yield File() print(’關(guān)閉文件’)with open() as f: f.query() # 結(jié)果 打開文件 查詢文件 關(guān)閉文件

實(shí)例二：

from contextlib import contextmanagerclass DataBase(): def query(self): print(’寫入操作’) @contextmanager def open(self): try: yield self.commit() except Exception as e: self.rollback() raise edb = DataBase()with db.open(): db.query()

到此這篇關(guān)于python中@contextmanager實(shí)例用法的文章就介紹到這了,更多相關(guān)python中@contextmanager是什么內(nèi)容請搜索好吧啦網(wǎng)以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持好吧啦網(wǎng)！