python中如何使用@contextmanage?

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

1、實(shí)現(xiàn)邏輯

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

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

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

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

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

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

2、使用實(shí)例

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

classMyContent:

defquery(self):

print('querydata')

#需要引入contextmanager

fromcontextlibimportcontextmanager

@contextmanager

defmake_resource():

print('firstconnecttoresource')

yieldMyContent()

print('closeresourceconnection')

#應(yīng)用上下文管理器

withmake_resource()asr:

r.query()

以上就是python中使用@contextmanage的方法，大家在操作之前，一定要先就具體的@contextmanage實(shí)現(xiàn)邏輯進(jìn)行了解，然后再展開相關(guān)的實(shí)例操作。更多Python學(xué)習(xí)教程請(qǐng)關(guān)注IT培訓(xùn)機(jī)構(gòu):千鋒教育。