python 终止协程和异常处理

2019-09-08 18:36:51 阅读：548 来源： 互联网

标签：协程 exc python coro 生成器 print 终止异常

协程中未处理的异常会向上冒泡，传给 next 函数或 send 方法的调用方（即触发协程的对象）。下面示例举例说明如何使用之前博客示例中由装饰器定义的 averager 协程。

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

"""
预激协程的装饰器

"""


from inspect import getgeneratorstate
from functools import wraps


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

    # 把被装饰的生成器函数替换成这里的 primer 函数；
    # 调用 primer 函数时，返回预激后的 生成器。
    @wraps(func)
    def primer(*args, **kwargs):
        # 调用被装饰的函数，获取生成器对象。
        gen = 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'))
    # 由于在协程内没有处理异常，协程会终止。
    # 如果试图重新激活协程，会抛出 StopIteration 异常。
    print(coro_avg.send(60))

上面示例，暗示了终止协程的一种方式：发送某个哨符值，让协程退出。内置的 None 和 Ellipsis 等常量经常用作哨符值。Ellipsis 的优点是，数据流中不太常有这个值。我还见过有人把 StopIteration 类（类本身，而不是实例，也不抛出）作为哨符值；也就是说，是像这样使用的：my_coro.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(getgeneratorstate(exc_coro))

标签：协程,exc,python,coro,生成器,print,终止,异常
来源： https://blog.csdn.net/MZP_man/article/details/100555806

本站声明： 1. iCode9 技术分享网（下文简称本站）提供的所有内容，仅供技术学习、探讨和分享；
2. 关于本站的所有留言、评论、转载及引用，纯属内容发起人的个人观点，与本站观点和立场无关；
3. 关于本站的所有言论和文字，纯属内容发起人的个人观点，与本站观点和立场无关；
4. 本站文章均是网友提供，不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属；如您发现该文章侵犯了您的权益，可联系我们第一时间进行删除；
5. 本站为非盈利性的个人网站，所有内容不会用来进行牟利，也不会利用任何形式的广告来间接获益，纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。

ICode9

python 终止协程和异常处理