forked from Muzifs/LGBot
Добавление команды /warn
This commit is contained in:
@@ -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 дней (множественные нарушения).'
|
||||
|
||||
}
|
||||
@@ -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()
|
||||
@@ -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"),
|
||||
|
||||
244
src/modules/warn.py
Normal file
244
src/modules/warn.py
Normal file
@@ -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)
|
||||
Reference in New Issue
Block a user