什么是上下文管理器？

基本語法

with open('test.txt') as f:print f.readlines() 1. 上下文表達式：with open('test.txt') as f: 2. 上下文管理器：open('test.txt') 3. f 不是上下文管理器，應(yīng)該是資源對象。

如何寫上下文管理器

要自己實現(xiàn)這樣一個上下文管理，要先知道上下文管理協(xié)議。

簡單點說，就是在一個類里，實現(xiàn)了__enter__和__exit__的方法，這個類的實例就是一個上下文管理器。

例如這個示例：

class Resource():def __enter__(self):print('===connect to resource===')return selfdef __exit__(self, exc_type, exc_val, exc_tb):print('===close resource connection===')def operate(self):print('===in operation===')with Resource() as res:res.operate() 輸出結(jié)果 ===connect to resource=== ===in operation=== ===close resource connection===

為什么你要用上下文管理器

學(xué)習(xí)時多問自己幾個為什么，養(yǎng)成對一些細節(jié)的思考，有助于加深對知識點的理解。

為什么要使用上下文管理器？

在我看來，這和 Python 崇尚的優(yōu)雅風(fēng)格有關(guān)。

可以以一種更加優(yōu)雅的方式，操作（創(chuàng)建/獲取/釋放）資源，如文件操作、數(shù)據(jù)庫連接；
可以以一種更加優(yōu)雅的方式，處理異常；
第一種，我們上面已經(jīng)以資源的連接為例講過了。

而第二種，會被大多數(shù)人所忽略。這里會重點講一下。

大家都知道，處理異常，通常都是使用 try…execept… 來捕獲處理的。這樣做一個不好的地方是，在代碼的主邏輯里，會有大量的異常處理代理，這會很大的影響我們的可讀性。

好一點的做法呢，可以使用 with 將異常的處理隱藏起來。

仍然是以上面的代碼為例，我們將1/0 這個一定會拋出異常的代碼寫在 operate 里

class Resource():def __enter__(self):print('===connect to resource===')return selfdef __exit__(self, exc_type, exc_val, exc_tb):print('===close resource connection===')return Truedef operate(self):1/0with Resource() as res:res.operate()

運行一下，驚奇地發(fā)現(xiàn)，居然不會報錯。

這就是上下文管理協(xié)議的一個強大之處，異?？梢栽赺_exit__ 進行捕獲并由你自己決定如何處理，是拋出呢還是在這里就解決了。在__exit__ 里返回 True（沒有return 就默認為 return False），就相當于告訴 Python解釋器，這個異常我們已經(jīng)捕獲了，不需要再往外拋了。

在 寫__exit__ 函數(shù)時，需要注意的事，它必須要有這三個參數(shù)：

exc_type：異常類型
exc_val：異常值
exc_tb：異常的錯誤棧信息
當主邏輯代碼沒有報異常時，這三個參數(shù)將都為None。

理解并使用 contextlib

在上面的例子中，我們只是為了構(gòu)建一個上下文管理器，卻寫了一個類。如果只是要實現(xiàn)一個簡單的功能，寫一個類未免有點過于繁雜。這時候，我們就想，如果只寫一個函數(shù)就可以實現(xiàn)上下文管理器就好了。

這個點Python早就想到了。它給我們提供了一個裝飾器，你只要按照它的代碼協(xié)議來實現(xiàn)函數(shù)內(nèi)容，就可以將這個函數(shù)對象變成一個上下文管理器。

我們按照 contextlib 的協(xié)議來自己實現(xiàn)一個打開文件（with open）的上下文管理器。

import contextlib@contextlib.contextmanager def open_func(file_name):# __enter__方法print('open file:', file_name, 'in __enter__')file_handler = open(file_name, 'r')# 【重點】：yieldyield file_handler# __exit__方法print('close file:', file_name, 'in __exit__')file_handler.close()returnwith open_func('/Users/MING/mytest.txt') as file_in:for line in file_in:print(line)

在被裝飾函數(shù)里，必須是一個生成器（帶有yield），而yield之前的代碼，就相當于__enter__里的內(nèi)容。yield 之后的代碼，就相當于__exit__ 里的內(nèi)容。

上面這段代碼只能實現(xiàn)上下文管理器的第一個目的（管理資源），并不能實現(xiàn)第二個目的（處理異常）。

如果要處理異常，可以改成下面這個樣子。

import contextlib@contextlib.contextmanager def open_func(file_name):# __enter__方法print('open file:', file_name, 'in __enter__')file_handler = open(file_name, 'r')try:yield file_handlerexcept Exception as exc:# deal with exceptionprint('the exception was thrown')finally:print('close file:', file_name, 'in __exit__')file_handler.close()returnwith open_func('/Users/MING/mytest.txt') as file_in:for line in file_in:1/0print(line)

總結(jié)起來，使用上下文管理器有三個好處：

提高代碼的復(fù)用率；
提高代碼的優(yōu)雅度；
提高代碼的可讀性；

總結(jié)

以上是生活随笔為你收集整理的python上下文管理器的全部內(nèi)容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網(wǎng)站內(nèi)容還不錯，歡迎將生活随笔推薦給好友。