我們還可以用裝飾器(decorators)和生成器(generators)來實(shí)現(xiàn)上下文管理器。
Python有個(gè)contextlib模塊專門用于這個(gè)目的。我們可以使用一個(gè)生成器函數(shù)來實(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)方式看起來更加直觀和簡(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ì)象封裝過的生成器。GeneratorContextManager被賦值給open_file函數(shù),我們實(shí)際上是在調(diào)用GeneratorContextManager對(duì)象。那現(xiàn)在我們既然知道了所有這些,我們可以用這個(gè)新生成的上下文管理器了,像這樣:
with open_file('some_file') as f:
f.write('hola!')