python中终止协程和异常处理方式

目录

• 未处理的异常会导致协程终止
• 下面举例说明
• 总结

协程中未处理的异常会向上冒泡，传给 next 函数或 send 方法的调用方（即触发协程的对 象）。

下面示例举例说明如何使用之前博客示例中由装饰器定义的 averager 协程。

未处理的异常会导致协程终止

"""
预激协程的装饰器

"""


from inspect import getgeneratorstate
from functools import wraps


def coroutine(func):
  """装饰器：向前执行到第一个`yield`表达式，预激`func`"""

  # 把被装饰的生成器函数替换成这里的 primer 函数；
  # 调用 primer 函数时，返回预激后的 生成器。
  @wraps(func)
  def primer(*args, **kwargs):
    # 调用被装饰的函数，获取生成器对象。
    geskWEoPoBFpn = func(*args, **kwargs)
    # 预激生成器。
    next(gen)
    # 返回生成器。
    return gen

  return primer


@coroutine
def averager():
  total = 0.0
  count = 0
  average = None
  while True:
    term = yield average
    total += term
    count += 1
    average = total / count


if __name__ == '__main__':
  coro_avg = averager()
  # print(getgeneratorstate(coro_avg))
  print(coro_avg.send(10))
  print(coro_avg.send(30))
  # 发送的值不是数字，导致协程内部有异常抛出。
  print(coro_avg.send('spam'))
  # 由于在协程内没有处理异常，协程会终止。
  # 如果试图重新激活协程，android会抛出 StopIteration 异常。
  print(coro_avg.send(60))

上面示例，暗示了终止协程的一种方式：发送某个哨符值，让协程退出。内置的 None 和 Ellipsis 等常量经常用作哨符值。Ellipsis 的优点是，数据流中不js太常有这个值。我还见 过有人把 StopIteratihttp://www.devze.comon 类（类本身，而不是实例，也不抛出）作为哨符值；也就是说， 是像这样使用的：my_cowww.devze.comro.send(StopIteration)。

从 python 2.5 开始，客户代码可以在生成器对象上调用两个方法，显式地把异常发给协程。

这两个方法是 throw 和 close。

generator.throw(exc_type[, exc_value[, traceback]])

致使生成器在暂停的 yield 表达式处抛出指定的异常。

如果生成器处理了抛出的异常，代码会向前执行到下一个 yield 表达式，而产出的值会成为调用 generator.throw 方法 得到的返回值。

如果生成器没有处理抛出的异常，异常会向上冒泡，传到调用方的上下 文中。

generator.close()

致使生成器在暂停的yield 表达式处抛出GeneratorExit 异常。

如果生成器没有处 理这个异常，或者抛出了StopIteration 异常（通常是指运行到结尾），调用方不会 报错。

如果收到GeneratorExit 异常，生成器一定不能产出值，否则解释器会抛出 RuntimeError 异常。

生成器抛出的其他异常会向上冒泡，传给调用方。

下面举例说明

如何使用 close 和 throw 方法控制协程：

"""
学习在协程中处理异常的测试代码
"""
from inspect import getgeneratorstate

class DemoException(Exception):
  """为这次演示定义的异常类型。"""

def demo_exc_handling():
  print('-> coroutine started')
  try:
    while True:
      try:
        x = yield
      # 特别处理 DemoException 异常
      except DemoException:
        print('*** DemoException handled. Continuing...')
      # 如果没有异常，那么显示接收到的值。
      else:
        print('-> coroutine received: {!r}'.format(x))
  finally:
    # 如果不管协程如何结束都想做些清理工作，
    # 要把协程定义体中相关的代码放入try/ finally 块中
    print('-> coroutine ending')

if __name__ == '__main__':
  exc_coro = demo_exc_handling()
  next(exc_coro)
  exc_coro.send(11)
  exc_coro.send(22)
  # 激活和关闭 demo_exc_handling，没有异常
  # exc_coro.close()

  # 如果把 DemoException 异常传入 demo_exc_handling 协程，
  # 它会处理，然后继续运行
  # exc_coro.throw(DemoException)
  # exc_coro.send(33)

  # 如果无法处理传入的异常，协程会终止
  exc_coro.throw(ZeroDivisionError)

  print(g开发者_Python培训etgeneratorstate(exc_coro))

总结

以上为个人经验，希望能给大家一个参考，也希望大家多多支持我们。