如何捕獲協程StopIteration異常？

Question

我使用python2.7。

def printtext():
    try:
        line = yield
        print line
    except StopIteration:
        pass

if __name__ == '__main__':
    p = printtext()
    p.send(None)
    p.send('Hello, World')

我嘗試捕獲StopIteration異常，但仍被引發而沒有被捕獲。

您能否給我一些提示，為什么在這種情況下逃避StopIteration異常？

Answer 1

引發StopIteration時，您會誤會。 當生成器函數退出時（而不是在yield表達式期間），將引發StopIteration 。 因此，捕獲此錯誤的唯一方法是在函數外部執行此操作...

def printtext():
    line = yield
    print line

if __name__ == '__main__':
    p = printtext()
    p.send(None)
    try:
        p.send('Hello, World')
    except StopIteration:
        pass