详解Python生成器和基于生成器的协程_Python

详解Python生成器和基于生成器的协程

2021-11-22 10:11WinvenChang Python

说到Python协程就会想到,进程和线程,当然更离不开生成器.今天就给大家整理了本篇文章,文中有非常详细的介绍,需要的朋友可以参考下

一、什么是生成器

Generator

1.生成器就是可以生成值的函数
2.当一个函数里有了 yield关键字就成了生成器
3.生成器可以挂起执行并且保持当前执行的状态

代码示例：

				?

									def simple_gen():

									    yield 'hello'

									    yield 'world'

									gen = simple_gen()

									print(type(gen))  # 'generator' object

									print(next(gen))  # 'hello'

									print(next(gen))  # 'world'

二、基于生成器的协程

Python3之前没有原生协程，只有基于生成器的协程

1.pep 342(Coroutines via Enhanced Generators)增强生成器功能
2.生成器可能通过 yield 暂停执行和产出数据
3.同时支持send()向生成器发送数据和throw()向生成器抛出异常

Generator Based Corouteine代码示例：

				?

									def coro():

									    hello = yield 'hello'  # yield 关键字在 = 右边作为表达式，可以被 send 值

									    yield hello

									c = coro()

									# 输出 'hello', 这里调用 next 产出第一个值 'hello'，之后函数暂停

									print(next(c))

									# 再次调用  send 发送值，此时 hello 变量赋值为 'world'，然后 yield 产出 hello 变量的值 'world'

									print(c.send('world'))

									# 之后协程结束，后续再 send 值会抛出异常 StopIteration

运行结果：

详解Python生成器和基于生成器的协程

三、协程的注意点

协程注意点

1.协程需要使用send(None)或者next(coroutine)来预激(prime)才能启动
2.在yield处协程会暂停执行
3.单独的yield value会产出值给调用方
4.可以通过 coroutine.send(value)来给协程发送值，发送的值会赋值给 yield表达式左边的变量value = yield
5.协程执行完成后（没有遇到下一个yield语句）会抛出StopIteration异常

四、协程装饰器

避免每次都要用 send 预激它

				?

									from functools import wraps

									def coroutine(func):  # 这样就不用每次都用 send(None) 启动了

									    “”“装饰器：向前执行到一个 `yield` 表达式，预激 `func` ”“”

									    @wrops(func)

									    def primer(*args, **kwargs):   # 1

									        gen = func(*args, **kwargs)  # 2

									        next(gen)  # 3

									        return gen  # 4

									    return primer

五、python3原生协程

python3.5引入 async/await支持原生协程(native coroutine)

				?

									import asyncio

									import datetime

									import random

									async def display_date(num, loop):

									    end_time = loop.time() + 50.0

									    while True:

									        print('Loop: {} Time: {}'.format(num, datetime.datetime.now())

									        if (loop.time() + 1.0) >= end_time:

									            break

									        await asyncio.sleep(random.randint(0, 5))

									loop = asyncio.get_event_loop()

									asyncio.ensure_future(display_date(1, loop))

									asyncio.ensure_future(display_date(2, loop))

									loop.run_forever()