处理在生成器中引发的异常

Question

我有一个生成器和一个使用它的函数：

def read():
    while something():
        yield something_else()

def process():
    for item in read():
        do stuff

如果生成器引发异常，我想在使用者函数中处理该异常，然后继续使用迭代器，直到耗尽为止。请注意，我不想在生成器中有任何异常处理代码。

我想到了类似的东西：

reader = read()
while True:
    try:
        item = next(reader)
    except StopIteration:
        break
    except Exception as e:
        log error
        continue
    do_stuff(item)

但这对我来说看起来很尴尬。

Answer 1

这也是我不确定是否正确/优雅处理的事情。

我要做的是从生成器中获取yield一个Exception，然后将其提升到其他位置。喜欢：

class myException(Exception):
    def __init__(self, ...)
    ...

def g():
    ...
    if everything_is_ok:
        yield result
    else:
        yield myException(...)

my_gen = g()
while True:
    try:
        n = next(my_gen)
        if isinstance(n, myException):
            raise n
    except StopIteration:
        break
    except myException as e:
        # Deal with exception, log, print, continue, break etc
    else:
        # Consume n

这样，我仍然继承了Exception而没有引发它，这将导致生成器功能停止。主要缺点是我需要isinstance在每次迭代时检查产生的结果。我不喜欢可以产生不同类型结果的生成器，但是将其用作最后的选择。