@contextmanager 的作用就是我們可以把一個(gè)非自定義類改成一個(gè)上下文管理器。

這個(gè)會(huì)經(jīng)常遇到，所以有必要理解一下。

上下文管理器 是有 enter__和__exit ，那么我們可以為需要改寫的普通類創(chuàng)建一個(gè)子類，在其中添加上__enter__和__exit__方法通過(guò)繼承父類的方式實(shí)現(xiàn)上下文管理器。

但是通過(guò)contextlib模塊下提供的@contextmanager裝飾器，我們能夠更方便的將一個(gè)普通類變成上下文管理器。

@contextmanager通過(guò)將一個(gè)函數(shù)變成生成器的方式來(lái)為普通類添加進(jìn)入和退出時(shí)的處理代碼，從而實(shí)現(xiàn)了將普通類變成一個(gè)上下文管理器。

還是拿文件來(lái)舉個(gè)例子 # 結(jié)果 打開(kāi)文件 查詢文件 關(guān)閉文件

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()

執(zhí)行的順序：

①with語(yǔ)句調(diào)用open函數(shù)=>②執(zhí)行open中yield之前的代碼(打開(kāi)文件)=>③執(zhí)行yield語(yǔ)句中的代碼(File())=>④執(zhí)行with語(yǔ)句中的代碼(f.query)=>⑤執(zhí)行yiled語(yǔ)句后的代碼(關(guān)閉文件)

注意：

• 此時(shí)的with語(yǔ)句后是調(diào)用被修飾的函數(shù)，而不是實(shí)例化上下文管理器對(duì)象
• with open() as f 中的f是yield語(yǔ)句中的內(nèi)容，上述示例中即所創(chuàng)建的File對(duì)象
• yield前、中、后的代碼都可以省略不寫

栗子2：

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()

重點(diǎn)理解執(zhí)行順序 ①with語(yǔ)句調(diào)用open函數(shù)=>②執(zhí)行try異常判斷=>③執(zhí)行with語(yǔ)句中的寫入操作=>④執(zhí)行數(shù)據(jù)庫(kù)的提交操作=>⑤如果出現(xiàn)異常執(zhí)行數(shù)據(jù)庫(kù)回滾操作

小結(jié)：

1.在思考過(guò)程中我們可以假設(shè)with語(yǔ)句中的內(nèi)容整體移到了yield語(yǔ)句后，從而將with語(yǔ)句的執(zhí)行簡(jiǎn)化為@contextmanager所修飾的函數(shù)的執(zhí)行

2.當(dāng)我們需要在一段代碼的執(zhí)行前后加上其他的處理代碼時(shí)都可以使用@contextmanager來(lái)完成，并不需要一定是對(duì)上下文管理器的操作

到此這篇關(guān)于python中@contextmanager裝飾器的用法詳解的文章就介紹到這了,更多相關(guān)python的@contextmanager裝飾器內(nèi)容請(qǐng)搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家！

最新評(píng)論