Как закончить функции вызванные через asyncio.gather()?

Question

[Maxwell012]

Мой код:

try:
        tasks = [
            asyncio.create_task(start_notifier(bot), ),
            asyncio.create_task(dp.start_polling(bot))
        ]
        await asyncio.gather(*tasks, return_exceptions=True)
    except Exception as ex:
        main_logger.critical(f"Bot stopped due to exception: {ex}", exc_info=True)
        # await dp.start_polling(bot)  # restart the bot
    finally:
        await bot.send_message(chat_id=Config.ADMIN_ID, text="<b>BOT STOPPED</b>")
        await bot.session.close()

Я вызываю 2 метода через asyncio.gather(), первый это телеграм бот и второй это метод которые отправляет сообщения. Проблема в том что после запуска и попытки остановить процес с помощью Ctrl + C ничего не происходит так как asyncio просто игнорирует это. Даже если я получаю ошибку от телеграм бота, скрипт все равно не заканчивает работа так как нотификатор продолжает работать, нотификатор это закольцованный метод (и в итоге, код работает безконечно пока я его принудительно не остановлю через пайчарм, но если это на сервере, то мне приходиться выходить с сервера чтобы сервер принудительно закончил мою программу), а ошибку от телеграмм бота я тоже не могу отловить так как asyncio перехватывает ее. Я пробовывал выставить параметр return_exceptions как True но это не помогает, ошибку я все равно не получаю.

Comments

[Everything_is_not_so_bad]

после запуска и попытки остановить процес с помощью Ctrl + C ничего не происходит так как asyncio просто игнорирует это

А зачем прерывать?

[Maxwell012]

Иерокопус Таманский, обнову когда например хочу загрузить нужно остановить и заново запустить

Answer 1

[Everything_is_bad]

Зачем ты вообще так криво стал делать, надо просто запустить dp.start_polling как рекомендовано в aiogram, а start_notifier перед этим запускать через dp.startup.register() или просто как create_task, не нужен тебе тут gather
ЗЫ ну и сразу разберись с webhook, прод боты лучше уж через него запускать

но если это на сервере, то мне приходиться выходить с сервера чтобы сервер принудительно закончил мою программу

а это что за? 100500 раз сказано, делайте сервис для systemd

Comments

[Maxwell012]

а это что за? 100500 раз сказано, делайте сервис для systemd

Я делаю сервис, но когда нужно быстро протетсить что-то на сервисе то я его просто запускаю в консоле

[Maxwell012]

Зачем ты вообще так криво стал делать, надо просто запустить dp.start_polling как рекомендовано в aiogram, а start_notifier перед этим запускать через dp.startup.register() или просто как create_task, не нужен тебе тут gather
ЗЫ ну и сразу разберись с webhook, прод боты лучше уж через него запускать

Notifier - это код который исполняется раз в 6м, пауза на 6м сделано с помощью await asyncio.sleep(300), пофакту это получается закольцованный код который должен исполняться с ботом в одной петле, поэтому я их и вызывал с помощью гейзер чтобы они могли передавать исполнение между собой

[Everything_is_bad]

Maxwell012, описанное мной поведение тоже вызывает Notifier в этом же eventloop что и start_polling. А ты на ровном месте сделал непонятные усложнения, после чего получил проблемы.

[Maxwell012]

Everything_is_bad, Вы правы, я написал криво, я переписал код:

async def setup_services(bot: Bot):
    # CREATING A CONNECTION POOL IN DATABASE
    await db.create_pool()

    # COMMANDS
    await set_commands(bot=bot)

    # LAUNCH NOTIFIER
    notifier = Notifier(bot=bot)
    await notifier.start_notifier()


async def on_startup(bot: Bot):
    await bot.set_webhook(f'{BASE_WEBHOOK_URL}{WEBHOOK_PATH}')
    await setup_services(bot=bot)


async def last_message(bot: Bot):
    await bot.send_message(chat_id=Config.ADMIN_ID, text="<b>BOT STOPPED</b>")
    await bot.session.close()


def main():
    # ALL STUFFS FOR BOT
    bot = Bot(token=Config.TOKEN_BOT, parse_mode=ParseMode.HTML)
    dp = Dispatcher(bot=bot)
    dp.startup.register(on_startup)
    app = web.Application()
    webhook_requests_handler = SimpleRequestHandler(dispatcher=dp, bot=bot)
    webhook_requests_handler.register(app, path=WEBHOOK_PATH)
    setup_application(app, dp, bot=bot)

    # REGISTER HANDLERS
    register_classes(dp=dp)
    basic.register_handlers(dp=dp)
    specialist.register_handlers(dp=dp)
    admin.register_handlers(dp=dp)

    try:
        web.run_app(app, host=WEB_SERVER_HOST, port=WEB_SERVER_PORT)
    except Exception as ex:
        main_logger.critical(f"Bot stopped due to exception: {ex}", exc_info=True)
        # await dp.start_polling(bot)  # restart the bot
    finally:
        asyncio.run(last_message(bot=bot))


if __name__ == "__main__":
    main()

но если мой нотифаер будет вызываться так как щас, то он просто тормозит главную петлю и код дальше не исполняется

[Everything_is_bad]

Maxwell012, ну так ты сам начал писать кривости, зачем ты сделал await notifier.start_notifier() а не asyncio.create_task(notifier.start_notifier())? Уже не первый раз замечаю не понимание между эти базовыми операциями в разных моментах кода, давай поработай над их пониманием.

[Maxwell012]

Everything_is_bad, Да, вы правы, я использовал просто create_task в основном в связке с asyncio.gather и у меня создалось впечатление что не create_task вызывает задачу а asyncio.gather, но благодаря вам я осознал это, благодарю)

Answer 2

[Everything_is_not_so_bad]

В добавок к замечанию насчёт best practices и корректного использования бота, для корректной обработки выхода из программы нужно добавить обработку сигналов SIGINT, SIGTERM - для плавного завершения работы программы. При получении сигнала нужно выйти из основного цикла.
https://stackoverflow.com/questions/1112343/how-do...