Coroutine 预激装饰器

预激装饰器

讨论如何终止协程之前，我们要先谈谈如何启动协程。使用协程之前必须预激，可是这一
步容易忘记。为了避免忘记，可以在协程上使用一个特殊的装饰器。接下来介绍这样一个
装饰器。

预激协程的装饰器,
from functools import wraps

def corountine(func):
    '''
        装饰器：向前执行到第一个 yield 表达式，预激协程 func
    :param func:
    :return:
    '''
    @wraps(func)                       # functools.wraps 则可以将原函数对象的指定属性复制给包装函数对象, 默认有 __module__、__name__、__doc__,或者通过参数选择
    def primer(*args, **kwargs):       # 把被装饰的生成器函数替换成这里的 primer 函数；调用 primer 函数时，返回预激后的生成器
        gen = func(*args, **kwargs)    # 调用被装饰的函数，获取生成器对象。
        next(gen)                      # 预激生成器
        return gen                     # 返回生成器
    return primer


@corountine               # 预激装饰器
def coro_average():
    total = 0.0
    count = 0
    average = None
    while 1:
        term = yield average
        total += term
        count += 1
        average = total/count

coro3 = coro_average()
#print (coro3.send(None))  # 若没有 预激装饰器  需要 调用 send(None) 或 next(coro) 完成预激，即让代码跑到第一个 yield 处
print (coro3.send(5))
print (coro3.send(7))
print (coro3.send(10))
coro3.close()

这个无限循环表明，只要调用方不断把值发给这个协程，它就会一直接收值，然后生
成结果。仅当调用方在协程上调用 .close() 方法，或者没有对协程的引用而被垃圾回收
程序回收时，这个协程才会终止。
这里的 yield 表达式用于暂停执行协程，把结果发给调用方；还用于接收调用方后面
发给协程的值，恢复无限循环。
使用协程的好处是， total 和 count 声明为局部变量即可，无需使用实例属性或闭包在
多次调用之间保持上下文。

调用 next(coro3) 函数后，协程会向前执行到
yield 表达式，产出 average 变量的初始值——None，因此不会出现在控制台中。此
时，协程在 yield 表达式处暂停，等到调用方发送值。 coro3.send(5) 那一行发送
一个值，激活协程，把发送的值赋给 term，并更新 total、 count 和 average 三个变量
的值，然后开始 while 循环的下一次迭代，产出 average 变量的值，等待下一次为
term 变量赋值.