4

如何防止异常过早终止 Python 生成器/迭代器?

例如,考虑以下情况:

#!/usr/bin/env python

def MyIter():
    yield 1
    yield 2
    yield 3
    raise Exception, 'blarg!'
    yield 4 # this never happens

my_iter = MyIter()
while 1:
    try:
        value = my_iter.next()
        print 'value:',value
    except StopIteration:
        break
    except Exception, e:
        print 'An error occurred: %s' % (e,)

我希望输出:

value: 1
value: 2
value: 3
An error occurred: blarg!
value: 4

但是,在异常之后,迭代器终止并且值 4 永远不会产生,即使我已经处理了异常并且没有破坏我的循环。为什么会这样?

4

2 回答 2

8

生成器需要自己处理异常;如果它引发未处理的异常,它将被终止,就像任何其他函数一样。

于 2013-04-26T19:58:27.313 回答
2

您需要处理生成器中可能发生的任何异常:

def MyIter():
    yield 1
    yield 2
    yield 3
    try:
       raise Exception, 'blarg!'
    except Exception:
       pass
    yield 4 # this will happen
于 2013-04-26T20:01:37.757 回答