我們還可以用裝飾器(decorators)和生成器(generators)來實現(xiàn)上下文管理器。
Python有個contextlib
模塊專門用于這個目的。我們可以使用一個生成器函數(shù)來實現(xiàn)一個上下文管理器,而不是使用一個類。
讓我們看看一個基本的,沒用的例子:
from contextlib import contextmanager
@contextmanager
def open_file(name):
f = open(name, 'w')
yield f
f.close()
OK啦!這個實現(xiàn)方式看起來更加直觀和簡單。然而,這個方法需要關(guān)于生成器、yield
和裝飾器的一些知識。在這個例子中我們還沒有捕捉可能產(chǎn)生的任何異常。它的工作方式和之前的方法大致相同。
讓我們小小地剖析下這個方法。
yield
關(guān)鍵字。因為這個緣故它創(chuàng)建了一個生成器而不是一個普通的函數(shù)。contextmanager
會被調(diào)用并傳入函數(shù)名(open_file
)作為參數(shù)。contextmanager
函數(shù)返回一個以GeneratorContextManager
對象封裝過的生成器。GeneratorContextManager
被賦值給open_file
函數(shù),我們實際上是在調(diào)用GeneratorContextManager
對象。那現(xiàn)在我們既然知道了所有這些,我們可以用這個新生成的上下文管理器了,像這樣:
with open_file('some_file') as f:
f.write('hola!')