协程与yield表达式
在函数内,yield语句可以作为表达式使用,出现在赋值运算符的右边,例如:
def receiver(): print(Ready to receive) while True: n = (yield) print(go %s, n)
以这种方式使用yield语句的函数称为协程
,向函数发送值时函数将执行,它的行为类似于生成器
r = reveiver() r.__next__() 或者 next(r) # 向前执行到第一条yield语句 for i in range(5): r.send(i) go %s 0 go %s 1 go %s 2 go %s 3 go %s 4
在这个例子中,一开始调用next()是必不可少的,这样协程才能执行第一个yield表示式之前的语句。这时,协程会挂起,
等待相关生成器对象r的send()方法给它发送一个值。传递给send()的值由协程中的(yield)表达式返回。接收到值后,协程
就会执行语句,直至遇到下一条yield语句。
在协程中需要首先调用next()这件事很容易被忽略,这经常成为错误出现的原因。因此,建议使用一个能自动完成该步骤的
装饰器来包装协程。
def coroutine(func): def start(*args, **kwargs): g = func(*args, **kwargs) g.next() return g return start
使用这个装饰器就可以像下面这样编写和使用协程:
@coroutine def receiver(): print(Ready to receive) while True: n = (yield) print(go %s, n) # 示例用法 r = receiver() r.send(Hello Wrold)
协程一般会不断的执行下去,除非被现式关闭或者自己退出。像下面这样使用方法close()可以关闭输入值的流
r.close() r.send(1) # 报错
关闭后,如果继续给协程发送值,就会引发StopIteration异常
如果yield表达式中提供了值,协程可以使用yield语句同时接收和发出返回值,例如
def line_splitter(delimiter=None): print(Ready to split) result = None while True: line = (yield result) result = line.split(delimiter)
在这个例子中,我们使用协程的方式与前面相同。但是,现在调用send()方法也会生成一个结果,例如:
s = line_splitter(',') s.next() # Ready to split s.send(A,B,C) # ['A', 'B', 'C'] s.send(100,200,300) # ['100', '200', '300']
理解这个例子中的先后顺序,首个next()调用让协程向前执行到(yield result),这将返回result的初始值None.
在接下来的send()调用中,接收到的值被放在line中并拆分到result中。send()方法的返回值就是传递给下一条
yield语句的值。换句话说,send()方法的返回值来自下一个yield表达式,而不是接收send()传递的值的yield表达式。