Merge pull request #5 from profcomff/copilot/fix-142d4adc-0389-484f-bf4b-7aecef7c32d2

Fix ban functionality to accept Telegram ID and username with collision detection

Author: MarkShidran

handlers/admin/ban_handlers.py

@@ -45,7 +45,11 @@ async def ban_list_add(message: types.Message):
     logger.info("Начало процесса добавления пользователя в бан.")
     await bot.send_message(
         message.from_user.id,
-        "Введите id пользователя, которого необходимо забанить:",
+        "Введите идентификатор пользователя, которого необходимо забанить:\n"
+        "• Внутренний ID базы данных (например: 123 или db:123)\n"
+        "• Telegram ID (например: 987654321)\n"
+        "• Username (например: @username)\n"
+        "\n💡 При совпадении ID используйте db:123 для точного указания внутреннего ID",
         reply_markup=admin_cancel_markup(),
     )
     await AdminData.user_ban.set()
@@ -119,7 +123,11 @@ async def ban_list_remove(message: types.Message):
     logger.info("Начало процесса вывода пользователя из бана.")
     await bot.send_message(
         message.from_user.id,
-        "Введите id пользователя, которого необходимо убрать из бан листа:",
+        "Введите идентификатор пользователя, которого необходимо убрать из бан листа:\n"
+        "• Внутренний ID базы данных (например: 123 или db:123)\n"
+        "• Telegram ID (например: 987654321)\n"
+        "• Username (например: @username)\n"
+        "\n💡 При совпадении ID используйте db:123 для точного указания внутреннего ID",
         reply_markup=admin_cancel_markup(),
     )
     await AdminData.user_unban.set()
@@ -168,7 +176,9 @@ async def comment_to_unban_answer(message: types.Message, state: FSMContext):
 async def save_to_unban(unbanned_user_id, comment):
     """Сохранение в БД, что пользователь выведен из бана."""
     with Session() as db_session:
