Question

我在python中有一个协程（增强型生成器），在数据结束后要执行一些代码：

def mycoroutine():
  try:
    while True:
      data = (yield)
      print data
  finally:
    raise ValueError
    print "END"

co = mycoroutine()
co.next()

for i in (1,2,3):
  co.send(i)

不会引发ValueError异常，但解释器只会打印：

Exception ValueError: ValueError() in <generator object mycoroutine at 0x2b59dfa23d20> ignored

有没有办法在调用者中捕获异常？

Answer 1

异常是引发的。生成器关闭时执行finally块。关闭生成器是通过在生成器上下文中引发GeneratorExit exception来完成的。

忽略异常，因为生成器在被删除之前不会被关闭（在这种情况下，当Python退出时自动生成）;生成器__del__处理程序关闭生成器，触发finally:块：

>>> def mycoroutine():
...   try:
...     while True:
...       data = (yield)
...       print data
...   finally:
...     raise ValueError
...     print "END"
... 
>>> co = mycoroutine()
>>> co.next()
>>> co.close()
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
  File "<stdin>", line 7, in mycoroutine
ValueError
>>> co = mycoroutine()
>>> co.next()
>>> del co
Exception ValueError: ValueError() in <generator object mycoroutine at 0x1046a9fa0> ignored

清理期间引发的异常总是被忽略;请参阅object.__del__() documentation：

警告：由于__del__()的不稳定情况调用方法，执行期间发生的异常忽略，而是向sys.stderr打印警告。

解决方案是在清理生成器时不会引发异常，或通过显式关闭生成器来捕获异常：

>>> co = mycoroutine()
>>> co.next()
>>> try:
...     co.close()
... except ValueError:
...     pass
... 
>>> del co
>>> # No exception was raised
...

您还可以捕获GeneratorExit异常并在此时执行一些清理：

def mycoroutine():
  try:
    while True:
      data = (yield)
      print data
  except GeneratorExit:
    print "Generator exiting!"

但请注意，StopIteration或GeneratorExit以外的任何例外都会传播;请参阅generator.close() documentation：

如果生成器函数然后引发StopIteration（通过正常退出，或者由于已经关闭）或GeneratorExit（通过不捕获异常），则关闭返回其调用者。如果生成器产生一个值，则会引发RuntimeError。如果生成器引发任何其他异常，它将传播给调用者。

避免在python增强型生成器中“忽略异常”

1 个答案: