Добавление команды /warn

src/config.py

@@ -22,6 +22,7 @@ COMMAND_MESSAGES = {
         "• <code>/help</code> - Этот справочник\n"
         "• <code>/log</code> - Инструкция по созданию логов\n\n"
         "<u>🛠 Команды модерации:</u>\n"
+        "• <code>/warn help</code> - Выдать предупреждение\n"
         "• <code>/mute help</code> - Инструкция по муту\n"
         "• <code>/unmute help</code> - Снятие мута\n"
         "• <code>/ban help</code> - Инструкция по бану\n"
@@ -135,6 +136,25 @@ COMMAND_MESSAGES = {
         "3. По ID пользователя:\n"
         "   <code>/reset_violations 123456789</code>\n\n"
         "<i>ℹ️ Сбрасывает все записи об автомутах пользователя</i>"
-    )
+    ),
+    'manual_warn': (
+        "<b>⚠️ Команда /warn</b>\n\n"
+        "<i>Выдает официальное предупреждение пользователю</i>\n\n"
+        "<u>🎯 Способы использования:</u>\n"
+        "1. Ответ на сообщение:\n"
+        "   <code>/warn причина</code>\n"
+        "2. По тегу пользователя:\n"
+        "   <code>/warn @username причина</code>\n"
+        "3. По ID пользователя:\n"
+        "   <code>/warn 123456789 причина</code>\n\n"
+        "<b>📋 Система накопления:</b>\n"
+        "• 1-й варн: просто предупреждение\n"
+        "• 2-й варн за неделю: автомут на сутки\n"
+        "• Повтор в течение 2 недель: мут на неделю\n\n"
+        "<i>ℹ️ Причину обязательно указывайте для прозрачности</i>"
+    ),
+    'warned': '⚠️ Пользователь получил предупреждение.',
+    'warned_auto_mute_day': '⚠️ Пользователь получил предупреждение и автомут на 1 день (повторное нарушение за неделю).',
+    'warned_auto_mute_week': '⚠️ Пользователь получил предупреждение и автомут на 7 дней (множественные нарушения).'
 
 }

src/database.py

