python中如何使用@contextmanage?

本文教程操作環(huán)境：windows7系統(tǒng)、Python3.9.1，DELLG3電腦。

1、實現(xiàn)邏輯

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

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

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

(4)__enter__返回yield的結(jié)果

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

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

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實現(xiàn)邏輯進行了解，然后再展開相關(guān)的實例操作。更多Python學習教程請關(guān)注IT培訓機構(gòu):千鋒教育。