python的上下文管理器 1

reference:

來看看如何正確關閉乙個檔案。

普通版：

def m1():
f = open("output.txt", "w")
f.write("python之禪")
f.close()

這樣寫有乙個潛在的問題，如果在呼叫 write 的過程中，出現了異常進而導致後續**無法繼續執行，close 方法無法被正常呼叫，因此資源就會一直被該程式占用而無法被釋放。那麼該如何改進**呢？

高階版：

def m2():
f = open("output.txt", "w")
try:
f.write("python之禪")
except ioerror:
print("oops error")
finally:
f.close()

改良版本的程式是對可能發生異常的**處進行 try 捕獲，使用 try/finally 語句，該語句表示如果在 try **塊中程式出現了異常，後續**就不再執行，而直接跳轉到 except **塊。而無論如何，finally 塊的**最終都會被執行。因此，只要把 close 放在 finally **中，檔案就一定會關閉。

高階版：

def m3():
with open("output.txt", "w") as f:
f.write("python之禪")

一種更加簡潔、優雅的方式就是用 with 關鍵字。open 方法的返回值賦值給變數 f，當離開 with **塊的時候，系統會自動呼叫 f.close() 方法， with 的作用和使用 try/finally 語句是一樣的。那麼它的實現原理是什麼？在講 with 的原理前要涉及到另外乙個概念，就是上下文管理器（context manager）。

上下文管理器

任何實現了 __enter__() 和 __exit__() 方法的物件都可稱之為上下文管理器，上下文管理器物件可以使用 with 關鍵字。顯然，檔案（file）物件也實現了上下文管理器。

那麼檔案物件是如何實現這兩個方法的呢？我們可以模擬實現乙個自己的檔案類，讓該類實現 __enter__() 和 __exit__() 方法。

class
file():
def__init__
(self, filename, mode):
self.filename =filename
self.mode =mode
def__enter__
(self):
print("
entering")
self.f =open(self.filename, self.mode)
return
self.f
def__exit__(self, *args):
print("
will exit")
self.f.close()

view code

__enter__() 方法返回資源物件，這裡就是你將要開啟的那個檔案物件，__exit__() 方法處理一些清除工作。

因為 file 類實現了上下文管理器，現在就可以使用 with 語句了。

class
file():
def__init__
(self, filename, mode):
self.filename =filename
self.mode =mode
def__enter__
(self):
print("
entering")
self.f =open(self.filename, self.mode)
return self.f #
這個返回值賦值給了as 後面的變數f
def__exit__(self, *args):
print("
will exit")
self.f.close()
with file(
'out.txt
', 'w'
) as f:
print("
writing")
f.write(
'hello, python')
'''執行結果：
entering
writing
will exit
'''

view code

這樣，你就無需顯示地呼叫 close 方法了，由系統自動去呼叫，哪怕中間遇到異常 close 方法也會被呼叫。

python的上下文管理器 1

python 上下文管理器

python上下文管理器

Python 上下文管理器

python的上下文管理器 1

python 上下文管理器

python上下文管理器

Python 上下文管理器

相關推薦