@@ -34,6 +34,17 @@ class Database: # Инициализация класса
                     FOREIGN KEY (user_id) REFERENCES users (id)
                 )
             ''')
+            cursor.execute('''
+                CREATE TABLE IF NOT EXISTS warnings (
+                    id INTEGER PRIMARY KEY AUTOINCREMENT,
+                    user_id INTEGER NOT NULL,
+                    chat_id INTEGER NOT NULL,
+                    warn_date INTEGER NOT NULL,
+                    reason TEXT NOT NULL,
+                    admin_id INTEGER NOT NULL,
+                    FOREIGN KEY (user_id) REFERENCES users (id)
+                )
+            ''')
             connect.commit()
     
     # Возвращает соединение с базой данных
@@ -156,5 +167,56 @@ class Database: # Инициализация класса
             logger.info(f"Сброшено {deleted_count} нарушений пользователя {user_id} в чате {chat_id}")
             return deleted_count
 
+    # Добавляет предупреждение в базу данных
+    def add_warning(self, user_id: int, chat_id: int, reason: str, admin_id: int):
+        import time
+        with self._get_connection() as connect:
+            cursor = connect.cursor()
+            cursor.execute('''
+                INSERT INTO warnings (user_id, chat_id, warn_date, reason, admin_id)
+                VALUES (?, ?, ?, ?, ?)
+            ''', (user_id, chat_id, int(time.time()), reason, admin_id))
+            connect.commit()
+            logger.info(f"Предупреждение пользователю {user_id} выдано администратором {admin_id} в чате {chat_id}")
+
+    # Получает количество предупреждений за период
+    def get_warnings_count(self, user_id: int, chat_id: int, period_seconds: int) -> int:
+        import time
+        with self._get_connection() as connect:
+            cursor = connect.cursor()
+            cutoff_time = int(time.time()) - period_seconds
+            cursor.execute('''
+                SELECT COUNT(*)
+                FROM warnings
+                WHERE user_id = ? AND chat_id = ? AND warn_date > ?
+            ''', (user_id, chat_id, cutoff_time))
+            result = cursor.fetchone()
+            return result[0] if result else 0
+
+    # Получает все предупреждения пользователя
+    def get_user_warnings(self, user_id: int, chat_id: int):
+        with self._get_connection() as connect:
+            cursor = connect.cursor()
+            cursor.execute('''
+                SELECT id, warn_date, reason, admin_id
+                FROM warnings
+                WHERE user_id = ? AND chat_id = ?
+                ORDER BY warn_date DESC
+            ''', (user_id, chat_id))
+            return cursor.fetchall()
+
+    # Сбрасывает все предупреждения пользователя в чате
+    def reset_user_warnings(self, user_id: int, chat_id: int):
+        with self._get_connection() as connect:
+            cursor = connect.cursor()
+            cursor.execute('''
+                DELETE FROM warnings
+                WHERE user_id = ? AND chat_id = ?
+            ''', (user_id, chat_id))
+            deleted_count = cursor.rowcount
+            connect.commit()
+            logger.info(f"Сброшено {deleted_count} предупреждений пользователя {user_id} в чате {chat_id}")
+            return deleted_count
+
 # Создаем экземпляр базы данных для импорта в других модулях
 db = Database()

src/main.py

@@ -141,6 +141,7 @@ async def setup_bot_commands():
         BotCommand("start", "Начало работы с ботом"),
         BotCommand("help", "Справка по всем командам"),
         BotCommand("log", "Инструкция по созданию лога ошибки"),
+        BotCommand("warn", "Выдать предупреждение. Использование: /warn help"),
         BotCommand("ban", "Забанить пользователя. Использование: /ban help"),
         BotCommand("unban", "Разбанить пользователя. Использование: /unban help"),
         BotCommand("mute", "Замутить пользователя. Использование: /mute help"),

src/modules/warn.py

@@ -0,0 +1,244 @@
+from telebot.async_telebot import AsyncTeleBot
+from telebot.types import Message, User, ChatPermissions
+import logging
+import time
+
+from database import db
+from action_reporter import action_reporter
+from utils import (
+    delete_messages,
+    check_admin_status,
+    check_target_status,
+)
+
+from config import COMMAND_MESSAGES
+
+# Получаем логгер для текущего модуля
+logger = logging.getLogger(__name__)
+
+# Регистрирует все обработчики команд
+def register_handlers(bot: AsyncTeleBot):
+
+    # Обработчик команды /warn
+    @bot.message_handler(commands=['warn'])
+    async def _warn_command_wrapper(message: Message):
+        await warn_command(bot, message)
+
+# Основная функция команды /warn
+async def warn_command(bot: AsyncTeleBot, message: Message):
+
+    # Определяем целевого пользователя
+    target_user = None
+
+    # Определяем причину
+    reason = None
+
+    # Разбиваем текст сообщения на части
+    parts_msg = message.text.split()
+
+    # Команда /warn help
+    if len(parts_msg) == 2 and parts_msg[1].strip() in ('help', 'помощь'):
+
+        # Отправляем инструкцию
+        await bot.send_message(
+            chat_id=message.chat.id,
+            text=COMMAND_MESSAGES['manual_warn'],
+            message_thread_id=message.message_thread_id,
+        )
+
+        # Удаляем сообщения через 30 секунд
+        await delete_messages(bot, message, time_sleep=30, number_message=2)
+        return
+
+    try:
+
+        # Проверяем, является ли отправитель администратором с правом ограничения
+        if await check_admin_status(bot, message) == 1:
+            return
+
+        # Если одно слово (/warn)
+        if len(parts_msg) == 1:
+
+            # Удаляем сообщение через 3 секунды
+            await delete_messages(bot, message, time_sleep=3, number_message=1)
+            return
+
+        # Команда через ответ на сообщение, если два или более слов (/warn причина)
+        if message.reply_to_message and (not message.is_topic_message or message.message_thread_id != message.reply_to_message.message_id):
+
+            # Собираем данные
+            target_user = message.reply_to_message.from_user
+            reason = ' '.join(parts_msg[1:]) if len(parts_msg) > 1 else 'отсутствует'
+
+        # Если второе слово это тег или ID
+        elif len(parts_msg) >= 2 and (parts_msg[1].strip().isdigit() or parts_msg[1].startswith('@')):
+
+            # Собираем данные
+            identifier = parts_msg[1].strip()
+            reason = ' '.join(parts_msg[2:]) if len(parts_msg) > 2 else 'отсутствует'
+
+            # Делаем поиск по ID
+            if identifier.isdigit():
+
+                # Ищем пользователя в базе данных
+                user_info = db.get_user(int(identifier))
+
+                # Если нашли пользователя
+                if user_info:
+
+                    # Создаем объект пользователя
+                    target_user = User(
+                        id=user_info[0],
+                        first_name=user_info[1],
+                        username=user_info[2],
+                        is_bot=False
+                    )
+
+            # Делаем поиск по тегу
+            elif identifier.startswith('@'):
+
+                # Ищем пользователя в базе данных (убрали @)
+                user_info = db.get_user_by_username(identifier[1:])
+
+                # Если нашли пользователя
+                if user_info:
+
+                    # Создаем объект пользователя
+                    target_user = User(
+                        id=user_info[0],
+                        first_name=user_info[1],
+                        username=user_info[2],
+                        is_bot=False
+                    )
+
+        # Если команда неправильная
+        else:
+
+            # Удаляем сообщение через 3 секунды
+            await delete_messages(bot, message, time_sleep=3, number_message=1)
+            return
+
+        # Если пользователь не найден в базе данных
+        if not target_user:
+
+            # Отправляем предупреждение
+            await bot.send_message(
+                chat_id=message.chat.id,
+                text=COMMAND_MESSAGES['user_not_found'],
+                message_thread_id=message.message_thread_id,
+            )
+
+            # Удаляем сообщения через 5 секунд
+            await delete_messages(bot, message, time_sleep=5, number_message=2)
+            return
+
+        # Проверяем статус целевого пользователя
+        if await check_target_status(bot, message, target_user) == 1:
+            return
+
+        # Добавляем предупреждение в БД
+        db.add_warning(
+            user_id=target_user.id,
+            chat_id=message.chat.id,
+            reason=reason,
+            admin_id=message.from_user.id
+        )
+
+        # Проверяем количество предупреждений
+        ONE_WEEK = 604800  # 7 дней в секундах
+        TWO_WEEKS = 1209600  # 14 дней в секундах
+
+        warns_week = db.get_warnings_count(target_user.id, message.chat.id, ONE_WEEK)
+        warns_two_weeks = db.get_warnings_count(target_user.id, message.chat.id, TWO_WEEKS)
+
+        logger.info(f"Предупреждений за неделю: {warns_week}, за 2 недели: {warns_two_weeks}")
+
+        # Определяем, нужно ли применять мут
+        mute_applied = False
+        mute_duration = 0
+        mute_duration_text = ""
+        response_message = COMMAND_MESSAGES['warned']
+
+        # Если это уже 2+ предупреждение за неделю -> мут на неделю
+        if warns_week >= 2:
+            mute_duration = 604800  # 7 дней
+            mute_duration_text = "7 дней"
+            response_message = COMMAND_MESSAGES['warned_auto_mute_week']
+            mute_applied = True
+            logger.info(f"Применен мут на неделю (предупреждений за неделю: {warns_week})")
+
+        # Если это 2-е предупреждение за 2 недели (но не за неделю) -> мут на сутки
+        elif warns_two_weeks >= 2:
+            mute_duration = 86400  # 1 день
+            mute_duration_text = "1 день"
+            response_message = COMMAND_MESSAGES['warned_auto_mute_day']
+            mute_applied = True
+            logger.info(f"Применен мут на сутки (предупреждений за 2 недели: {warns_two_weeks})")
+
+        # Применяем мут если нужно
+        if mute_applied:
+            try:
+                # Вычисляем время окончания мута
+                until_date = int(time.time()) + mute_duration
+
+                # Устанавливаем ограничения (только чтение)
+                permissions = ChatPermissions(
+                    can_send_messages=False,
+                    can_send_media_messages=False,
+                    can_send_polls=False,
+                    can_send_other_messages=False,
+                    can_add_web_page_previews=False,
+                    can_change_info=False,
+                    can_invite_users=False,
+                    can_pin_messages=False,
+                )
+
+                # Выполняем мут
+                await bot.restrict_chat_member(
+                    chat_id=message.chat.id,
+                    user_id=target_user.id,
+                    permissions=permissions,
+                    until_date=until_date
+                )
+
+                logger.info(f"Пользователь {target_user.id} получил автомут на {mute_duration_text} после варна")
+
+            except Exception as e:
+                logger.error(f"Ошибка при применении мута после варна: {str(e)}")
+
+        # Отправляем сообщение-лог в админ-чат
+        await action_reporter.log_action(
+            action="ВАРН" if not mute_applied else f"ВАРН + МУТ ({mute_duration_text})",
+            user_id=target_user.id,
+            admin_id=message.from_user.id,
+            reason=reason,
+            duration=mute_duration_text if mute_applied else None,
+        )
+
+        # Отправляем сообщение в чат
+        await bot.send_message(
+            chat_id=message.chat.id,
+            text=response_message,
+            message_thread_id=message.message_thread_id,
+        )
+
+        # Записываем действие в логи
+        logger.info(f"Администратор {message.from_user.id} выдал предупреждение пользователю {target_user.id}. Причина: {reason}")
+
+        # Удаляем сообщения через 5 секунд
+        await delete_messages(bot, message, time_sleep=5, number_message=2)
+
+    except Exception as e:
+
+        # Отправляем ошибку
+        await bot.send_message(
+            chat_id=message.chat.id,
+            text=COMMAND_MESSAGES['general_error'],
+            message_thread_id=message.message_thread_id,
+        )
+
+        # Записываем ошибку в логи
+        logger.error(f"Общая ошибка в warn_command: {str(e)}")
+
+        # Удаляем сообщения через 5 секунд
+        await delete_messages(bot, message, time_sleep=5, number_message=2)