首先，我們來(lái)看一下with的語(yǔ)法格式：

withcontext_expression[astarget(s)]:with-body

with語(yǔ)法非常簡(jiǎn)單，我們只需要with一個(gè)表達(dá)式，然后就可以執(zhí)行自定義的業(yè)務(wù)邏輯。

但是，with后面的表達(dá)式是可以任意寫(xiě)的嗎?

答案是否定的。要想使用with語(yǔ)法塊，with后面的的對(duì)象需要實(shí)現(xiàn)「上下文管理器協(xié)議」。

什么是「上下文管理器協(xié)議」?

一個(gè)類(lèi)在Python中，只要實(shí)現(xiàn)以下方法，就實(shí)現(xiàn)了「上下文管理器協(xié)議」：

__enter__：在進(jìn)入with語(yǔ)法塊之前調(diào)用，返回值會(huì)賦值給with的target

__exit__：在退出with語(yǔ)法塊時(shí)調(diào)用，一般用作異常處理

我們來(lái)看實(shí)現(xiàn)了這2個(gè)方法的例子：

classTestContext:def__enter__(self):print('__enter__')return1def__exit__(self,exc_type,exc_value,exc_tb):print('exc_type:%s'%exc_type)print('exc_value:%s'%exc_value)print('exc_tb:%s'%exc_tb)withTestContext()ast:print('t:%s'%t)#Output:#__enter__#t:1#exc_type:None#exc_value:None#exc_tb:None

在這個(gè)例子中，我們定義了TestContext類(lèi)，它分別實(shí)現(xiàn)了__enter__和exit方法。

這樣一來(lái)，我們就可以把TestContext當(dāng)做一個(gè)「上下文管理器」來(lái)使用，也就是通過(guò)withTestContext()ast方式來(lái)執(zhí)行。

從輸出結(jié)果我們可以看到，具體的執(zhí)行流程如下：

__enter__在進(jìn)入with語(yǔ)句塊之前被調(diào)用，這個(gè)方法的返回值賦給了with后的t變量

__exit__在執(zhí)行完with語(yǔ)句塊之后被調(diào)用

如果在with語(yǔ)句塊內(nèi)發(fā)生了異常，那么__exit__方法可以拿到關(guān)于異常的詳細(xì)信息：

exc_type：異常類(lèi)型

exc_value：異常對(duì)象

exc_tb：異常堆棧信息

我們來(lái)看一個(gè)發(fā)生異常的例子，觀(guān)察__exit__方法拿到的異常信息是怎樣的：

withTestContext()ast:#這里會(huì)發(fā)生異常a=1/0print('t:%s'%t)#Output:#__enter__#exc_type:#exc_value:integerdivisionormodulobyzero#exc_tb:#Traceback(mostrecentcalllast):#File"base.py",line16,in#a=1/0#ZeroDivisionError:integerdivisionormodulobyzero

從輸出結(jié)果我們可以看到，當(dāng)with語(yǔ)法塊內(nèi)發(fā)生異常后，__exit__輸出了這個(gè)異常的詳細(xì)信息，其中包括異常類(lèi)型、異常對(duì)象、異常堆棧。

如果我們需要對(duì)異常做特殊處理，就可以在這個(gè)方法中實(shí)現(xiàn)自定義邏輯。

回到最開(kāi)始我們講的，使用with讀取文件的例子。之所以with能夠自動(dòng)關(guān)閉文件資源，就是因?yàn)閮?nèi)置的文件對(duì)象實(shí)現(xiàn)了「上下文管理器協(xié)議」，這個(gè)文件對(duì)象的__enter__方法返回了文件句柄，并且在__exit__中實(shí)現(xiàn)了文件資源的關(guān)閉，另外，當(dāng)with語(yǔ)法塊內(nèi)有異常發(fā)生時(shí)，會(huì)拋出異常給調(diào)用者。

偽代碼可以這么寫(xiě)：

classFile:def__enter__(self):returnfile_objdef__exit__(self,exc_type,exc_value,exc_tb):#with退出時(shí)釋放文件資源file_obj.close()#如果with內(nèi)有異常發(fā)生拋出異常ifexc_typeisnotNone:raiseexception

這里我們小結(jié)一下，通過(guò)對(duì)with的學(xué)習(xí)，我們了解到，with非常適合用需要對(duì)于上下文處理的場(chǎng)景，例如操作文件、Socket，這些場(chǎng)景都需要在執(zhí)行完業(yè)務(wù)邏輯后，釋放資源。

以上內(nèi)容為大家介紹了python的上下文管理器，希望對(duì)大家有所幫助，如果想要了解更多Python相關(guān)知識(shí)，請(qǐng)關(guān)注IT培訓(xùn)機(jī)構(gòu):千鋒教育。http://m.hangjiakeji.com/