75 lines
2.4 KiB
Python
75 lines
2.4 KiB
Python
# 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
|