Python: дождитесь окончания генератора
Текущие версии Python (декабрь 2022 г.) по-прежнему позволяют использовать декоратор @coroutine, и генерация может быть такой:
import asyncio
asyncify = asyncio.coroutine
data_ready = False # Status of a pipe, just to test
def gen():
global data_ready
while not data_ready:
print("not ready")
data_ready = True # Just to test
yield
return "done"
async def main():
result = await asyncify(gen)()
print(result)
loop = asyncio.new_event_loop()
loop.create_task(main())
loop.run_forever()
Тем не менее, новые версии Python 3.8+ не будут поддерживать декоратор @coroutine (псевдоним функции asyncify), как дождаться (дождаться) завершения работы генератора, как указано выше?
Я пытался использовать async def, как и ожидалось в предупреждении, но не работал:
import asyncio
asyncify = asyncio.coroutine
data_ready = False # Just to test
async def gen():
global data_ready
while not data_ready:
print("not ready")
data_ready = True # Just to test
yield
yield "done"
return
async def main():
# this has error: TypeError: object async_generator can't be used in 'await' expression
result = await gen()
print(result)
loop = asyncio.new_event_loop()
loop.create_task(main())
loop.run_forever()
Ручное использование yield в генераторе, помеченном asyncio.coroutine, — чрезвычайно необычная вещь, которую можно делать где-либо за пределами самого ядра asyncio — я не думаю, что семантика этого на самом деле является частью общедоступного API asyncio.
я пытаюсь опросить, пока данные не будут готовы, а не блокировать основной поток, для этой цели предназначен первый образец. Второй пример — это просто попытка конвертировать в async def, чтобы заменить устаревший декоратор @coroutine.
(Кроме того, asyncio.coroutine уже устарела с версии 3.8 и уже удалена в версии 3.11.)
Может быть, вы просто ищете asyncio.sleep?
Ответы 1
Асинхронные генераторы наследуют асинхронный итератор и предназначены для асинхронных итераций.
Вы не можете напрямую await их как обычные сопрограммы.
Имея это в виду, возвращаясь к вашему экспериментальному случаю и вашему вопросу «как дождаться (дождаться) окончания генератора?»: чтобы получить окончательное значение yielded, выполните асинхронные итерации:
import asyncio
data_ready = False # Just to test
async def gen():
global data_ready
while not data_ready:
print("not ready")
data_ready = True # Just to test
yield "processing"
yield "done"
return
async def main():
a_gen = gen()
async for result in a_gen: # assign to result on each async iteration
pass
print('result:', result)
asyncio.run(main())
Отпечатки:
not ready
result: done
Естественно, вы также можете продвигать асинхронный генератор пошагово с помощью anext:
a_gen = gen()
val_1 = await anext(a_gen)
Подводя итог, следуйте рекомендациям PEP 525 — Асинхронные генераторы и старайтесь не смешивать устаревшие вещи с актуальными.
спасибо, тоже нашел похожее решение await a_gen.asend(None) :) на python <3.10
Эти два примера кода показывают очень разные вещи. Первый — ожидаемая сопрограмма, второй — асинхронный генератор. Ни один из них не подходит для «ожидания окончания генератора», потому что ни один из них не ждет генератор. Что именно ты пытаешься сделать? Чему должен уступать генератор, и что он должен уступать? Другими словами, зачем вообще нужен генератор?