Как во время работы кода подключать неопределённое количество новых Discord ботов асинхронно?

Question

[john_does13]

Я использую disnake.py и asyncio для реализации моей идеи. Её суть в том, что-бы запустить некий событийный цикл, который будет ожидать, что в любой момент мы захотим подключить к нашему коду нового Discord бота (количество заранее неизвестно). В процессе реализации я столкнулся с проблемой: у меня не получаеться подключить нового бота в асинхронном режиме работы.

import asyncio
import disnake


class BotHandler:
    def __init__(self, token: str):
        self.token = token
        intents = disnake.Intents.all()
        self.bot = disnake.Client(intents=intents)

        @self.bot.event
        async def on_ready():
            print(f"'{self.bot.user.name}' is online!")

    async def start(self):
        await self.bot.start(self.token)


async def start_bots():
    bot1 = BotHandler(token1)
    bot2 = BotHandler(token2)

    tasks = [bot1.start(), bot2.start()]
    await asyncio.gather(*tasks)


if __name__ == '__main__':
    asyncio.run(start_bots())

Код, написанный мной выше, работает, но не так, как мне хотелось бы. Боты действительно запускаются, они работают. Но, при таком подходе я не имею возможности подключить ещё одного бота.

Comments

[Модератор]

Не надо ставить как можно больше тэгов. Лучше оставить один, но конкретный, с которым проблема.
См.п.3.1 Регламента. Также обратите внимание на п.3.4

Answer 1

[Vindicar]

asyncio.create_task()

Comments

[john_does13]

Таски позволяют отойти от gather, но всё же это не решает моих проблем.

async def start_bots():
    while True:
        token = input('token: ')

        task_bot = asyncio.create_task(BotHandler(token).start())
        await task_bot


if __name__ == '__main__':
    asyncio.run(start_bots())

Что я могу использовать при такой реализации, но что-бы строка "await task_bot" не блокировала цикл while (и у меня оставалась возможность подключать новых ботов)?

[Vindicar]

john_does13, а на кой тебе вообще await task_bot?
Таск будет выполняться, как только он создан.

[john_does13]

Vindicar, нет, в моём случае это так не работает. Если убрать await task_bot, то цикл всё же будет работать, но боты не станут запускаться.

[Vindicar]

john_does13, почему не будут, когда будут? Вот тебе синтетический пример:

import asyncio

async def test(name: str):
    await asyncio.sleep(0.1)
    print(name, 'is running')
    await asyncio.sleep(2)
    print(name, 'is done')


async def run_all():
    for n in ['bot1', 'bot2', 'bot3']:
        asyncio.create_task(test(n))
    await asyncio.sleep(1)
    # дозапускаем еще таск
    asyncio.create_task(test('bot4'))
    await asyncio.sleep(5)  # так как мы не ожидаем выполнения тасков, нам нужно задержаться в рабочем цикле


asyncio.run(run_all())

Альтернативно, можно убрать последний sleep() в run_all() и заменить asyncio.run() на

loop = asyncio.get_event_loop()
loop.create_task(run_all())
loop.run_forever()

[john_does13]

Vindicar, действительно, код такого вида работает:

async def start_bots():
    while True:
        token = input("Enter your token: ")
        asyncio.create_task(BotHandler(token).start())
        await asyncio.sleep(3)


if __name__ == '__main__':
    loop = asyncio.new_event_loop()
    loop.create_task(start_bots())
    loop.run_forever()

Но мне не нравиться "await asyncio.sleep(3)", я уверен, что этого можно избежать. Ведь боту нужно время для запуска (в моём случае магические 3 секунды). Можно ли дождаться запуска бота без этой строчки? (Без неё цикл сразу продолжиться, а боты не будут запущены)
Есть "asyncio.wait" и "asyncio.wait_for", но насколько я знаю, они дожидаются окончательного выполнения корутин, что мне не подходит.

[Vindicar]

john_does13, можно. Создай внутри BotHandler объект asyncio.Event, и настрой запускаемого бота так, чтобы по завершении инициализации он делал event.set(). Тогда ты сможешь сделать
asyncio.wait_for(event.wait(), timeout) и работать дальше в зависимости от этого.

[john_does13]

Vindicar, спасибо за помощь, действительно помогли! Можете поделиться литературой, которую использовали при изучении asyncio?

[Vindicar]

john_does13, да всего понемножку, отдельной литературы не искал. Когда главную идею (рабочий цикл - реактор) ухватишь, дальше уже вопрос знания стандартной библиотеки.