Добавлена запись группы, для пересылки сообщений.

Author: PyRusAdmin

database/database.py

@@ -41,6 +41,19 @@ class Meta:
     return Keywords  # Возвращаем класс модели
 
 
+def create_group_model(user_id):
+    class Group(Model):
+        """Модель для хранения технической группы"""
+        id = AutoField()  # <-- добавляем первичный ключ (иначе всё пишется в одну строку)
+        user_group = CharField(unique=True)  # Поле для хранения технической группы
+
+        class Meta:
+            database = db  # Указываем, что модель использует базу данных
+            table_name = f"group_{user_id}"  # Имя таблицы
+
+    return Group  # Возвращаем класс модели
+
+
 # Создаём таблицы при первом запуске
 db.connect()
 db.create_tables([User], safe=True)

handlers/connect_group.py

@@ -0,0 +1,62 @@
+# -*- coding: utf-8 -*-
+from aiogram import F
+from aiogram.fsm.context import FSMContext
+from aiogram.types import Message
+from loguru import logger
+
+from database.database import User, create_group_model
+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 handle_connect_message_group(message: Message, state: FSMContext):
+    """Ввод username группы для сообщений"""
+    user_tg = message.from_user
+    user = User.get(User.user_id == user_tg.id)
+
+    logger.info(
+        f"Пользователь {user_tg.id} {user_tg.username} {user_tg.first_name} {user_tg.last_name} перешел в меню Подключить группу для сообщений")
+
+    await message.answer(
+        get_text(user.language, "enter_group"),
+        reply_markup=back_keyboard()  # клавиатура назад
+    )
+    await state.set_state(MyStates.entering_group)
+
+
+@router.message(MyStates.entering_group)
+async def handle_group_username_submission(message: Message, state: FSMContext):
+    """Обработка введённого ключевого слова, словосочетания"""
+
+    group_username = message.text.strip()
+    user_tg = message.from_user
+    logger.info(f"Пользователь ввёл ссылку: {group_username}")
+
+    # Создаём модель с таблицей, уникальной для конкретного пользователя
+    GroupModel = create_group_model(user_id=user_tg.id)  # Создаём таблицу для групп / ключевых слов
+
+    # Проверяем, существует ли таблица (если нет — создаём)
+    if not GroupModel.table_exists():
+        GroupModel.create_table()
+        logger.info(f"Создана новая таблица для пользователя {user_tg.id}")
+
+    # Добавляем запись в таблицу
+    try:
+        group_record = GroupModel.create(user_group=group_username)
+        await message.answer(f"✅ Группа {group_username} добавлена для отправки сообщений.")
+        logger.info(f"username {group_username} добавлено пользователем {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()  # Очищаем состояние
+
+
+def register_entering_group_handler():
+    """Регистрация обработчиков"""
+    router.message.register(handle_connect_message_group)  # Регистрация обработчика

keyboards/keyboards.py

@@ -42,10 +42,9 @@ def settings_keyboard():
     return ReplyKeyboardMarkup(
         keyboard=[
             [KeyboardButton(text="🔁 Обновить список")],
-
             [KeyboardButton(text="Ввод ключевого слова")],
-
             [KeyboardButton(text="Подключить аккаунт")],
+            [KeyboardButton(text="Подключить группу для сообщений")],
             [KeyboardButton(text="Сменить язык")],
             [KeyboardButton(text="🔙 Назад")]
         ],

locales/locales.py

@@ -36,6 +36,9 @@
         "enter_keyword": (
             "🔍 Введите ключевое слово / словосочетание для отслеживания"
         ),
+        "enter_group": (
+            "🔍 Введите ссылку на группу в формате @username в которую будет пересылаться сообщение обнаруженное по ключевому слову"
+        ),
     },
 
     "en": {
@@ -73,6 +76,9 @@
         "enter_keyword": (
             "🔍 Enter a keyword / phrase to track"
         ),
+        "enter_group": (
+            "🔍 Enter a link to the group in the format @username to which the message will be forwarded when a keyword is detected"
+        ),
     }
 }
 

main.py

@@ -5,6 +5,7 @@
 
 from loguru import logger
 
+from handlers.connect_group import register_entering_group_handler
 from handlers.entering_keyword import register_entering_keyword_handler
 from handlers.handlers import register_greeting_handlers
 from system.dispatcher import dp, bot
@@ -21,6 +22,8 @@ async def main() -> None:
 
     register_entering_keyword_handler()  # Регистрация обработчика для ввода и записи в БД ключевых слов
 
+    register_entering_group_handler()  # Регистрация обработчика для ввода и записи в БД групп (техническая группа)
+
     await dp.start_polling(bot)
 
 

states/states.py

@@ -6,3 +6,4 @@ class MyStates(StatesGroup):
     """Прием ссылки на группу / канал"""
     waiting_username_group = State()  # ожидание ввода имени группы в формате @username
     entering_keyword = State()  # ожидание ввода ключевого слова
+    entering_group = State()  # ожидание ввода группы в формате @username (техническая)