Отладка записи слов для отслеживания

Author: PyRusAdmin

handlers/entering_keyword.py

@@ -5,20 +5,20 @@
 from loguru import logger
 
 from database.database import User, create_keywords_model
-from keyboards.keyboards import (back_keyboard)
+from keyboards.keyboards import back_keyboard
 from locales.locales import get_text
 from states.states import MyStates
 from system.dispatcher import router
 
 
 @router.message(F.text == "Ввод ключевого слова")
-async def entering_keyword(message: Message, state: FSMContext):
+async def handle_enter_keyword_menu(message: Message, state: FSMContext):
     """Ввод ключевого слова"""
-    user_tg = message.from_user
-    user = User.get(User.user_id == user_tg.id)
+    telegram_user = message.from_user
+    user = User.get(User.user_id == telegram_user.id)
 
     logger.info(
-        f"Пользователь {user_tg.id} {user_tg.username} {user_tg.first_name} {user_tg.last_name} перешел в меню Ввод ключевого слова")
+        f"Пользователь {telegram_user.id} {telegram_user.username} {telegram_user.first_name} {telegram_user.last_name} перешел в меню Ввод ключевого слова")
 
     await message.answer(
         get_text(user.language, "enter_keyword"),
@@ -28,35 +28,95 @@ async def entering_keyword(message: Message, state: FSMContext):
 
 
 @router.message(MyStates.entering_keyword)
-async def handle_keyword(message: Message, state: FSMContext):
+async def handle_keywords_submission(message: Message, state: FSMContext):
     """Обработка введённого ключевого слова, словосочетания"""
 
-    user_keyword = message.text.strip()
-    user_tg = message.from_user
-    logger.info(f"Пользователь ввёл ключевое слово: {user_keyword}")
+    raw_input = message.text.strip()
+    telegram_user = message.from_user
+    logger.info(f"Пользователь ввёл ключевое слово: {raw_input}")
+
+    keywords_list = [
+        keyword.strip()
+        for keyword in raw_input.split('\n')
+        if keyword.strip()
+    ]
+
+    if not keywords_list:
+        await message.answer("⚠️ Вы не указали ни одного ключевого слова.")
+        await state.clear()
+        return
 
     # Создаём модель с таблицей, уникальной для конкретного пользователя
-    Keywords = create_keywords_model(user_id=user_tg.id)  # Создаём таблицу для групп / ключевых слов
+    KeywordsModel = create_keywords_model(user_id=telegram_user.id)  # Создаём таблицу для групп / ключевых слов
 
     # Проверяем, существует ли таблица (если нет — создаём)
-    if not Keywords.table_exists():
-        Keywords.create_table()
-        logger.info(f"Создана новая таблица для пользователя {user_tg.id}")
-
-    # Добавляем запись в таблицу
-    try:
-        keywords = Keywords.create(user_keyword=user_keyword)
-        await message.answer(f"✅ Слово / словосочетание {user_keyword} добавлено в отслеживание.")
-        logger.info(f"Ключевое слово {user_keyword} добавлено пользователем {user_tg.id}")
-    except Exception as e:
-        if "UNIQUE constraint failed" in str(e):
-            await message.answer("⚠️ Такое слово / словосочетание уже есть в отслеживаемых.")
-        else:
-            await message.answer("⚠️ Ошибка при добавлении слова / словосочетания.")
-        logger.error(f"Ошибка при добавлении ключевого слова: {e}")
-    await state.clear()  # Очищаем состояние
+    if not KeywordsModel.table_exists():
+        KeywordsModel.create_table()
+        logger.info(f"Создана новая таблица для пользователя {telegram_user.id}")
+
+    added_keywords = []
+    skipped_keywords = []
+    error_keywords = []
+
+    # Add each keyword one by one
+    for keyword in keywords_list:
+        try:
+            KeywordsModel.create(user_keyword=keyword)
+            added_keywords.append(keyword)
+        except Exception as e:
+            if "UNIQUE constraint failed" in str(e):
+                skipped_keywords.append(keyword)
+            else:
+                error_keywords.append((keyword, str(e)))
+                logger.error(f"Error adding keyword {keyword}: {e}")
+
+    # Format response message
+    response_parts = []
+
+    if added_keywords:
+        keywords_preview = added_keywords[:10]  # Show first 10
+        keywords_text = "\n".join(f"• {kw}" for kw in keywords_preview)
+        if len(added_keywords) > 10:
+            keywords_text += f"\n... и ещё {len(added_keywords) - 10}"
+        response_parts.append(
+            f"✅ Добавлено ключевых слов: {len(added_keywords)}\n{keywords_text}"
+        )
+
+    if skipped_keywords:
+        skipped_preview = skipped_keywords[:5]  # Show first 5
+        skipped_text = "\n".join(f"• {kw}" for kw in skipped_preview)
+        if len(skipped_keywords) > 5:
+            skipped_text += f"\n... и ещё {len(skipped_keywords) - 5}"
+        response_parts.append(
+            f"⚠️ Уже были добавлены ({len(skipped_keywords)}):\n{skipped_text}"
+        )
+
+    if error_keywords:
+        error_text = "\n".join(f"• {kw}: {err}" for kw, err in error_keywords[:3])
+        if len(error_keywords) > 3:
+            error_text += f"\n... и ещё {len(error_keywords) - 3} ошибок"
+        response_parts.append(f"❌ Ошибки при добавлении:\n{error_text}")
+
+    # Summary
+    summary = (
+        f"\n📊 Итого:\n"
+        f"• Добавлено: {len(added_keywords)}\n"
+        f"• Пропущено (дубликаты): {len(skipped_keywords)}\n"
+        f"• Ошибки: {len(error_keywords)}"
+    )
+    response_parts.append(summary)
+
+    await message.answer("\n\n".join(response_parts))
+
+    # Log statistics
+    logger.info(
+        f"Keywords import for user {telegram_user.id}: "
+        f"added={len(added_keywords)}, skipped={len(skipped_keywords)}, errors={len(error_keywords)}"
+    )
+
+    await state.clear()
 
 
 def register_entering_keyword_handler():
     """Регистрация обработчиков"""
-    router.message.register(entering_keyword)  # Регистрация обработчика
+    router.message.register(handle_enter_keyword_menu)  # Регистрация обработчика