Как использовать "yield" внутри асинхронной функции?

Question

Я хочу использовать генератор yield и асинхронные функции. Я прочитал эту тему и написал следующий код:

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

Но я получил ошибку:

SyntaxError: "yield" внутри асинхронной функции

Как использовать генератор урожайности в асинхронной функции?

Answer 1

Upd:

Начиная с Python 3.6 у нас есть асинхронные генераторы и возможность использовать 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.AbstractEventLoop.shutdown_asyncgens
    loop.close()

Старый ответ для Python 3.5:

Ты не можешь yield внутри сопрограмм. Единственный способ - реализовать асинхронный итератор вручную, используя __aiter__/__anext__ магические методы. В твоем случае:

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

Вот еще два примера: 1, 2

Answer 2

Новый Python 3.6 поставляется с поддержкой асинхронных генераторов.

ОПТОСОЗ 0525

Что нового в Python 3.6

PS: На момент написания Python 3.6 все еще бета. Если вы работаете в GNU/Linux или OS X и не можете ждать, попробуйте новый Python с pyenv.

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