Как использовать "yield" внутри асинхронной функции?
Я хочу использовать генератор 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" внутри асинхронной функции
Как использовать генератор урожайности в асинхронной функции?
3 ответа
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
Новый Python 3.6 поставляется с поддержкой асинхронных генераторов.
PS: На момент написания Python 3.6 все еще бета. Если вы работаете в GNU/Linux или OS X и не можете ждать, попробуйте новый Python с pyenv.
Это должно работать с 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