-        db_session.query(BanList).filter(BanList.banned_user_id == unbanned_user_id).update(
+        db_session.query(BanList).filter(
+            BanList.banned_user_id == unbanned_user_id
+        ).update(
             {
                 "ban_status": 0,
                 "date_of_unban": datetime.date.today(),

handlers/admin/validators/validators.py

@@ -4,7 +4,7 @@
 from sqlalchemy import exists
 
 from controllerBD.db_loader import Session
-from controllerBD.models import Users
+from controllerBD.models import Username, Users
 from handlers.user.ban_check import check_id_in_ban_with_status
 from loader import bot, logger
 
@@ -17,37 +17,136 @@ async def comment_validator(text):
 
 
 async def ban_validator(message: types.Message):
-    """Валидация id пользователя для добавления в бан."""
-    if re.fullmatch(r"^\d{1,10}$", message.text):
-        if await check_id_in_base(message.text):
-            if not await check_id_in_ban_with_status(message.text, 1):
-                logger.info("Валидация пройдена.")
-                return True
-            await bot.send_message(message.from_user.id, "Пользователь уже забаннен.")
-            return False
+    """Валидация пользователя для добавления в бан."""
+    success, db_id, result_message = await resolve_user_input_to_db_id(message.text)
+
+    if not success:
+        await bot.send_message(message.from_user.id, result_message)
+        return False
+
+    # Проверяем, не забаннен ли уже пользователь
+    if await check_id_in_ban_with_status(db_id, 1):
         await bot.send_message(
-            message.from_user.id, "Пользователя с таким id не существует."
+            message.from_user.id, f"Пользователь уже забаннен. {result_message}"
         )
         return False
-    await bot.send_message(message.from_user.id, "Неверный ввод, введите число.")
-    return False
+
+    logger.info(f"Валидация пройдена. {result_message}")
+    # Сохраняем найденный DB ID в message.text для дальнейшего использования
+    message.text = str(db_id)
+    return True
 
 
 async def unban_validator(message: types.Message):
-    """Валидация id пользователя для вывода из бана."""
-    if re.fullmatch(r"^\d{1,10}$", message.text):
-        if await check_id_in_base(message.text):
-            if await check_id_in_ban_with_status(message.text, 1):
-                logger.info("Валидация пройдена.")
-                return True
-            await bot.send_message(message.from_user.id, "Пользователь не забаннен.")
-            return False
+    """Валидация пользователя для вывода из бана."""
+    success, db_id, result_message = await resolve_user_input_to_db_id(message.text)
+
+    if not success:
+        await bot.send_message(message.from_user.id, result_message)
+        return False
+
+    # Проверяем, забаннен ли пользователь
+    if not await check_id_in_ban_with_status(db_id, 1):
         await bot.send_message(
-            message.from_user.id, "Пользователя с таким id не существует."
+            message.from_user.id, f"Пользователь не забаннен. {result_message}"
         )
         return False
-    await bot.send_message(message.from_user.id, "Неверный ввод, введите число.")
-    return False
+
+    logger.info(f"Валидация пройдена. {result_message}")
+    # Сохраняем найденный DB ID в message.text для дальнейшего использования
+    message.text = str(db_id)
+    return True
+
+
+async def resolve_user_input_to_db_id(user_input: str):
+    """
+    Пытается определить внутренний ID пользователя по различным типам ввода:
+    - Внутренний ID базы данных (числовой)
+    - Telegram ID (числовой, но больше)
+    - Username (начинается с @)
+    - Явный внутренний ID (db:123) для разрешения коллизий
+
+    Возвращает кортеж (success: bool, db_id: int | None, message: str)
+    """
+    user_input = user_input.strip()
+
+    with Session() as db_session:
+        # Случай 1: Ввод - это username (начинается с @)
+        if user_input.startswith("@"):
+            username = user_input[1:]  # Убираем @
+            user_record = (
+                db_session.query(Username).filter(Username.username == username).first()
+            )
+            if user_record:
+                return (
+                    True,
+                    user_record.id,
+                    f"Найден пользователь с username @{username}",
+                )
+            return False, None, f"Пользователь с username @{username} не найден"
+
+        # Случай 2: Явное указание внутреннего ID (db:123)
+        if user_input.startswith("db:"):
+            try:
+                db_id = int(user_input[3:])
+                internal_user = (
+                    db_session.query(Users).filter(Users.id == db_id).first()
+                )
+                if internal_user:
+                    return (
+                        True,
+                        db_id,
+                        f"Найден пользователь по явно указанному внутреннему ID: {internal_user.name}",
+                    )
+                return False, None, f"Пользователь с внутренним ID {db_id} не найден"
+            except ValueError:
+                return False, None, "Неверный формат после 'db:'. Используйте db:123"
+
+        # Случай 3: Числовой ввод (может быть коллизия)
+        if re.fullmatch(r"^\d{1,15}$", user_input):
+            user_id = int(user_input)
+
+            # Проверяем оба типа ID одновременно для обнаружения коллизий
+            internal_user = db_session.query(Users).filter(Users.id == user_id).first()
+            telegram_user = (
+                db_session.query(Users).filter(Users.teleg_id == user_id).first()
+            )
+
+            # Случай 1: Найден только по внутреннему ID
+            if internal_user and not telegram_user:
+                return (
+                    True,
+                    user_id,
+                    f"Найден пользователь по внутреннему ID: {internal_user.name}",
+                )
+
+            # Случай 2: Найден только по Telegram ID
+            if telegram_user and not internal_user:
+                return (
+                    True,
+                    telegram_user.id,
+                    f"Найден пользователь по Telegram ID: {telegram_user.name}",
+                )
+
+            # Случай 3: КОЛЛИЗИЯ - найден и по внутреннему ID, и по Telegram ID
+            if internal_user and telegram_user:
+                # Приоритет отдаем Telegram ID, так как это чаще используется админами
+                return (
+                    True,
+                    telegram_user.id,
+                    f"⚠️ ВНИМАНИЕ: ID {user_id} найден у двух пользователей! "
+                    f"Выбран по Telegram ID: {telegram_user.name}. "
+                    f"Для точности используйте @username или укажите 'db:{user_id}' для внутреннего ID.",
+                )
+
+            # Случай 4: Не найден ни по одному типу ID
+            return (
+                False,
+                None,
+                f"Пользователь с ID {user_id} не найден ни в внутренней базе, ни среди Telegram ID",
+            )
+
+        return False, None, "Неверный формат ввода. Введите число (ID) или username с @"
 
 
 async def check_id_in_base(user_id):