Управление асинцио сопрограммы в dict - PullRequest
Новая
       23

Управление асинцио сопрограммы в dict

0 голосов
/ 21 января 2020

Я хочу управлять некоторыми сопрограммами в диктовке, пока выполняется основная сопрограмма. В частности, я хочу запустить бесконечные сопрограммы, поместить их обработчик в dict и снова отменить их с помощью dict call. В моем примере я хочу запустить 4 сопрограммы, и они отменят одну за другой с сопрограммой doomsday. Я использую Python 3.6. 

import asyncio
import traceback


async def add_to_handler(node, func):
    func_handler[node] = asyncio.ensure_future(func, loop=loop)
    return

async def test_func1():
    while True:
        print("1 - HeNlO")
        await asyncio.sleep(1)

async def test_func2():
    while True:
        print("2 - TestFunc2")
        await asyncio.sleep(2)


async def test_func3():
    while True:
        print("3 - Tukan")
        await asyncio.sleep(3)


async def test_func4():
    while True:
        print("4 - Do Coro!")
        await asyncio.sleep(4)


async def doomsday():
    # Cancel coroutine every 10 seconds
    print("launch doomsday")
    for i in range(len(func_handler)):
        await asyncio.sleep(10)
        print("start cancelling with {}".format(i))
        func_handler[str(i + 1)].cancel()
    return

async def main():
    await add_to_handler("1", test_func1)
    await add_to_handler("2", test_func2)
    await add_to_handler("3", test_func3)
    await add_to_handler("4", test_func4)
    await doomsday()
    while True:
        print("z..z..Z..Z...Z")
        print(func_handler)
        await asyncio.sleep(5)

func_handler = {}

loop = asyncio.get_event_loop()
try:
    loop.run_until_complete(main())
except KeyboardInterrupt:
    print("stop loop")
    loop.close()

Я попробовал это с помощью метода .call_later AbstractEventL oop вместо бесконечного, пока l oop, но все равно не хочет работа, и кажется, что мои сопрограммы рассматриваются как функции, но я не знаю почему. Где моя вина?

1 Ответ

2 голосов
/ 21 января 2020

попробуйте изменить эту функцию: 

async def add_to_handler(node, func):
    func_handler[node] = asyncio.ensure_future(func(), loop=loop)
    return None

обратите внимание на asyncio.ensure_future ( fun c () , loop = l oop)

...