python中@contextmanager裝飾器的用法詳解

更新時間：2023年07月21日 09:24:45 作者：發(fā)條宇

這篇文章主要介紹了python中@contextmanager裝飾器的用法詳解,@contextmanager 的作用就是我們可以把一個非自定義類改成一個上下文管理器,需要的朋友可以參考下

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

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

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

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

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

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

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

執(zhí)行的順序：

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

注意：

此時的with語句后是調(diào)用被修飾的函數(shù)，而不是實例化上下文管理器對象
with open() as f 中的f是yield語句中的內(nèi)容，上述示例中即所創(chuàng)建的File對象
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()

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

小結(jié)：

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

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

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

您可能感興趣的文章: