上下文管理器(context manager)是Python2.5開始支持的一種語法,用于規(guī)定某個對象的使用范圍。一旦進入或者離開該使用范圍,會有特殊操作被調(diào)用 (比如為對象分配或者釋放內(nèi)存)。它的語法形式是with...as...
關(guān)閉文件
我們會進行這樣的操作:打開文件,讀寫,關(guān)閉文件。程序員經(jīng)常會忘記關(guān)閉文件。上下文管理器可以在不需要文件的時候,自動關(guān)閉文件。
下面我們看一下兩段程序:
代碼如下:
# 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)
兩段程序?qū)嶋H上執(zhí)行的是相同的操作。我們的第二段程序就使用了上下文管理器 (with...as...)。上下文管理器有隸屬于它的程序塊。當(dāng)隸屬的程序塊執(zhí)行結(jié)束的時候(也就是不再縮進),上下文管理器自動關(guān)閉了文件 (我們通過f.closed來查詢文件是否關(guān)閉)。我們相當(dāng)于使用縮進規(guī)定了文件對象f的使用范圍。
上面的上下文管理器基于f對象的__exit__()特殊方法(還記得我們?nèi)绾卫锰厥夥椒▉韺崿F(xiàn)各種語法?參看特殊方法與多范式)。當(dāng)我們使用上下文管理器的語法時,我們實際上要求Python在進入程序塊之前調(diào)用對象的__enter__()方法,在結(jié)束程序塊的時候調(diào)用__exit__()方法。對于文件對象f來說,它定義了__enter__()和__exit__()方法(可以通過dir(f)看到)。在f的__exit__()方法中,有self.close()語句。所以在使用上下文管理器時,我們就不用明文關(guān)閉f文件了。
自定義
任何定義了__enter__()和__exit__()方法的對象都可以用于上下文管理器。文件對象f是內(nèi)置對象,所以f自動帶有這兩個特殊方法,不需要自定義。
下面,我們自定義用于上下文管理器的對象,就是下面的myvow:
代碼如下:
# 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
新聞熱點
疑難解答
圖片精選