Python - 异常似乎正在跳过块[重复]

Question

以下是相关代码：

    @contextlib.contextmanager
    def make_temp_dir():
      temp_dir = tempfile.mkdtemp()
      yield temp_dir
      shutil.rmtree(temp_dir)

    with make_temp_dir(listing_id) as tmpdir:
      pass
      # Sometimes something in here throws an exception that gets caught higher up

好的，写完这一切，我现在明白发生了什么。我正在使用装饰器创建的上下文管理器中的 exit 方法 正在运行，但这当然不会将流返回到我的生成器。

那么应该我该怎么做呢？

Answer 1

这里发生的情况如下：

• 在__enter__() 上，生成器启动。它产生的返回值是__enter__()。
• 在__exit__() 上，以正常方式或通过注入异常来恢复生成器。相关代码在$PYTHONROOT/contextlib.py 中，您可以看到在生成器上调用了next() 或throw()。

如果在生成器上调用throw()，则会在它内部引发异常，这正是我们上次离开它的位置，即。 e.然后yield 表达式引发异常。

因此，您必须将yield 包含在try: 语句中。只有这样，您才能对异常做一些事情。

如果您不这样做，您的生成器将在不执行任何操作的情况下引发异常。

你可能想要

@contextlib.contextmanager
def make_temp_dir():
    temp_dir = tempfile.mkdtemp()
    try:
        yield temp_dir
    finally:
        shutil.rmtree(temp_dir)