init

shared/telegram/telegram_bot.py

@@ -0,0 +1,50 @@
+import asyncio
+import logging
+
+import pydantic
+from aiogram import Bot, Dispatcher, Router
+
+
+class TelegramConfig(pydantic.BaseModel):
+    TOKEN: str
+    DROP_PENDING_UPDATES: bool = True
+
+
+class TelegramHelper:
+    def __init__(self, config: TelegramConfig, *routers: Router) -> None:
+        self.bot: Bot = Bot(token=config.TOKEN)
+        self.dispatcher: Dispatcher = Dispatcher()
+        self._polling_task: asyncio.Task | None = None
+        self._drop_pending_updates: bool = config.DROP_PENDING_UPDATES
+
+        for router in routers:
+            self.dispatcher.include_router(router)
+
+        logging.info('Telegram bot initialized')
+
+    async def start_polling(self) -> None:
+        """Start bot polling in background"""
+        if self._polling_task is not None:
+            logging.warning('Bot polling already started')
+            return
+
+        self._polling_task = asyncio.create_task(
+            self.dispatcher.start_polling(self.bot, drop_pending_updates=self._drop_pending_updates)
+        )
+
+        logging.info('Telegram bot polling started')
+
+    async def stop_polling(self) -> None:
+        """Stop bot polling and cleanup"""
+        if self._polling_task is None:
+            logging.warning('Bot polling not started')
+            return
+
+        self._polling_task.cancel()
+
+        if self.bot.session is not None:
+            await self.bot.session.close()
+
+        self._polling_task = None
+
+        logging.info('Telegram bot stopped')