获取使用@contextmanager创建的上下文管理器可以正常使用异常

Question

我有以下代码

from contextlib import contextmanager

@contextmanager
def simple_context_manager():
    print("starting context manager")
    yield
    print("finished context manager")

try:
    with simple_context_manager():
        raise RuntimeError
except RuntimeError:
    print("Caught the error")
print("Moving on")

现在它打印出来了

starting context manager
Caught the error
Moving on

这告诉我上下文管理器没有关闭。 如何让它关闭并打印“完成的上下文管理器”行？

因为我正在使用装饰器，所以我没有专门的__exit__函数，我认为应该根据它调用它 。

因此，我不确定如何在其上下文中发生错误的情况下让我的上下文管理器退出。

Answer 1

你需要一个尝试 - 终于：

@contextmanager
def simple_context_manager():
    print("starting context manager")
    try:
        yield
    finally:
        print("finished context manager")

如果一个异常传播出的with说法， @contextmanager装饰会扔在点外进入装饰发生器yield 。 finally让我们执行一个清理块，无论是否发生异常，所以我们使用它。