Support-BOT

handlers_commands.py

@@ -0,0 +1,195 @@
+from aiogram import Bot, F, Router
+from aiogram.exceptions import TelegramBadRequest
+from aiogram.filters import Command, CommandObject
+from aiogram.types import Message
+
+from app.bot.get_reports import get_report_from_db
+from app.bot.utils import get_user_name, check_input_date_correct, \
+    stringdate_to_date, check_user_is_banned, \
+    get_telegram_user_from_resend_message, parse_ban_command
+from app.core.config import settings
+from app.core.db import get_async_session
+from app.crud.user import crud_user
+from app.crud.message import crud_message
+
+router = Router()
+
+
+@router.message(Command(commands=["start"]))
+async def command_start(message: Message):
+    await message.answer(settings.START_MESSAGE)
+    session_generator = get_async_session()
+    session = await session_generator.__anext__()
+    db_user = await crud_user.get_or_create_user_by_tg_message(message,
+                                                               session)
+    if check_user_is_banned(db_user):
+        return
+
+
+@router.message(Command(commands="info"),
+                F.chat.id == int(settings.GROUP_ID),
+                F.reply_to_message)
+async def get_user_info(message: Message, bot: Bot):
+    session_generator = get_async_session()
+    session = await session_generator.__anext__()
+    telegram_user = await get_telegram_user_from_resend_message(message, bot)
+    if not telegram_user:
+        return
+    messages_count = await crud_message.get_count_user_messages(
+        telegram_user.id, session
+    )
+    ans_count = await crud_message.get_count_answers_to_user(
+        telegram_user.id, session
+    )
+    username = f"@{telegram_user.username}" if telegram_user.username else "отсутствует"
+    await message.reply(text=f'Имя: {get_user_name(telegram_user)}\n'
+                             f'Id: {telegram_user.id}\n'
+                             f'username: {username}\n'
+                             f'Сообщений от пользователя: {messages_count}\n'
+                             f'Ответов пользователю: {ans_count}\n')
+
+
+@router.message(Command(commands='report'),
+                F.chat.id == int(settings.GROUP_ID))
+async def get_report(message: Message,
+                     bot: Bot,
+                     command: CommandObject):
+    session_generator = get_async_session()
+    session = await session_generator.__anext__()
+    if command.args:
+        if not check_input_date_correct(command.args):
+            answer_text = 'Неверный формат даты'
+            await bot.send_message(
+                chat_id=int(settings.GROUP_ID), text=answer_text
+            )
+            return
+        from_date, to_date = stringdate_to_date(command.args)
+        report = await get_report_from_db(session, from_date, to_date)
+        report['period'] = 'выбранный период'
+    else:
+        report = await get_report_from_db(session)
+        report['period'] = 'последний месяц'
+
+    answer_text = (f"Отчет за {report['period']}, c {report['from_date']} до "
+                   f"{report['to_date']}.\n"
+                   f"Всего было получено {report['questions_amount']} "
+                   f"сообщений от {report['users_amount']} клиентов.\n"
+                   f"Количество ответов от администраторов: {report['answers_amount']}")
+    await bot.send_message(chat_id=int(settings.GROUP_ID), text=answer_text)
+
+
+@router.message(Command(commands='ban'),
+                F.chat.id == int(settings.GROUP_ID))
+async def handler_ban_user(message: Message,
+                           bot: Bot,
+                           command: CommandObject):
+    if not command.args and not message.reply_to_message:
+        return await message.reply(
+            text='Команда некорректна. Укажите ID или ответьте на сообщение')
+    session_generator = get_async_session()
+    session = await session_generator.__anext__()
+    if command.args:
+        telegram_user_id = parse_ban_command(command)
+        if not telegram_user_id:
+            return await message.reply(
+                text='Невозможно извлечь id пользователя. '
+                     'Нужно ввести id в формате "12345", либо ответить на '
+                     'сообщение пользователя, которого хотите забанить')
+        try:
+            telegram_user = await bot.get_chat(telegram_user_id)
+        except TelegramBadRequest:
+            return await message.reply(
+                text='Пользователя с таким id не существует ')
+
+    else:
+        telegram_user = await get_telegram_user_from_resend_message(message, bot)
+        if not telegram_user:
+            return
+    db_user = await crud_user.get_user_by_telegram_id(telegram_user.id,
+                                                      session)
+    await crud_user.ban_user(db_user, session)
+    await message.reply(text=f'Пользователь {db_user.first_name} '
+                             f'{db_user.last_name} забанен.'
+                             f'Чтобы разбанить, отправьте /unban\n'
+                             f'Тикет: #id{db_user.telegram_id}')
+
+
+@router.message(Command(commands='unban'),
+                F.chat.id == int(settings.GROUP_ID))
+async def handler_unban_user(message: Message,
+                             bot: Bot,
+                             command: CommandObject):
+    if not command.args and not message.reply_to_message:
+        return await message.reply(
+            text='Команда некорректна. Укажите ID или ответьте на сообщение')
+    session_generator = get_async_session()
+    session = await session_generator.__anext__()
+    if command.args:
+        telegram_user_id = parse_ban_command(command)
+        if not telegram_user_id:
+            return await message.reply(
+                text='Невозможно извлечь id пользователя. '
+                     'Нужно ввести id в формате "12345", либо ответить на '
+                     'сообщение пользователя, которого хотите забанить')
+        db_user = await crud_user.get_user_by_telegram_id(telegram_user_id,
+                                                          session)
+    else:
+        telegram_user = await get_telegram_user_from_resend_message(message, bot)
+        if not telegram_user:
+            return
+        db_user = await crud_user.get_user_by_telegram_id(telegram_user.id,
+                                                          session)
+    await crud_user.unban_user(db_user, session)
+    await message.reply(text=f'Пользователь с id '
+                             f'{db_user.first_name} {db_user.last_name} разбанен\n'
+                             f'Тикет: #id{db_user.telegram_id}')
+
+
+@router.message(Command(commands='banlist'),
+                F.chat.id == int(settings.GROUP_ID))
+async def handler_unban_user(message: Message,
+                             bot: Bot):
+    session_generator = get_async_session()
+    session = await session_generator.__anext__()
+    banned_users = await crud_user.get_banned_users(session)
+    text = 'Список забанненых пользователей:\n'
+    for user in banned_users:
+        text += f'{user.telegram_id} - {user.first_name} {user.last_name}\n'
+    await message.reply(text=text)
+
+
+@router.message(Command(commands='registeradmin'),
+                F.chat.id == int(settings.GROUP_ID))
+async def handle_register_admin(message: Message,
+                                bot: Bot):
+    if not message.reply_to_message:
+        return message.reply(text="Введите команду как ответ на сообщение "
+                                  "пользователя")
+    session_generator = get_async_session()
+    session = await session_generator.__anext__()
+    db_user = await crud_user.get_or_create_user_by_tg_message(
+        message.reply_to_message,
+        session
+    )
+    db_user = await crud_user.register_admin(db_user, session)
+    text = (f'Пользователь {db_user.first_name} {db_user.last_name} теперь '
+            f'администратор')
+    await message.reply(text=text)
+
+
+@router.message(Command(commands='deleteadmin'),
+                F.chat.id == int(settings.GROUP_ID))
+async def handle_remove_admin(message: Message,
+                              bot: Bot):
+    if not message.reply_to_message:
+        return message.reply(text="Введите команду как ответ на сообщение "
+                                  "пользователя")
+    session_generator = get_async_session()
+    session = await session_generator.__anext__()
+    db_user = await crud_user.get_or_create_user_by_tg_message(
+        message.reply_to_message,
+        session
+    )
+    db_user = await crud_user.remove_admin(db_user, session)
+    text = f'Администратор {db_user.first_name} {db_user.last_name} удален'
+    await message.reply(text=text)