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

当循环中断时,有没有办法使python自动调用上下文()的__exit__方法？还是其他一些实现同样目标的方法？我对于生成器和上下文管理器的了解使我怀疑这是不可能的,但是这使得内部管理器在生成器中是没有用的,不是吗？在我看来,一个块中的yield语句应该会引发一个红旗,上下文管理器__exit__可能不会运行.

解决方法

那么你可以用一个try / finally子句将context()函数中的yield包装起来：

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

编辑：如果您尝试：help(contextmanager),它将显示它的“典型”使用示例,它们用yield / finally子句包装yield.