telegram-application/Dockerfile

@@ -28,4 +28,4 @@ RUN poetry install --no-root
 COPY . .
 
 # Команда для запуска приложения
-CMD ["poetry", "run", "python", "src/main.py"]
+CMD ["poetry", "run", "uvicorn", "src.main:app"]

telegram-application/src/main.py

@@ -1,3 +1,7 @@
+from contextlib import asynccontextmanager
+
+from fastapi import FastAPI
+
 from pyrogram import Client, filters, idle
 from pyrogram.handlers import MessageHandler
 
@@ -6,22 +10,26 @@ from src.core.tg_service.messages_handler import message_listener
 from src.core.rabbitmq.connect import broker, init_queue_and_publisher
 
 
-app = Client(
+tg_app = Client(
     name=settings.ACCOUNT.NAME,
     api_id=settings.ACCOUNT.API_ID,
     api_hash=settings.ACCOUNT.API_HASH,
 )
 
-app.add_handler(MessageHandler(
+
+@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(
     callback=message_listener,
     filters=filters.all
 ))
 
-async def main():
-    await app.start()
-    await broker.start()
-    await init_queue_and_publisher()
-    await idle()
-    await app.stop()
-
-app.run(main())
+app = FastAPI(lifespan=lifespan)