如何在异步函数中使用 'yield'？

Question

我想使用生成器 yield 和 async 函数。我读了 this topic，然后写了下一个代码：

import asyncio

async def createGenerator():
    mylist = range(3)
    for i in mylist:
        await asyncio.sleep(1)
        yield i*i

async def start():
    mygenerator = await createGenerator()
    for i in mygenerator:
        print(i)

loop = asyncio.get_event_loop()

try:
    loop.run_until_complete(start())

except KeyboardInterrupt:
    loop.stop()
    pass

但是我得到了错误：

语法错误：'yield' 内部异步函数

如何在异步函数中使用yield生成器？

Answer 1

更新：

从 Python 3.6 开始，我们有 asynchronous generators 并且能够在协程中直接使用 yield。

import asyncio


async def async_generator():
    for i in range(3):
        await asyncio.sleep(1)
        yield i*i


async def main():
    async for i in async_generator():
        print(i)


loop = asyncio.get_event_loop()
try:
    loop.run_until_complete(main())
finally:
    loop.run_until_complete(loop.shutdown_asyncgens())  # see: https://docs.python.org/3/library/asyncio-eventloop.html#asyncio.loop.shutdown_asyncgens
    loop.close()

Python3.5 的旧答案：

你不能 yield 在协同程序中。唯一的方法是使用 __aiter__/__anext__ 魔术方法手动实施 Asynchronous Iterator。在你的情况下：

import asyncio


class async_generator:
    def __init__(self, stop):
        self.i = 0
        self.stop = stop

    async def __aiter__(self):
        return self

    async def __anext__(self):
        i = self.i
        self.i += 1
        if self.i <= self.stop:
            await asyncio.sleep(1)
            return i * i
        else:
            raise StopAsyncIteration


async def main():
    async for i in async_generator(3):
        print(i)


if __name__ == "__main__":
    loop = asyncio.get_event_loop()
    loop.run_until_complete(main())

输出：

0
1
4

这里还有两个例子：,

Answer 2

新 Python 3.6 支持异步发电机。

PEP 0525

What's new in Python 3.6

PS: 在撰写本文时 Python 3.6 仍处于测试阶段。如果您在 GNU/Linux 或 OS X 上并且迫不及待，您可以尝试使用 pyenv.

的新 Python

Answer 3

这应该适用于 python 3.6（使用 3.6.0b1 测试）：

import asyncio

async def createGenerator():
    mylist = range(3)
    for i in mylist:
        await asyncio.sleep(1)
        yield i*i

async def start():
    async for i in createGenerator():
        print(i)

loop = asyncio.get_event_loop()

try:
    loop.run_until_complete(start())

except KeyboardInterrupt:
    loop.stop()
    pass

如何在异步函数中使用 'yield'？

How to use 'yield' inside async function?

python

yield

python-asyncio

python-3.5