上下文管理器可以在不需要檔案的時候,自動關閉檔案。
下面我們看一下兩段程式:
**如下:
# without context manager
f = open("new.txt", "w")
print(f.closed) # whether the file is open
f.write("hello world!")
f.close()
print(f.closed)
**如下:
# with context manager
with open("new.txt", "w") as f:
print(f.closed)
f.write("hello world!")
print(f.closed)
兩段程式實際上執行的是相同的操作。第二段程式就使用了上下文管理器 (with…as…)。上下文管理器有隸屬於它的程式塊。當隸屬的程式塊執行結束的時候(也就是不再縮排),上下文管理器自動關閉了檔案 (我們通過f.closed來查詢檔案是否關閉)。我們相當於使用縮排規定了檔案物件f的使用範圍。
上面的上下文管理器基於f物件的exit()特殊方法。當我們使用上下文管理器的語法時,我們實際上要求python在進入程式塊之前呼叫物件的enter()方法,在結束程式塊的時候呼叫exit()方法。對於檔案物件f來說,它定義了enter()和exit()方法(可以通過dir(f)看到)。在f的exit()方法中,有self.close()語句。所以在使用上下文管理器時,我們就不用明文關閉f檔案了。
任何定義了enter()和exit()方法的物件都可以用於上下文管理器。檔案物件f是內建物件,所以f自動帶有這兩個特殊方法,不需要自定義。
下面,我們自定義用於上下文管理器的物件:
**如下:
# customized object
class
vow(object):
def__init__
(self, text):
self.text = text
def__enter__
(self):
self.text = "i say: " + self.text # add prefix
return self # note: return an object
def__exit__
(self,exc_type,exc_value,traceback):
self.text = self.text + "!"
# add suffix
with vow("i'm fine") as myvow:
print(myvow.text)
print(myvow.text)
我們的執行結果如下:
**如下:
i say: i'm fine
i say: i'm fine!
我們可以看到,在進入上下文和離開上下文時,物件的text屬性發生了改變(最初的text屬性是」i』m fine」)。
enter()返回乙個物件。上下文管理器會使用這一物件作為as所指的變數,也就是myvow。在enter()中,我們為myvow.text增加了字首 (「i say: 「)。在exit()中,我們為myvow.text增加了字尾(「!」)。
注意:exit()中有四個引數。當程式塊中出現異常(exception),exit()的引數中exc_type, exc_value, traceback用於描述異常。我們可以根據這三個引數進行相應的處理。如果正常執行結束,這三個引數都是none。在我們的程式中,我們並沒有用到這一特性。
通過上下文管理器,我們控制物件在程式不同區間的特性。上下文管理器(with expr as var)大致相當於如下流程:
**如下:
# with expr as
var:
var = expr
var = var.__enter__()
try:
block
finally:
var.__exit__()
Python 學習筆記 上下文
python裡面有個特殊的模組叫做上下文模組,可以和with搭配來實現一些功能。他的執行方式是使用者需要乙個定義乙個生成器的函式 with後面跟這個函式,他會執行yield之前的 然後跳出來,執行with語句下面的 然後再切換回函式,執行finally後面的 比如import contextlib ...
Python筆記 上下文管理器
with 語句後面的結果物件,需要重寫 enter 和 exit 方法 當進入到 with 塊時,會自動呼叫 enter 方法裡的 當 with 塊執行完成以後,會自動呼叫 exit 方法 class demo object def enter self print enter 方法被執行了 ret...
26 python筆記 上下文管理器
個人部落格 注 enter方法可以寫連線資料庫的操作,exit方法寫斷開連線的操作,用with去管理 注 with開啟的檔案物件,就是1個上下文管理器 注 with語句可以去管理 上下文管理器物件 注 使用with管理,上下文管理器 不需要自己呼叫 enter exit 方法 示例 上下文管理器 上...