From b1ee865124b3254dccf74d2c86ce4b141134be97 Mon Sep 17 00:00:00 2001 From: harold Date: Mon, 21 Apr 2025 17:27:45 +0500 Subject: [PATCH] fix main --- telegram-application/src/main.py | 28 ++++++++++------------------ 1 file changed, 10 insertions(+), 18 deletions(-) diff --git a/telegram-application/src/main.py b/telegram-application/src/main.py index faa8dd4..c01e0b6 100644 --- a/telegram-application/src/main.py +++ b/telegram-application/src/main.py @@ -1,7 +1,3 @@ -from contextlib import asynccontextmanager - -from fastapi import FastAPI - from pyrogram import Client, filters, idle from pyrogram.handlers import MessageHandler @@ -10,26 +6,22 @@ from src.core.tg_service.messages_handler import message_listener from src.core.rabbitmq.connect import broker, init_queue_and_publisher -tg_app = Client( +app = Client( name=settings.ACCOUNT.NAME, api_id=settings.ACCOUNT.API_ID, api_hash=settings.ACCOUNT.API_HASH, ) - -@asynccontextmanager -async def lifespan(fastapi_app: FastAPI): - await tg_app.start() - await broker.start() - await init_queue_and_publisher() - await idle() - yield - await tg_app.stop() - - -tg_app.add_handler(MessageHandler( +app.add_handler(MessageHandler( callback=message_listener, filters=filters.all )) -app = FastAPI(lifespan=lifespan) +async def main(): + await app.start() + await broker.start() + await init_queue_and_publisher() + await idle() + await app.stop() + +app.run(main())