Click to Visit Homepage : zzyzz.top


Coroutine 预激装饰器

 1 预激装饰器
 2 
 3 讨论如何终止协程之前,我们要先谈谈如何启动协程。使用协程之前必须预激,可是这一
 4 步容易忘记。为了避免忘记,可以在协程上使用一个特殊的装饰器。接下来介绍这样一个
 5 装饰器。
 6 
 7 预激协程的装饰器,
 8 from functools import wraps
 9 
10 def corountine(func):
11     '''
12         装饰器:向前执行到第一个 yield 表达式,预激协程 func
13     :param func:
14     :return:
15     '''
16     @wraps(func)                       # functools.wraps 则可以将原函数对象的指定属性复制给包装函数对象, 默认有 __module__、__name__、__doc__,或者通过参数选择
17     def primer(*args, **kwargs):       # 把被装饰的生成器函数替换成这里的 primer 函数;调用 primer 函数时,返回预激后的生成器
18         gen = func(*args, **kwargs)    # 调用被装饰的函数,获取生成器对象。
19         next(gen)                      # 预激生成器
20         return gen                     # 返回生成器
21     return primer
22 
23 
24 @corountine               # 预激装饰器
25 def coro_average():
26     total = 0.0
27     count = 0
28     average = None
29     while 1:
30         term = yield average
31         total += term
32         count += 1
33         average = total/count
34 
35 coro3 = coro_average()
36 #print (coro3.send(None))  # 若没有 预激装饰器  需要 调用 send(None) 或 next(coro) 完成预激,即让代码跑到第一个 yield 处
37 print (coro3.send(5))
38 print (coro3.send(7))
39 print (coro3.send(10))
40 coro3.close()
41 
42 这个无限循环表明,只要调用方不断把值发给这个协程,它就会一直接收值,然后生
43 成结果。仅当调用方在协程上调用 .close() 方法,或者没有对协程的引用而被垃圾回收
44 程序回收时,这个协程才会终止。
45 这里的 yield 表达式用于暂停执行协程,把结果发给调用方;还用于接收调用方后面
46 发给协程的值,恢复无限循环。
47 使用协程的好处是, total 和 count 声明为局部变量即可,无需使用实例属性或闭包在
48 多次调用之间保持上下文。
49 
50 调用 next(coro3) 函数后,协程会向前执行到
51 yield 表达式,产出 average 变量的初始值——None,因此不会出现在控制台中。此
52 时,协程在 yield 表达式处暂停,等到调用方发送值。 coro3.send(5) 那一行发送
53 一个值,激活协程,把发送的值赋给 term,并更新 total、 count 和 average 三个变量
54 的值,然后开始 while 循环的下一次迭代,产出 average 变量的值,等待下一次为
55 term 变量赋值.

 

posted @ 2017-11-19 21:06  zzYzz  阅读(322)  评论(0编辑  收藏  举报


Click to Visit Homepage : zzyzz.top