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