python中如何使用@contextmanage?

本文教程操作環境：windows7系統、Python3.9.1，DELLG3電腦。

1、實現邏輯

(1)初始化一個_GeneratorContextManager類，構造方法接受了一個生成器gen

(2)這個類實現了上下文管理器協議__enter__和__exit__

(3)執行with時會進入到__enter__方法，然后執行這個生成器，執行時會運行到with語法塊內的yield處

(4)__enter__返回yield的結果

(5)如果with語法塊沒有發生異常，with執行結束后，會進入到__exit__方法，再次執行生成器，這時會運行yield之后的代碼邏輯

(6)如果with語法塊發生了異常，__exit__會把這個異常通過生成器，傳入到with語法塊內，也就是把異常拋給調用者

2、使用實例

在Python中，可以使用裝飾器@contextmanager來簡化上下文管理器，也就是不需要定義__enter__和__exit__方法。

classMyContent:

defquery(self):

print('querydata')

#需要引入contextmanager

fromcontextlibimportcontextmanager

@contextmanager

defmake_resource():

print('firstconnecttoresource')

yieldMyContent()

print('closeresourceconnection')

#應用上下文管理器

withmake_resource()asr:

r.query()

以上就是python中使用@contextmanage的方法，大家在操作之前，一定要先就具體的@contextmanage實現邏輯進行了解，然后再展開相關的實例操作。更多Python學習教程請關注IT培訓機構:千鋒教育。