有没有一种有意义的方式来使用发生器内的上下文管理器？

Question

from contextlib import contextmanager 

@contextmanager 
def context(): 
    print "entering" 
    yield 
    print "exiting" 

def test(): 
    with context(): 
     for x in range(10): 
      yield x 

for x in test(): 
    if x == 5: 
     break # or raise 

输出：

entering 

有没有一种方法，使蟒蛇自动调用上下文的__exit__（）方法时，for -loop中断？或者达到相同目标的其他方式？我对发生器和上下文管理器的了解让我怀疑这是不可能的，但是这使得上下文管理器在发生器内部毫无用处，不是吗？在我看来，with块内的yield声明应该引起红旗，上下文管理器__exit__可能不运行。

Answer 1

嗯，你可以包裹在上下文中的收益率（）函数与一个try/finally从句：

from contextlib import contextmanager 

@contextmanager 
def context(): 
    print "entering" 
    try: 
     yield 
    finally: 
     print "exiting" 

def test(): 
    with context(): 
     for x in range(10): 
      yield x 

for x in test(): 
    if x == 5: 
     break # or raise 

输出：

entering 
exiting 

编辑：如果你尝试：帮助（contextmanager），它会显示它是“典型的”用例，它们用try/finally子句包装yield。