miltu-bot refactor

bot/bot.py

@@ -0,0 +1,74 @@
+# bot/bot.py
+import logging
+from typing import Optional, Tuple 
+from telegram.constants import ParseMode
+
+from telegram.ext import (
+    Application,
+    ApplicationBuilder,
+    CommandHandler,
+    MessageHandler,
+    ChatMemberHandler,
+    filters,
+    Defaults
+)
+
+from .models import TelegramBot, BotConfig
+from .handlers import start, ping, echo, my_chat_member_update, error_handler
+
+logger = logging.getLogger(__name__)
+
+
+def _resolve_parse_mode(pm: Optional[str]):
+    """
+    Преобразуем строковое значение из BotConfig.parse_mode в объект ParseMode.
+    'None' -> None (без парсинга), иначе HTML/MarkdownV2.
+    """
+    if not pm or pm == "None":
+        return None
+    # Безопасно берём атрибут из ParseMode, по умолчанию HTML
+    return getattr(ParseMode, pm, ParseMode.HTML)
+
+
+def _get_active_bot_and_config() -> Tuple[TelegramBot, BotConfig]:
+    """
+    Возвращает (активный TelegramBot, его BotConfig).
+    Бросает понятные ошибки, если данных нет.
+    """
+    bot = TelegramBot.objects.filter(is_active=True).first()
+    if not bot:
+        raise RuntimeError(
+            "Нет активного TelegramBot (is_active=True). Создайте запись TelegramBot и включите её."
+        )
+
+    cfg = BotConfig.objects.filter(bot=bot).first()
+    if not cfg:
+        raise RuntimeError(
+            f"Для бота '{bot}' не найден BotConfig. Создайте связанную запись BotConfig."
+        )
+
+    return bot, cfg
+
+
+def build_application() -> Application:
+    tb, cfg = _get_active_bot_and_config()
+
+    parse_mode = _resolve_parse_mode(cfg.parse_mode)
+    allowed_updates = cfg.allowed_updates or None
+
+    defaults = Defaults(parse_mode=parse_mode) if parse_mode else None
+
+    builder: ApplicationBuilder = Application.builder().token(tb.token)
+    if defaults:
+        builder = builder.defaults(defaults)
+
+    app = builder.build()
+
+    app.add_handler(CommandHandler("start", start))
+    app.add_handler(CommandHandler("ping", ping))
+    app.add_handler(ChatMemberHandler(my_chat_member_update, ChatMemberHandler.MY_CHAT_MEMBER))
+    app.add_handler(MessageHandler(filters.TEXT & ~filters.COMMAND, echo))
+    app.add_error_handler(error_handler)
+
+    app.bot_data["allowed_updates"] = allowed_updates
+    return app