73 lines
3.5 KiB
Python
73 lines
3.5 KiB
Python
from django.core.management.base import BaseCommand, CommandError
|
|
from telegram.ext import ApplicationBuilder, CommandHandler, MessageHandler, ChatMemberHandler, filters
|
|
from django.db import transaction
|
|
from bot.models import TelegramBot, BotConfig
|
|
from bot.handlers import on_my_chat_member, upsert_chat_from_update
|
|
|
|
async def cmd_start(update, context):
|
|
await upsert_chat_from_update(update, context)
|
|
await update.message.reply_text("Я на связи. Добавьте меня в группу — запомню чат автоматически.")
|
|
|
|
async def cmd_groups(update, context):
|
|
"""
|
|
/groups — покажем список чатов, где бот сейчас состоит.
|
|
(Лучше ограничить доступ: например, только admin_user_ids из BotConfig)
|
|
"""
|
|
bot_conf = await context.application.bot_data.get("bot_conf")
|
|
admin_ids = bot_conf.admin_user_ids if bot_conf else []
|
|
if update.effective_user and admin_ids and update.effective_user.id not in admin_ids:
|
|
return
|
|
|
|
from bot.models import TelegramChat
|
|
qs = TelegramChat.objects.filter(is_member=True).exclude(type="private").order_by("type", "title")
|
|
if not qs.exists():
|
|
await update.message.reply_text("Пока ни одной группы/канала не найдено.")
|
|
return
|
|
|
|
lines = []
|
|
for ch in qs:
|
|
label = ch.title or ch.username or ch.id
|
|
lines.append(f"• {label} [{ch.type}] — {ch.id}")
|
|
await update.message.reply_text("\n".join(lines)[:3900])
|
|
|
|
class Command(BaseCommand):
|
|
help = "Запуск телеграм-бота (polling) на основе активного TelegramBot"
|
|
|
|
def add_arguments(self, parser):
|
|
parser.add_argument("--bot-id", type=int, help="ID TelegramBot (если не указан — берём первый активный)")
|
|
|
|
def handle(self, *args, **options):
|
|
bot_id = options.get("bot_id")
|
|
|
|
try:
|
|
if bot_id:
|
|
bot = TelegramBot.objects.get(id=bot_id, is_active=True)
|
|
else:
|
|
bot = TelegramBot.objects.filter(is_active=True).first()
|
|
if not bot:
|
|
raise CommandError("Нет активного TelegramBot.")
|
|
except TelegramBot.DoesNotExist:
|
|
raise CommandError("Указанный TelegramBot не найден или неактивен.")
|
|
|
|
conf = getattr(bot, "config", None)
|
|
|
|
app = ApplicationBuilder().token(bot.token).build()
|
|
# Сохраним конфиг в bot_data, чтобы был доступ в хендлерах:
|
|
app.bot_data["bot_conf"] = conf
|
|
|
|
# Команды
|
|
app.add_handler(CommandHandler("start", cmd_start))
|
|
app.add_handler(CommandHandler("groups", cmd_groups))
|
|
|
|
# Обновление списка чатов при изменении статуса бота
|
|
app.add_handler(ChatMemberHandler(on_my_chat_member, ChatMemberHandler.MY_CHAT_MEMBER))
|
|
|
|
# Любые входящие сообщения — обновляем last_message_at для чатов
|
|
app.add_handler(MessageHandler(filters.ALL, upsert_chat_from_update))
|
|
|
|
# allowed_updates (если заданы в конфиге)
|
|
allowed_updates = conf.allowed_updates if conf and conf.allowed_updates else None
|
|
|
|
self.stdout.write(self.style.SUCCESS(f"Бот {bot} запущен (polling)."))
|
|
app.run_polling(allowed_updates=allowed_updates, drop_pending_updates=True)
|