生成器扩展知识

生成器扩展知识

前面说过函数内只要有yield关键字，那么该函数的执行结果就是一个生成器。生成器本身也是一个迭代器对象，可以使用next()方法去取值，还有一个格式可以支持send()去传值给yield并执行next()。

def foo():
while True:
    x = yield 'aaa'
    print('value:',x)
g=foo()
print(g)
print(g.send(None))    #第一次必须要next()触发，这里使用send(None)和next(等价)
print(g.send(2))      #send传值给yield并执行next()

#执行结果：
<generator object foo at 0x000001D39EC61DB0>
aaa
value: 2
aaa

总结

• 要使用send方法传值并触发生成器必须至少next()一次，或者send(None)一次，不然会报错:TypeError: can't send non-None value to a just-started generator

• send(2)的作用就是先执行传值，将2传给yield，然后执行next()

加装饰器实现先next()一次生成器

def deco(func):
	def wrapper(*args,**kwargs):
		res = func(*args,**kwargs)
		next(res)
		return(res)
	return wrapper
@deco
def foo():
	print('starting ...')
	while True:
    	x = yield 'aaa'
    	print('value:', x)
g = foo()
print(g.send(2))

#执行结果
starting ...     #g = foo()执行结果
value: 2
aaa

与其它函数协同工作

def deco(func):
	def wrapper(*args,**kwargs):
		res = func(*args,**kwargs)
		next(res)
		return(res)
	return wrapper
list_eat = []
@deco
def foo(name):
	print('starting ...')
	while True:
    	food = yield list_eat   #yield接受send过来的值，传给food
    	list_eat.append(food)
    	print('%s eat %s' %(name,food))


def make_egg(people,num):
	for i in range(1,num):
    	people.send('egg%s' %i)    #发送给people，peopl必须是一个生成器

g = foo('jack')
make_egg(g,5)

#执行结果：
starting ...
jack eat egg1
jack eat egg2
jack eat egg3
jack eat egg4

多个函数通过生成器协同
两个函数间传值，接受值格式：

def deco(func):
	def wrapper(*args,**kwargs):
    	res = func(*args,**kwargs)
    	next(res)
    	return res
	return wrapper
@deco
def foo1():
	while True:
		x1 = yield    #yield接受值，可以是其它函数传值过来
		print(x1)

def foo2(target):    #target传入foo1
	target.send(值)

三个或三个以上类似：

def deco(func):
	def wrapper(*args,**kwargs):
    	res = func(*args,**kwargs)
    	next(res)
    	return res
	return wrapper
@deco
def foo1():
	while True:
		x1 = yield    #yield接受值，可以是其它函数传值过来
		print(x1)

@deco
def foo2(target):
	while True:
		x2 = yield    
		target.send(x2)
def foo3(target):
	target.send(值)