added utils.py
1. Добавил message_thread_id для всех команд, убрав костыль, который отвечал за корректную отправку сообщений в топик. 2. Функции (определение администратора, удаление сообщений) вынес в utils.py, от куда они будут вызываться в командах. Модули стали более читаемы из-за уменьшения количества строк кода в них. 3. Дописал manual_unban и добавил error в config.py 4. Оптимизация
This commit is contained in:
@ -68,9 +68,11 @@ COMMAND_MESSAGES = {
|
||||
"<b>✅ Команда /unban</b>\n\n"
|
||||
"<i>Снимает бан с пользователя</i>\n\n"
|
||||
"<u>🎯 Способы использования:</u>\n"
|
||||
"1. По тегу пользователя:\n"
|
||||
"1. Ответ на сообщение:\n"
|
||||
" <code>/unban</code>\n"
|
||||
"2. По тегу пользователя:\n"
|
||||
" <code>/unban @username</code>\n"
|
||||
"2. По ID пользователя:\n"
|
||||
"3. По ID пользователя:\n"
|
||||
" <code>/unban 123456789</code>\n\n"
|
||||
"<b>⚠️ Работает только для забаненных через /ban</b>\n"
|
||||
"<i>ℹ️ Пользователь сможет снова присоединиться</i>"
|
||||
@ -87,6 +89,7 @@ COMMAND_MESSAGES = {
|
||||
'unmuted': '✅ Пользователь размучен.',
|
||||
'banned': '✅ Пользователь успешно забанен.',
|
||||
'unbanned': '✅ Пользователь успешно разбанен.',
|
||||
'error': '⚠️ Ошибка: {e}',
|
||||
'general_error': '⚠️ Произошла непредвиденная ошибка.'
|
||||
|
||||
}
|
||||
@ -5,27 +5,16 @@ import logging
|
||||
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 extract_reason(words: str) -> str:
|
||||
|
||||
if words == []:
|
||||
reason = 'отсутствует'
|
||||
else:
|
||||
reason = ' '.join(words)
|
||||
|
||||
return reason
|
||||
|
||||
# Удаляет два последних сообщения
|
||||
async def delete_messages(bot: AsyncTeleBot, message: Message, time_sleep: int):
|
||||
await asyncio.sleep(time_sleep)
|
||||
await bot.delete_message(message.chat.id, message.message_id)
|
||||
await bot.delete_message(message.chat.id, message.message_id+1)
|
||||
|
||||
# Регистрирует все обработчики команд
|
||||
def register_handlers(bot: AsyncTeleBot):
|
||||
|
||||
@ -37,10 +26,6 @@ def register_handlers(bot: AsyncTeleBot):
|
||||
# Основная функция команды /ban
|
||||
async def ban_command(bot: AsyncTeleBot, message: Message, photo_path: str = None):
|
||||
|
||||
# Отправка сообщения в тему или обычный чат
|
||||
send_message = bot.reply_to if message.is_topic_message else bot.send_message
|
||||
chat_id = message if message.is_topic_message else message.chat.id
|
||||
|
||||
# Определяем целевого пользователя
|
||||
target_user = None
|
||||
|
||||
@ -50,91 +35,81 @@ async def ban_command(bot: AsyncTeleBot, message: Message, photo_path: str = Non
|
||||
# Разбиваем текст сообщения на части
|
||||
parts_msg = message.text.split()
|
||||
|
||||
# Выводим помощь (/ban help)
|
||||
if len(parts_msg) == 2 and parts_msg[1].strip() in ['help', 'помощь']:
|
||||
await send_message(chat_id, COMMAND_MESSAGES['manual_ban'])
|
||||
# Команда /ban 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_ban'],
|
||||
message_thread_id=message.message_thread_id,
|
||||
)
|
||||
|
||||
# Удаляем сообщения через 30 секунд
|
||||
await delete_messages(bot, message, 30)
|
||||
|
||||
await delete_messages(bot, message, time_sleep=30, number_message=2)
|
||||
return
|
||||
|
||||
try:
|
||||
|
||||
# Проверяем, является ли отправитель администратором
|
||||
try:
|
||||
admin_status = await bot.get_chat_member(message.chat.id, message.from_user.id)
|
||||
|
||||
# Проверяем статус администратора (создателя)
|
||||
if admin_status.status not in ['administrator', 'creator']:
|
||||
await send_message(chat_id, COMMAND_MESSAGES['no_admin_rights'])
|
||||
|
||||
# Удаляем сообщения через 5 секунд
|
||||
await delete_messages(bot, message, 5)
|
||||
|
||||
return
|
||||
|
||||
# Проверяем право администратора на бан
|
||||
if admin_status.status == 'administrator' and not admin_status.can_restrict_members:
|
||||
await send_message(chat_id, COMMAND_MESSAGES['no_restrict_rights'])
|
||||
|
||||
# Удаляем сообщения через 5 секунд
|
||||
await delete_messages(bot, message, 5)
|
||||
|
||||
return
|
||||
|
||||
except Exception as e:
|
||||
await send_message(chat_id, f"⚠️ Ошибка: {str(e)}")
|
||||
logger.error(f"Ошибка при получении статуса администратора: {str(e)}")
|
||||
|
||||
# Удаляем сообщения через 5 секунд
|
||||
await delete_messages(bot, message, 5)
|
||||
|
||||
if await check_admin_status(bot, message) == 1:
|
||||
return
|
||||
|
||||
# Если одно слово, то ответом на сообщение
|
||||
# Команда через ответ на сообщение, если одно слово (/ban)
|
||||
if len(parts_msg) == 1:
|
||||
|
||||
# Если это тема
|
||||
# Если это топик
|
||||
if message.is_topic_message:
|
||||
|
||||
# Если без ответа на сообщение, ошибка
|
||||
# Если без ответа на сообщение
|
||||
if message.message_thread_id == message.reply_to_message.message_id:
|
||||
await asyncio.sleep(3)
|
||||
await bot.delete_message(message.chat.id, message.message_id)
|
||||
|
||||
# Удаляем сообщение через 3 секунды
|
||||
await delete_messages(bot, message, time_sleep=3, number_message=1)
|
||||
return
|
||||
|
||||
# Если с ответом на сообщение
|
||||
else:
|
||||
target_user = message.reply_to_message.from_user
|
||||
reason = extract_reason(parts_msg[1:])
|
||||
|
||||
# Если это обычный чат
|
||||
# Собираем данные
|
||||
target_user = message.reply_to_message.from_user
|
||||
reason = 'отсутствует'
|
||||
|
||||
# Если это General (обычный чат)
|
||||
elif message.reply_to_message and message.is_topic_message is None:
|
||||
|
||||
# Собираем данные
|
||||
target_user = message.reply_to_message.from_user
|
||||
reason = extract_reason(parts_msg[1:])
|
||||
reason = 'отсутствует'
|
||||
message.message_thread_id = None
|
||||
|
||||
# Удаляем сообщение, если команда неправильная
|
||||
# Если команда неправильная
|
||||
else:
|
||||
await asyncio.sleep(3)
|
||||
await bot.delete_message(message.chat.id, message.message_id)
|
||||
|
||||
# Удаляем сообщение через 3 секунды
|
||||
await delete_messages(bot, message, time_sleep=3, number_message=1)
|
||||
return
|
||||
|
||||
# В сообщении больше одного слова
|
||||
else:
|
||||
|
||||
# Если второе слово это тег или ID
|
||||
if parts_msg[1].strip().isdigit() or parts_msg[1].startswith('@'):
|
||||
|
||||
identifier = parts_msg[1].strip()
|
||||
reason = extract_reason(parts_msg[2:])
|
||||
|
||||
# Поиск по ID
|
||||
# Собираем данные
|
||||
identifier = parts_msg[1].strip()
|
||||
reason = ' '.join(parts_msg[2:]) if parts_msg[2:] != [] else 'отсутствует'
|
||||
|
||||
# Делаем поиск по ID
|
||||
if identifier.isdigit():
|
||||
|
||||
# Делаем в int и ищем
|
||||
# Ищем пользователя в базе данных
|
||||
user_info = db.get_user(int(identifier))
|
||||
|
||||
# Если нашли пользователя
|
||||
if user_info:
|
||||
# Создаем объект пользователя из данных базы
|
||||
|
||||
# Создаем объект пользователя
|
||||
target_user = User(
|
||||
id=user_info[0],
|
||||
first_name=user_info[1],
|
||||
@ -142,14 +117,16 @@ async def ban_command(bot: AsyncTeleBot, message: Message, photo_path: str = Non
|
||||
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],
|
||||
@ -157,67 +134,66 @@ async def ban_command(bot: AsyncTeleBot, message: Message, photo_path: str = Non
|
||||
is_bot=False
|
||||
)
|
||||
|
||||
# Команда через ответ на сообщение с указанием причины
|
||||
else:
|
||||
|
||||
# Если это тема
|
||||
# Если это топик
|
||||
if message.is_topic_message:
|
||||
|
||||
# Если без ответа на сообщение, ошибка
|
||||
# Если без ответа на сообщение
|
||||
if message.message_thread_id == message.reply_to_message.message_id:
|
||||
await asyncio.sleep(3)
|
||||
await bot.delete_message(message.chat.id, message.message_id)
|
||||
|
||||
# Удаляем сообщение через 3 секунды
|
||||
await delete_messages(bot, message, time_sleep=3, number_message=1)
|
||||
return
|
||||
|
||||
# Если с ответом на сообщение
|
||||
else:
|
||||
target_user = message.reply_to_message.from_user
|
||||
reason = extract_reason(parts_msg[1:])
|
||||
|
||||
# Если это обычный чат
|
||||
# Собираем данные
|
||||
target_user = message.reply_to_message.from_user
|
||||
reason = ' '.join(parts_msg[1:])
|
||||
|
||||
# Если это General (обычный чат)
|
||||
elif message.reply_to_message and message.is_topic_message is None:
|
||||
|
||||
# Собираем данные
|
||||
target_user = message.reply_to_message.from_user
|
||||
reason = extract_reason(parts_msg[1:])
|
||||
reason = ' '.join(parts_msg[1:])
|
||||
message.message_thread_id = None
|
||||
|
||||
# Удаляем сообщение, если команда неправильная
|
||||
# Если команда неправильная
|
||||
else:
|
||||
await asyncio.sleep(3)
|
||||
await bot.delete_message(message.chat.id, message.message_id)
|
||||
|
||||
# Удаляем сообщение через 3 секунды
|
||||
await delete_messages(bot, message, time_sleep=3, number_message=1)
|
||||
return
|
||||
|
||||
# Если пользователь не найден
|
||||
# Если пользователь не найден в базе данных
|
||||
if not target_user:
|
||||
await send_message(chat_id, COMMAND_MESSAGES['user_not_found'])
|
||||
|
||||
# Отправляем предупреждение
|
||||
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, 5)
|
||||
|
||||
await delete_messages(bot, message, time_sleep=5, number_message=2)
|
||||
return
|
||||
|
||||
# Проверяем статус целевого пользователя
|
||||
try:
|
||||
target_status = await bot.get_chat_member(message.chat.id, target_user.id)
|
||||
|
||||
# Проверяем, является ли цель администратором или создателем
|
||||
if target_status.status in ['administrator', 'creator']:
|
||||
await send_message(chat_id, COMMAND_MESSAGES['cant_ban_admin'])
|
||||
|
||||
# Удаляем сообщения через 5 секунд
|
||||
await delete_messages(bot, message, 5)
|
||||
|
||||
return
|
||||
|
||||
except Exception as e:
|
||||
await send_message(chat_id, f"⚠️ Ошибка: {str(e)}")
|
||||
logger.error(f"Ошибка при получении статуса пользователя: {str(e)}")
|
||||
|
||||
# Удаляем сообщения через 5 секунд
|
||||
await delete_messages(bot, message, 5)
|
||||
|
||||
if await check_target_status(bot, message, target_user) == 1:
|
||||
return
|
||||
|
||||
# Выполняем бан
|
||||
try:
|
||||
await bot.ban_chat_member(message.chat.id, target_user.id)
|
||||
|
||||
# Выполняем бан
|
||||
await bot.ban_chat_member(
|
||||
chat_id=message.chat.id,
|
||||
user_id=target_user.id,
|
||||
)
|
||||
|
||||
# Отправляем лог в админ-чат
|
||||
await action_reporter.log_action(
|
||||
@ -226,28 +202,48 @@ async def ban_command(bot: AsyncTeleBot, message: Message, photo_path: str = Non
|
||||
admin_id=message.from_user.id,
|
||||
reason=reason,
|
||||
duration=None,
|
||||
photo_path=photo_path
|
||||
photo_path=photo_path,
|
||||
)
|
||||
|
||||
# Отправляем сообщения, что пользователь получил бан
|
||||
await send_message(chat_id, COMMAND_MESSAGES['banned'])
|
||||
await bot.send_message(
|
||||
chat_id=message.chat.id,
|
||||
text=COMMAND_MESSAGES['banned'],
|
||||
message_thread_id=message.message_thread_id,
|
||||
)
|
||||
|
||||
# Записываем действие в логи
|
||||
logger.info(f"Пользователь {target_user.id} забанен администратором {message.from_user.id}.")
|
||||
|
||||
# Удаляем сообщения через 5 секунд
|
||||
await asyncio.sleep(5)
|
||||
await bot.delete_message(message.chat.id, message.message_id)
|
||||
await bot.delete_message(message.chat.id, message.message_id+2)
|
||||
await delete_messages(bot, message, time_sleep=5, number_message=2)
|
||||
|
||||
except Exception as e:
|
||||
await send_message(chat_id, f"⚠️ Ошибка: {str(e)}")
|
||||
|
||||
# Отправляем ошибку
|
||||
await bot.send_message(
|
||||
chat_id=message.chat.id,
|
||||
text=COMMAND_MESSAGES['error'].format(e=str(e)),
|
||||
message_thread_id=message.message_thread_id,
|
||||
)
|
||||
|
||||
# Записываем ошибку в логи
|
||||
logger.error(f"Ошибка бана: {str(e)}")
|
||||
|
||||
# Удаляем сообщения через 5 секунд
|
||||
await delete_messages(bot, message, 5)
|
||||
await delete_messages(bot, message, time_sleep=5, number_message=2)
|
||||
|
||||
except Exception as e:
|
||||
await send_message(chat_id, COMMAND_MESSAGES['general_error'])
|
||||
|
||||
# Отправляем ошибку
|
||||
await bot.send_message(
|
||||
chat_id=message.chat.id,
|
||||
text=COMMAND_MESSAGES['general_error'],
|
||||
message_thread_id=message.message_thread_id,
|
||||
)
|
||||
|
||||
# Записываем ошибку в логи
|
||||
logger.error(f"Общая ошибка в ban_command: {str(e)}")
|
||||
|
||||
# Удаляем сообщения через 5 секунд
|
||||
await delete_messages(bot, message, 5)
|
||||
await delete_messages(bot, message, time_sleep=5, number_message=2)
|
||||
@ -3,32 +3,46 @@ from telebot.types import Message
|
||||
import asyncio
|
||||
import logging
|
||||
|
||||
from utils import delete_messages
|
||||
|
||||
from config import COMMAND_MESSAGES
|
||||
|
||||
logger = logging.getLogger(__name__) # Получаем логгер для текущего модуля
|
||||
# Получаем логгер для текущего модуля
|
||||
logger = logging.getLogger(__name__)
|
||||
|
||||
def register_handlers(bot: AsyncTeleBot): # Регистрирует все обработчики команд
|
||||
# Регистрирует все обработчики команд
|
||||
def register_handlers(bot: AsyncTeleBot):
|
||||
|
||||
@bot.message_handler(commands=['help']) # Обработчик команды /help
|
||||
# Обработчик команды /help
|
||||
@bot.message_handler(commands=['help'])
|
||||
async def help_command(message: Message):
|
||||
|
||||
# Отправка сообщения в тему или обычный чат
|
||||
send_message = bot.reply_to if message.is_topic_message else bot.send_message
|
||||
chat_id = message if message.is_topic_message else message.chat.id
|
||||
|
||||
try:
|
||||
|
||||
# Отправляем сообщение
|
||||
await send_message(chat_id, COMMAND_MESSAGES['help'])
|
||||
await bot.send_message(
|
||||
chat_id=message.chat.id,
|
||||
text=COMMAND_MESSAGES['help'],
|
||||
message_thread_id=message.message_thread_id,
|
||||
)
|
||||
|
||||
# Записываем действие в логи
|
||||
logger.info(f"Пользователь {message.from_user.id} запустил /help.")
|
||||
|
||||
# Если пользователь писал в чат
|
||||
if message.chat.id != message.from_user.id:
|
||||
|
||||
# Удаляем сообщения через 30 секунд
|
||||
await asyncio.sleep(30)
|
||||
await bot.delete_message(message.chat.id, message.message_id)
|
||||
await bot.delete_message(message.chat.id, message.message_id+1)
|
||||
await delete_messages(bot, message, time_sleep=30, number_message=2)
|
||||
|
||||
except Exception as e:
|
||||
logger.error(f"Пользователь {message.from_user.id} запустил /help: {str(e)}")
|
||||
|
||||
# Отправляем ошибку
|
||||
await send_message(
|
||||
chat_id=message.chat.id,
|
||||
text=COMMAND_MESSAGES['general_error'],
|
||||
message_thread_id=message.message_thread_id,
|
||||
)
|
||||
|
||||
# Записываем ошибку в логи
|
||||
logger.error(f"Общая ошибка в help_command: {str(e)}")
|
||||
@ -6,90 +6,18 @@ import time
|
||||
from database import db
|
||||
|
||||
from action_reporter import action_reporter
|
||||
from utils import (
|
||||
delete_messages,
|
||||
check_admin_status,
|
||||
check_target_status,
|
||||
parse_mute_time,
|
||||
format_mute_time,
|
||||
)
|
||||
|
||||
from config import COMMAND_MESSAGES
|
||||
|
||||
logger = logging.getLogger(__name__) # Получаем логгер для текущего модуля
|
||||
|
||||
# Возвращает количество секунд
|
||||
def parse_mute_time(time_str: str) -> int:
|
||||
|
||||
# Парсим строку времени
|
||||
time_str = time_str.strip().lower()
|
||||
|
||||
# Минуты
|
||||
if time_str.endswith('m') or time_str.endswith('м'):
|
||||
try:
|
||||
minutes = int(time_str[:-1])
|
||||
return abs(minutes) * 60
|
||||
except:
|
||||
return None
|
||||
|
||||
# Часы
|
||||
elif time_str.endswith('h') or time_str.endswith('ч'):
|
||||
try:
|
||||
hours = int(time_str[:-1])
|
||||
return abs(hours) * 3600
|
||||
except:
|
||||
return None
|
||||
|
||||
# Дни
|
||||
elif time_str.endswith('d') or time_str.endswith('д'):
|
||||
try:
|
||||
days = int(time_str[:-1])
|
||||
return abs(days) * 86400
|
||||
except:
|
||||
return None
|
||||
|
||||
return None
|
||||
|
||||
# Возвращает причину мута
|
||||
def extract_reason(words: str) -> str:
|
||||
|
||||
if words == []:
|
||||
reason = 'отсутствует'
|
||||
else:
|
||||
reason = ' '.join(words)
|
||||
|
||||
return reason
|
||||
|
||||
# Форматирует время в нормальный вид
|
||||
def format_time(seconds: int) -> str:
|
||||
|
||||
# Для минут
|
||||
minutes = seconds // 60
|
||||
if minutes < 60:
|
||||
if minutes % 10 == 1 and minutes % 100 != 11:
|
||||
return f"{minutes} минута"
|
||||
elif 2 <= minutes % 10 <= 4 and minutes % 100 not in (12, 13, 14):
|
||||
return f"{minutes} минуты"
|
||||
else:
|
||||
return f"{minutes} минут"
|
||||
|
||||
# Для часов
|
||||
hours = minutes // 60
|
||||
if hours < 24:
|
||||
if hours % 10 == 1 and hours % 100 != 11:
|
||||
return f"{hours} час"
|
||||
elif 2 <= hours % 10 <= 4 and hours % 100 not in (12, 13, 14):
|
||||
return f"{hours} часа"
|
||||
else:
|
||||
return f"{hours} часов"
|
||||
|
||||
# Для дней
|
||||
days = hours // 24
|
||||
if days % 10 == 1 and days % 100 != 11:
|
||||
return f"{days} день"
|
||||
elif 2 <= days % 10 <= 4 and days % 100 not in (12, 13, 14):
|
||||
return f"{days} дня"
|
||||
else:
|
||||
return f"{days} дней"
|
||||
|
||||
# Удаляет два последних сообщения
|
||||
async def delete_messages(bot: AsyncTeleBot, message: Message, time_sleep: int):
|
||||
await asyncio.sleep(time_sleep)
|
||||
await bot.delete_message(message.chat.id, message.message_id)
|
||||
await bot.delete_message(message.chat.id, message.message_id+1)
|
||||
# Получаем логгер для текущего модуля
|
||||
logger = logging.getLogger(__name__)
|
||||
|
||||
# Регистрирует все обработчики команд
|
||||
def register_handlers(bot: AsyncTeleBot):
|
||||
@ -102,10 +30,6 @@ def register_handlers(bot: AsyncTeleBot):
|
||||
# Основная функция команды /mute
|
||||
async def mute_command(bot: AsyncTeleBot, message: Message, photo_path: str = None):
|
||||
|
||||
# Отправка сообщения в тему или обычный чат
|
||||
send_message = bot.reply_to if message.is_topic_message else bot.send_message
|
||||
chat_id = message if message.is_topic_message else message.chat.id
|
||||
|
||||
# Определяем целевого пользователя
|
||||
target_user = None
|
||||
|
||||
@ -118,100 +42,91 @@ async def mute_command(bot: AsyncTeleBot, message: Message, photo_path: str = No
|
||||
# Разбиваем текст сообщения на части
|
||||
parts_msg = message.text.split()
|
||||
|
||||
# Выводим помощь
|
||||
if len(parts_msg) == 2 and parts_msg[1].strip() in ['help', 'помощь']:
|
||||
await send_message(chat_id, COMMAND_MESSAGES['manual_mute'])
|
||||
# Команда /mute 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_mute'],
|
||||
message_thread_id=message.message_thread_id,
|
||||
)
|
||||
|
||||
# Удаляем сообщения через 30 секунд
|
||||
await delete_messages(bot, message, 30)
|
||||
|
||||
await delete_messages(bot, message, time_sleep=30, number_message=2)
|
||||
return
|
||||
|
||||
try:
|
||||
|
||||
# Проверяем, является ли отправитель администратором
|
||||
try:
|
||||
admin_status = await bot.get_chat_member(message.chat.id, message.from_user.id)
|
||||
|
||||
# Проверяем статус администратора (создателя)
|
||||
if admin_status.status not in ['administrator', 'creator']:
|
||||
await send_message(chat_id, COMMAND_MESSAGES['no_admin_rights'])
|
||||
|
||||
# Удаляем сообщения через 5 секунд
|
||||
await delete_messages(bot, message, 5)
|
||||
|
||||
return
|
||||
|
||||
# Проверяем право администратора на мут
|
||||
if admin_status.status == 'administrator' and not admin_status.can_restrict_members:
|
||||
await send_message(chat_id, COMMAND_MESSAGES['no_restrict_rights'])
|
||||
|
||||
# Удаляем сообщения через 5 секунд
|
||||
await delete_messages(bot, message, 5)
|
||||
|
||||
return
|
||||
|
||||
except Exception as e:
|
||||
await send_message(chat_id, f"⚠️ Ошибка: {str(e)}")
|
||||
logger.error(f"Ошибка при получении статуса администратора: {str(e)}")
|
||||
|
||||
# Удаляем сообщения через 5 секунд
|
||||
await delete_messages(bot, message, 5)
|
||||
|
||||
if await check_admin_status(bot, message) == 1:
|
||||
return
|
||||
|
||||
# Если одно слово, то удаляем сообщение. Ошибка
|
||||
# Если одно слово (/mute)
|
||||
if len(parts_msg) == 1:
|
||||
await asyncio.sleep(3)
|
||||
await bot.delete_message(message.chat.id, message.message_id)
|
||||
|
||||
# Удаляем сообщение через 3 секунды
|
||||
await delete_messages(bot, message, time_sleep=3, number_message=1)
|
||||
return
|
||||
|
||||
# Если два слово, то ответом на сообщение
|
||||
# Команда через ответ на сообщение, если два слова (/mute 2m)
|
||||
elif len(parts_msg) == 2:
|
||||
|
||||
# Если это тема
|
||||
# Если это топик
|
||||
if message.is_topic_message:
|
||||
|
||||
# Если без ответа на сообщение, ошибка
|
||||
# Если без ответа на сообщение
|
||||
if message.message_thread_id == message.reply_to_message.message_id:
|
||||
await asyncio.sleep(3)
|
||||
await bot.delete_message(message.chat.id, message.message_id)
|
||||
|
||||
# Удаляем сообщение через 3 секунды
|
||||
await delete_messages(bot, message, time_sleep=3, number_message=1)
|
||||
return
|
||||
|
||||
# Если с ответом на сообщение
|
||||
else:
|
||||
|
||||
# Собираем данные
|
||||
target_user = message.reply_to_message.from_user
|
||||
time_arg = parts_msg[1]
|
||||
reason = extract_reason(parts_msg[2:])
|
||||
reason = 'отсутствует'
|
||||
|
||||
# Если это обычный чат
|
||||
# Если это General (обычный чат)
|
||||
elif message.reply_to_message and message.is_topic_message is None:
|
||||
|
||||
# Собираем данные
|
||||
target_user = message.reply_to_message.from_user
|
||||
time_arg = parts_msg[1]
|
||||
reason = extract_reason(parts_msg[2:])
|
||||
reason = 'отсутствует'
|
||||
message.message_thread_id = None
|
||||
|
||||
# Удаляем сообщение, если команда неправильная
|
||||
# Если команда неправильная
|
||||
else:
|
||||
await asyncio.sleep(3)
|
||||
await bot.delete_message(message.chat.id, message.message_id)
|
||||
|
||||
# Удаляем сообщение через 3 секунды
|
||||
await delete_messages(bot, message, time_sleep=3, number_message=1)
|
||||
return
|
||||
|
||||
# В сообщении больше двух слов
|
||||
else:
|
||||
|
||||
# Если второе слово это тег или ID
|
||||
if parts_msg[1].strip().isdigit() or parts_msg[1].startswith('@'):
|
||||
|
||||
|
||||
# Собираем данные
|
||||
identifier = parts_msg[1].strip()
|
||||
time_arg = parts_msg[2]
|
||||
reason = extract_reason(parts_msg[3:])
|
||||
reason = ' '.join(parts_msg[3:]) if parts_msg[3:] != [] else 'отсутствует'
|
||||
|
||||
# Поиск по ID
|
||||
# Делаем поиск по ID
|
||||
if identifier.isdigit():
|
||||
|
||||
# Делаем в int и ищем
|
||||
# Ищем пользователя в базе данных
|
||||
user_info = db.get_user(int(identifier))
|
||||
|
||||
# Если нашли пользователя
|
||||
if user_info:
|
||||
# Создаем объект пользователя из данных базы
|
||||
|
||||
# Создаем объект пользователя
|
||||
target_user = User(
|
||||
id=user_info[0],
|
||||
first_name=user_info[1],
|
||||
@ -219,14 +134,16 @@ async def mute_command(bot: AsyncTeleBot, message: Message, photo_path: str = No
|
||||
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],
|
||||
@ -234,95 +151,106 @@ async def mute_command(bot: AsyncTeleBot, message: Message, photo_path: str = No
|
||||
is_bot=False
|
||||
)
|
||||
|
||||
# Команда через ответ на сообщение с указанием причины
|
||||
else:
|
||||
|
||||
# Если это тема
|
||||
# Если это топик
|
||||
if message.is_topic_message:
|
||||
|
||||
# Если без ответа на сообщение, ошибка
|
||||
# Если без ответа на сообщение
|
||||
if message.message_thread_id == message.reply_to_message.message_id:
|
||||
await asyncio.sleep(3)
|
||||
await bot.delete_message(message.chat.id, message.message_id)
|
||||
|
||||
# Удаляем сообщение через 3 секунды
|
||||
await delete_messages(bot, message, time_sleep=3, number_message=1)
|
||||
return
|
||||
|
||||
# Если с ответом на сообщение
|
||||
else:
|
||||
|
||||
# Собираем данные
|
||||
target_user = message.reply_to_message.from_user
|
||||
time_arg = parts_msg[1]
|
||||
reason = extract_reason(parts_msg[2:])
|
||||
reason = ' '.join(parts_msg[2:])
|
||||
|
||||
# Если это обычный чат
|
||||
# Если это General (обычный чат)
|
||||
elif message.reply_to_message and message.is_topic_message is None:
|
||||
|
||||
# Собираем данные
|
||||
target_user = message.reply_to_message.from_user
|
||||
time_arg = parts_msg[1]
|
||||
reason = extract_reason(parts_msg[2:])
|
||||
reason = ' '.join(parts_msg[2:])
|
||||
message.message_thread_id = None
|
||||
|
||||
# Удаляем сообщение, если команда неправильная
|
||||
# Если команда неправильная
|
||||
else:
|
||||
await asyncio.sleep(3)
|
||||
await bot.delete_message(message.chat.id, message.message_id)
|
||||
|
||||
# Удаляем сообщение через 3 секунды
|
||||
await delete_messages(bot, message, time_sleep=3, number_message=1)
|
||||
return
|
||||
|
||||
# Если пользователь не найден
|
||||
# Если пользователь не найден в базе данных
|
||||
if not target_user:
|
||||
await send_message(chat_id, COMMAND_MESSAGES['user_not_found'])
|
||||
|
||||
# Отправляем предупреждение
|
||||
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, 5)
|
||||
|
||||
await delete_messages(bot, message, time_sleep=5, number_message=2)
|
||||
return
|
||||
|
||||
# Парсинг времени мута
|
||||
mute_seconds = parse_mute_time(time_arg)
|
||||
|
||||
# Если не указали время
|
||||
if mute_seconds is None:
|
||||
await send_message(chat_id, COMMAND_MESSAGES['incorrect_time_format'])
|
||||
|
||||
# Отправляем предупреждение
|
||||
await bot.send_message(
|
||||
chat_id=message.chat.id,
|
||||
text=COMMAND_MESSAGES['incorrect_time_format'],
|
||||
message_thread_id=message.message_thread_id,
|
||||
)
|
||||
|
||||
# Удаляем сообщения через 5 секунд
|
||||
await delete_messages(bot, message, 5)
|
||||
|
||||
await delete_messages(bot, message, time_sleep=5, number_message=2)
|
||||
return
|
||||
|
||||
# Минимальный мут 1 минута (60 секунд)
|
||||
# Минимальное время мута - 1 минута (60 секунд)
|
||||
if mute_seconds < 60:
|
||||
await send_message(chat_id, COMMAND_MESSAGES['min_mute'])
|
||||
|
||||
# Отправляем предупреждение
|
||||
await bot.send_message(
|
||||
chat_id=message.chat.id,
|
||||
text=COMMAND_MESSAGES['min_mute'],
|
||||
message_thread_id=message.message_thread_id,
|
||||
)
|
||||
|
||||
# Удаляем сообщения через 5 секунд
|
||||
await delete_messages(bot, message, 5)
|
||||
|
||||
await delete_messages(bot, message, time_sleep=5, number_message=2)
|
||||
return
|
||||
|
||||
# Максимальный мут 30 дней (2592000 секунд)
|
||||
# Максимальное время мута - 30 дней (2592000 секунд)
|
||||
if mute_seconds > 2592000:
|
||||
await send_message(chat_id, COMMAND_MESSAGES['max_mute'])
|
||||
|
||||
# Отправляем предупреждение
|
||||
await bot.send_message(
|
||||
chat_id=message.chat.id,
|
||||
text=COMMAND_MESSAGES['max_mute'],
|
||||
message_thread_id=message.message_thread_id,
|
||||
)
|
||||
|
||||
# Удаляем сообщения через 5 секунд
|
||||
await delete_messages(bot, message, 5)
|
||||
|
||||
await delete_messages(bot, message, time_sleep=5, number_message=2)
|
||||
return
|
||||
|
||||
# Проверяем статус целевого пользователя
|
||||
try:
|
||||
target_status = await bot.get_chat_member(message.chat.id, target_user.id)
|
||||
|
||||
# Проверяем, является ли цель администратором или создателем
|
||||
if target_status.status in ['administrator', 'creator']:
|
||||
await send_message(chat_id, COMMAND_MESSAGES['cant_mute_admin'])
|
||||
|
||||
# Удаляем сообщения через 5 секунд
|
||||
await delete_messages(bot, message, 5)
|
||||
|
||||
return
|
||||
|
||||
except Exception as e:
|
||||
await send_message(chat_id, f"⚠️ Ошибка: {str(e)}")
|
||||
logger.error(f"Ошибка при получении статуса пользователя: {str(e)}")
|
||||
|
||||
# Удаляем сообщения через 5 секунд
|
||||
await delete_messages(bot, message, 5)
|
||||
|
||||
if await check_target_status(bot, message, target_user) == 1:
|
||||
return
|
||||
|
||||
# Выполняем мут
|
||||
try:
|
||||
|
||||
# Вычисляем время окончания мута
|
||||
@ -340,6 +268,7 @@ async def mute_command(bot: AsyncTeleBot, message: Message, photo_path: str = No
|
||||
can_pin_messages=False,
|
||||
)
|
||||
|
||||
# Выполняем мут
|
||||
await bot.restrict_chat_member(
|
||||
chat_id=message.chat.id,
|
||||
user_id=target_user.id,
|
||||
@ -347,36 +276,58 @@ async def mute_command(bot: AsyncTeleBot, message: Message, photo_path: str = No
|
||||
until_date=until_date
|
||||
)
|
||||
|
||||
# Форматирование времени
|
||||
time_display = format_time(mute_seconds)
|
||||
# Форматируем время в удобный формат
|
||||
time_display = format_mute_time(mute_seconds)
|
||||
|
||||
# Отправляем лог в админ-чат
|
||||
# Отправляем сообщение-лог в админ-чат
|
||||
await action_reporter.log_action(
|
||||
action="МУТ",
|
||||
user_id=target_user.id,
|
||||
admin_id=message.from_user.id,
|
||||
reason=reason,
|
||||
duration=time_display,
|
||||
photo_path=photo_path
|
||||
photo_path=photo_path,
|
||||
)
|
||||
|
||||
# Отправляем сообщения, что пользователь получил мут
|
||||
await send_message(chat_id, COMMAND_MESSAGES['muted'].format(time_display=time_display))
|
||||
# Отправляем сообщение, что пользователь получил мут
|
||||
await bot.send_message(
|
||||
chat_id=message.chat.id,
|
||||
text=COMMAND_MESSAGES['muted'].format(time_display=time_display),
|
||||
message_thread_id=message.message_thread_id,
|
||||
)
|
||||
|
||||
# Записываем действие в логи
|
||||
logger.info(f"Пользователь {target_user.id} получил мут на {time_display} от администратора {message.from_user.id}.")
|
||||
|
||||
# Удаляем сообщения через 5 секунд
|
||||
await delete_messages(bot, message, 5)
|
||||
await delete_messages(bot, message, time_sleep=5, number_message=2)
|
||||
|
||||
except Exception as e:
|
||||
await send_message(chat_id, f"⚠️ Ошибка: {str(e)}")
|
||||
|
||||
# Отправляем ошибку
|
||||
await bot.send_message(
|
||||
chat_id=message.chat.id,
|
||||
text=COMMAND_MESSAGES['error'].format(e=str(e)),
|
||||
message_thread_id=message.message_thread_id,
|
||||
)
|
||||
|
||||
# Записываем ошибку в логи
|
||||
logger.error(f"Ошибка мута: {str(e)}")
|
||||
|
||||
# Удаляем сообщения через 5 секунд
|
||||
await delete_messages(bot, message, 5)
|
||||
await delete_messages(bot, message, time_sleep=5, number_message=2)
|
||||
|
||||
except Exception as e:
|
||||
await send_message(chat_id, COMMAND_MESSAGES['general_error'])
|
||||
|
||||
# Отправляем ошибку
|
||||
await bot.send_message(
|
||||
chat_id=message.chat.id,
|
||||
text=COMMAND_MESSAGES['general_error'],
|
||||
message_thread_id=message.message_thread_id,
|
||||
)
|
||||
|
||||
# Записываем ошибку в логи
|
||||
logger.error(f"Общая ошибка в mute_command: {str(e)}")
|
||||
|
||||
# Удаляем сообщения через 5 секунд
|
||||
await delete_messages(bot, message, 5)
|
||||
await delete_messages(bot, message, time_sleep=5, number_message=2)
|
||||
@ -3,32 +3,46 @@ from telebot.types import Message
|
||||
import asyncio
|
||||
import logging
|
||||
|
||||
from utils import delete_messages
|
||||
|
||||
from config import COMMAND_MESSAGES
|
||||
|
||||
logger = logging.getLogger(__name__) # Получаем логгер для текущего модуля
|
||||
# Получаем логгер для текущего модуля
|
||||
logger = logging.getLogger(__name__)
|
||||
|
||||
def register_handlers(bot: AsyncTeleBot): # Регистрирует все обработчики команд
|
||||
# Регистрирует все обработчики команд
|
||||
def register_handlers(bot: AsyncTeleBot):
|
||||
|
||||
@bot.message_handler(commands=['start']) # Обработчик команды /start
|
||||
# Обработчик команды /start
|
||||
@bot.message_handler(commands=['start'])
|
||||
async def start_command(message: Message):
|
||||
|
||||
# Отправка сообщения в тему или обычный чат
|
||||
send_message = bot.reply_to if message.is_topic_message else bot.send_message
|
||||
chat_id = message if message.is_topic_message else message.chat.id
|
||||
|
||||
try:
|
||||
|
||||
# Отправляем сообщение
|
||||
await send_message(chat_id, COMMAND_MESSAGES['start'])
|
||||
await bot.send_message(
|
||||
chat_id=message.chat.id,
|
||||
text=COMMAND_MESSAGES['start'],
|
||||
message_thread_id=message.message_thread_id,
|
||||
)
|
||||
|
||||
# Записываем действие в логи
|
||||
logger.info(f"Пользователь {message.from_user.id} запустил /start.")
|
||||
|
||||
# Если пользователь писал в чат
|
||||
if message.chat.id != message.from_user.id:
|
||||
|
||||
# Удаляем сообщения через 5 секунд
|
||||
await asyncio.sleep(5)
|
||||
await bot.delete_message(message.chat.id, message.message_id)
|
||||
await bot.delete_message(message.chat.id, message.message_id+1)
|
||||
await delete_messages(bot, message, time_sleep=5, number_message=2)
|
||||
|
||||
except Exception as e:
|
||||
logger.error(f"Пользователь {message.from_user.id} запустил /start: {str(e)}")
|
||||
|
||||
# Отправляем ошибку
|
||||
await send_message(
|
||||
chat_id=message.chat.id,
|
||||
text=COMMAND_MESSAGES['general_error'],
|
||||
message_thread_id=message.message_thread_id,
|
||||
)
|
||||
|
||||
# Записываем ошибку в логи
|
||||
logger.error(f"Общая ошибка в start_command: {str(e)}")
|
||||
@ -1,116 +1,103 @@
|
||||
from telebot.async_telebot import AsyncTeleBot
|
||||
from telebot.types import Message, User
|
||||
from telebot.types import Message, User, ChatPermissions
|
||||
import asyncio
|
||||
import logging
|
||||
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__) # Получаем логгер для текущего модуля
|
||||
# Получаем логгер для текущего модуля
|
||||
logger = logging.getLogger(__name__)
|
||||
|
||||
# Удаляет два последних сообщения
|
||||
async def delete_messages(bot: AsyncTeleBot, message: Message, time_sleep: int):
|
||||
await asyncio.sleep(time_sleep)
|
||||
await bot.delete_message(message.chat.id, message.message_id)
|
||||
await bot.delete_message(message.chat.id, message.message_id+1)
|
||||
# Регистрирует все обработчики команд
|
||||
def register_handlers(bot: AsyncTeleBot):
|
||||
|
||||
def register_handlers(bot: AsyncTeleBot): # Регистрирует все обработчики команд
|
||||
|
||||
@bot.message_handler(commands=['unban']) # Обработчик команды /unban
|
||||
# Обработчик команды /unban
|
||||
@bot.message_handler(commands=['unban'])
|
||||
async def unban_command(message: Message):
|
||||
|
||||
# Отправка сообщения в тему или обычный чат
|
||||
send_message = bot.reply_to if message.is_topic_message else bot.send_message
|
||||
chat_id = message if message.is_topic_message else message.chat.id
|
||||
|
||||
# Определяем целевого пользователя
|
||||
target_user = None
|
||||
|
||||
# Разбиваем текст сообщения на части
|
||||
parts_msg = message.text.split()
|
||||
|
||||
# Выводим помощь (/unban help)
|
||||
if len(parts_msg) == 2 and parts_msg[1].strip() in ['help', 'помощь']:
|
||||
await send_message(chat_id, COMMAND_MESSAGES['manual_unban'])
|
||||
# Команда /unban 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_unban'],
|
||||
message_thread_id=message.message_thread_id,
|
||||
)
|
||||
|
||||
# Удаляем сообщения через 30 секунд
|
||||
await delete_messages(bot, message, 30)
|
||||
|
||||
await delete_messages(bot, message, time_sleep=30, number_message=2)
|
||||
return
|
||||
|
||||
try:
|
||||
|
||||
# Проверяем, является ли отправитель администратором
|
||||
try:
|
||||
admin_status = await bot.get_chat_member(message.chat.id, message.from_user.id)
|
||||
|
||||
# Проверяем статус администратора (создателя)
|
||||
if admin_status.status not in ['administrator', 'creator']:
|
||||
await send_message(chat_id, COMMAND_MESSAGES['no_admin_rights'])
|
||||
|
||||
# Удаляем сообщения через 5 секунд
|
||||
await delete_messages(bot, message, 5)
|
||||
|
||||
return
|
||||
|
||||
# Проверяем право администратора на разбан
|
||||
if admin_status.status == 'administrator' and not admin_status.can_restrict_members:
|
||||
await send_message(chat_id, COMMAND_MESSAGES['no_restrict_rights'])
|
||||
|
||||
# Удаляем сообщения через 5 секунд
|
||||
await delete_messages(bot, message, 5)
|
||||
|
||||
return
|
||||
|
||||
except Exception as e:
|
||||
await send_message(chat_id, f"⚠️ Ошибка: {str(e)}")
|
||||
logger.error(f"Ошибка при получении статуса администратора: {str(e)}")
|
||||
|
||||
# Удаляем сообщения через 5 секунд
|
||||
await delete_messages(bot, message, 5)
|
||||
|
||||
if await check_admin_status(bot, message) == 1:
|
||||
return
|
||||
|
||||
# Если одно слово, то ответом на сообщение
|
||||
# Команда через ответ на сообщение, если одно слово (/unban)
|
||||
if len(parts_msg) == 1:
|
||||
|
||||
# Если это тема
|
||||
# Если это топик
|
||||
if message.is_topic_message:
|
||||
|
||||
# Если без ответа на сообщение, ошибка
|
||||
# Если без ответа на сообщение
|
||||
if message.message_thread_id == message.reply_to_message.message_id:
|
||||
await asyncio.sleep(3)
|
||||
await bot.delete_message(message.chat.id, message.message_id)
|
||||
|
||||
# Удаляем сообщение через 3 секунды
|
||||
await delete_messages(bot, message, time_sleep=3, number_message=1)
|
||||
return
|
||||
|
||||
# Если с ответом на сообщение
|
||||
else:
|
||||
|
||||
# Собираем данные
|
||||
target_user = message.reply_to_message.from_user
|
||||
|
||||
# Если это обычный чат
|
||||
# Если это General (обычный чат)
|
||||
elif message.reply_to_message and message.is_topic_message is None:
|
||||
|
||||
# Собираем данные
|
||||
target_user = message.reply_to_message.from_user
|
||||
message.message_thread_id = None
|
||||
|
||||
# Удаляем сообщение, если команда неправильная
|
||||
else:
|
||||
await asyncio.sleep(3)
|
||||
await bot.delete_message(message.chat.id, message.message_id)
|
||||
|
||||
# Удаляем сообщение через 3 секунды
|
||||
await delete_messages(bot, message, time_sleep=3, number_message=1)
|
||||
return
|
||||
|
||||
# Если два, то через тег или ID
|
||||
# Команда через тег или ID, если два слова
|
||||
elif len(parts_msg) == 2:
|
||||
|
||||
# Собираем данные
|
||||
identifier = parts_msg[1].strip()
|
||||
|
||||
# Поиск по ID
|
||||
# Делаем поиск по ID
|
||||
if identifier.isdigit():
|
||||
|
||||
# Делаем в int и ищем
|
||||
# Ищем пользователя в базе данных
|
||||
user_info = db.get_user(int(identifier))
|
||||
|
||||
# Если нашли пользователя
|
||||
if user_info:
|
||||
# Создаем объект пользователя из данных базы
|
||||
|
||||
# Создаем объект пользователя
|
||||
target_user = User(
|
||||
id=user_info[0],
|
||||
first_name=user_info[1],
|
||||
@ -118,14 +105,16 @@ def register_handlers(bot: AsyncTeleBot): # Регистрирует все об
|
||||
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],
|
||||
@ -133,18 +122,31 @@ def register_handlers(bot: AsyncTeleBot): # Регистрирует все об
|
||||
is_bot=False
|
||||
)
|
||||
|
||||
# Если пользователь не найден
|
||||
# Если пользователь не найден в базе данных
|
||||
if not target_user:
|
||||
await send_message(chat_id, COMMAND_MESSAGES['user_not_found'])
|
||||
|
||||
# Отправляем предупреждение
|
||||
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, 5)
|
||||
|
||||
await delete_messages(bot, message, time_sleep=5, number_message=2)
|
||||
return
|
||||
|
||||
# Проверяем статус целевого пользователя
|
||||
if await check_target_status(bot, message, target_user) == 1:
|
||||
return
|
||||
|
||||
# Выполняем разбан
|
||||
try:
|
||||
await bot.unban_chat_member(message.chat.id, target_user.id)
|
||||
|
||||
# Выполняем разбан
|
||||
await bot.unban_chat_member(
|
||||
chat_id=message.chat.id,
|
||||
user_id=target_user.id,
|
||||
)
|
||||
|
||||
# Отправляем лог в админ-чат
|
||||
await action_reporter.log_action(
|
||||
@ -153,26 +155,48 @@ def register_handlers(bot: AsyncTeleBot): # Регистрирует все об
|
||||
admin_id=message.from_user.id,
|
||||
reason=None,
|
||||
duration=None,
|
||||
photo_path=None
|
||||
photo_path=None,
|
||||
)
|
||||
|
||||
# Отправляем сообщения, что пользователь получил разбан
|
||||
await send_message(chat_id, COMMAND_MESSAGES['unbanned'])
|
||||
await bot.send_message(
|
||||
chat_id=message.chat.id,
|
||||
text=COMMAND_MESSAGES['unbanned'],
|
||||
message_thread_id=message.message_thread_id,
|
||||
)
|
||||
|
||||
# Записываем действие в логи
|
||||
logger.info(f"Пользователь {target_user.id} разбанен администратором {message.from_user.id}.")
|
||||
|
||||
# Удаляем сообщения через 5 секунд
|
||||
await delete_messages(bot, message, 5)
|
||||
await delete_messages(bot, message, time_sleep=5, number_message=2)
|
||||
|
||||
except Exception as e:
|
||||
await send_message(chat_id, f"⚠️ Ошибка: {str(e)}")
|
||||
logger.error(f"Ошибка разбана: {str(e)}")
|
||||
|
||||
# Отправляем ошибку
|
||||
await bot.send_message(
|
||||
chat_id=message.chat.id,
|
||||
text=COMMAND_MESSAGES['error'].format(e=str(e)),
|
||||
message_thread_id=message.message_thread_id,
|
||||
)
|
||||
|
||||
# Записываем ошибку в логи
|
||||
logger.error(f"Ошибка размута: {str(e)}")
|
||||
|
||||
# Удаляем сообщения через 5 секунд
|
||||
await delete_messages(bot, message, 5)
|
||||
await delete_messages(bot, message, time_sleep=5, number_message=2)
|
||||
|
||||
except Exception as e:
|
||||
await send_message(chat_id, COMMAND_MESSAGES['general_error'])
|
||||
|
||||
# Отправляем ошибку
|
||||
await bot.send_message(
|
||||
chat_id=message.chat.id,
|
||||
text=COMMAND_MESSAGES['general_error'],
|
||||
message_thread_id=message.message_thread_id,
|
||||
)
|
||||
|
||||
# Записываем ошибку в логи
|
||||
logger.error(f"Общая ошибка в unban_command: {str(e)}")
|
||||
|
||||
# Удаляем сообщения через 5 секунд
|
||||
await delete_messages(bot, message, 5)
|
||||
await delete_messages(bot, message, time_sleep=5, number_message=2)
|
||||
@ -5,112 +5,99 @@ import logging
|
||||
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__) # Получаем логгер для текущего модуля
|
||||
# Получаем логгер для текущего модуля
|
||||
logger = logging.getLogger(__name__)
|
||||
|
||||
# Удаляет два последних сообщения
|
||||
async def delete_messages(bot: AsyncTeleBot, message: Message, time_sleep: int):
|
||||
await asyncio.sleep(time_sleep)
|
||||
await bot.delete_message(message.chat.id, message.message_id)
|
||||
await bot.delete_message(message.chat.id, message.message_id+1)
|
||||
# Регистрирует все обработчики команд
|
||||
def register_handlers(bot: AsyncTeleBot):
|
||||
|
||||
def register_handlers(bot: AsyncTeleBot): # Регистрирует все обработчики команд
|
||||
|
||||
@bot.message_handler(commands=['unmute']) # Обработчик команды /unmute
|
||||
# Обработчик команды /unmute
|
||||
@bot.message_handler(commands=['unmute'])
|
||||
async def unmute_command(message: Message):
|
||||
|
||||
# Отправка сообщения в тему или обычный чат
|
||||
send_message = bot.reply_to if message.is_topic_message else bot.send_message
|
||||
chat_id = message if message.is_topic_message else message.chat.id
|
||||
|
||||
# Определяем целевого пользователя
|
||||
target_user = None
|
||||
|
||||
# Разбиваем текст сообщения на части
|
||||
parts_msg = message.text.split()
|
||||
|
||||
# Выводим помощь (/unmute help)
|
||||
if len(parts_msg) == 2 and parts_msg[1].strip() in ['help', 'помощь']:
|
||||
await send_message(chat_id, COMMAND_MESSAGES['manual_unmute'])
|
||||
# Команда /unmute 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_unmute'],
|
||||
message_thread_id=message.message_thread_id,
|
||||
)
|
||||
|
||||
# Удаляем сообщения через 30 секунд
|
||||
await delete_messages(bot, message, 30)
|
||||
|
||||
await delete_messages(bot, message, time_sleep=30, number_message=2)
|
||||
return
|
||||
|
||||
try:
|
||||
|
||||
# Проверяем, является ли отправитель администратором
|
||||
try:
|
||||
admin_status = await bot.get_chat_member(message.chat.id, message.from_user.id)
|
||||
|
||||
# Проверяем статус администратора (создателя)
|
||||
if admin_status.status not in ['administrator', 'creator']:
|
||||
await send_message(chat_id, COMMAND_MESSAGES['no_admin_rights'])
|
||||
|
||||
# Удаляем сообщения через 5 секунд
|
||||
await delete_messages(bot, message, 5)
|
||||
|
||||
return
|
||||
|
||||
# Проверяем право администратора на разбан
|
||||
if admin_status.status == 'administrator' and not admin_status.can_restrict_members:
|
||||
await send_message(chat_id, COMMAND_MESSAGES['no_restrict_rights'])
|
||||
|
||||
# Удаляем сообщения через 5 секунд
|
||||
await delete_messages(bot, message, 5)
|
||||
|
||||
return
|
||||
|
||||
except Exception as e:
|
||||
await send_message(chat_id, f"⚠️ Ошибка: {str(e)}")
|
||||
logger.error(f"Ошибка при получении статуса администратора: {str(e)}")
|
||||
|
||||
# Удаляем сообщения через 5 секунд
|
||||
await delete_messages(bot, message, 5)
|
||||
|
||||
if await check_admin_status(bot, message) == 1:
|
||||
return
|
||||
|
||||
# Если одно слово, то ответом на сообщение
|
||||
# Команда через ответ на сообщение, если одно слово (/unmute)
|
||||
if len(parts_msg) == 1:
|
||||
|
||||
# Если это тема
|
||||
# Если это топик
|
||||
if message.is_topic_message:
|
||||
|
||||
# Если без ответа на сообщение, ошибка
|
||||
# Если без ответа на сообщение
|
||||
if message.message_thread_id == message.reply_to_message.message_id:
|
||||
await asyncio.sleep(3)
|
||||
await bot.delete_message(message.chat.id, message.message_id)
|
||||
|
||||
# Удаляем сообщение через 3 секунды
|
||||
await delete_messages(bot, message, time_sleep=3, number_message=1)
|
||||
return
|
||||
|
||||
# Если с ответом на сообщение
|
||||
else:
|
||||
|
||||
# Собираем данные
|
||||
target_user = message.reply_to_message.from_user
|
||||
|
||||
# Если это обычный чат
|
||||
# Если это General (обычный чат)
|
||||
elif message.reply_to_message and message.is_topic_message is None:
|
||||
|
||||
# Собираем данные
|
||||
target_user = message.reply_to_message.from_user
|
||||
message.message_thread_id = None
|
||||
|
||||
# Удаляем сообщение, если команда неправильная
|
||||
else:
|
||||
await asyncio.sleep(3)
|
||||
await bot.delete_message(message.chat.id, message.message_id)
|
||||
|
||||
# Удаляем сообщение через 3 секунды
|
||||
await delete_messages(bot, message, time_sleep=3, number_message=1)
|
||||
return
|
||||
|
||||
# Если два, то через тег или ID
|
||||
# Команда через тег или ID, если два слова
|
||||
elif len(parts_msg) == 2:
|
||||
|
||||
# Собираем данные
|
||||
identifier = parts_msg[1].strip()
|
||||
|
||||
# Поиск по ID
|
||||
# Делаем поиск по ID
|
||||
if identifier.isdigit():
|
||||
|
||||
# Делаем в int и ищем
|
||||
# Ищем пользователя в базе данных
|
||||
user_info = db.get_user(int(identifier))
|
||||
|
||||
# Если нашли пользователя
|
||||
if user_info:
|
||||
# Создаем объект пользователя из данных базы
|
||||
|
||||
# Создаем объект пользователя
|
||||
target_user = User(
|
||||
id=user_info[0],
|
||||
first_name=user_info[1],
|
||||
@ -118,14 +105,16 @@ def register_handlers(bot: AsyncTeleBot): # Регистрирует все об
|
||||
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],
|
||||
@ -133,17 +122,26 @@ def register_handlers(bot: AsyncTeleBot): # Регистрирует все об
|
||||
is_bot=False
|
||||
)
|
||||
|
||||
# Если пользователь не найден
|
||||
# Если пользователь не найден в базе данных
|
||||
if not target_user:
|
||||
await send_message(chat_id, COMMAND_MESSAGES['user_not_found'])
|
||||
|
||||
# Отправляем предупреждение
|
||||
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, 5)
|
||||
|
||||
await delete_messages(bot, message, time_sleep=5, number_message=2)
|
||||
return
|
||||
|
||||
# Проверяем статус целевого пользователя
|
||||
if await check_target_status(bot, message, target_user) == 1:
|
||||
return
|
||||
|
||||
# Выполняем размут
|
||||
try:
|
||||
|
||||
# Убираем ограничения (можно писать в чат)
|
||||
permissions = ChatPermissions(
|
||||
can_send_messages=True,
|
||||
@ -153,13 +151,14 @@ def register_handlers(bot: AsyncTeleBot): # Регистрирует все об
|
||||
can_add_web_page_previews=True,
|
||||
can_change_info=False,
|
||||
can_invite_users=True,
|
||||
can_pin_messages=False,
|
||||
can_pin_messages=False
|
||||
)
|
||||
|
||||
# Выполняем размут
|
||||
await bot.restrict_chat_member(
|
||||
chat_id=message.chat.id,
|
||||
user_id=target_user.id,
|
||||
permissions=permissions
|
||||
permissions=permissions,
|
||||
)
|
||||
|
||||
# Отправляем лог в админ-чат
|
||||
@ -169,26 +168,48 @@ def register_handlers(bot: AsyncTeleBot): # Регистрирует все об
|
||||
admin_id=message.from_user.id,
|
||||
reason=None,
|
||||
duration=None,
|
||||
photo_path=None
|
||||
photo_path=None,
|
||||
)
|
||||
|
||||
# Отправляем сообщения, что пользователь получил размут
|
||||
await send_message(chat_id, COMMAND_MESSAGES['unmuted'])
|
||||
await bot.send_message(
|
||||
chat_id=message.chat.id,
|
||||
text=COMMAND_MESSAGES['unmuted'],
|
||||
message_thread_id=message.message_thread_id,
|
||||
)
|
||||
|
||||
# Записываем действие в логи
|
||||
logger.info(f"Пользователь {target_user.id} получил размут от администратора {message.from_user.id}.")
|
||||
|
||||
# Удаляем сообщения через 5 секунд
|
||||
await delete_messages(bot, message, 5)
|
||||
await delete_messages(bot, message, time_sleep=5, number_message=2)
|
||||
|
||||
except Exception as e:
|
||||
await send_message(chat_id, f"⚠️ Ошибка: {str(e)}")
|
||||
|
||||
# Отправляем ошибку
|
||||
await bot.send_message(
|
||||
chat_id=message.chat.id,
|
||||
text=COMMAND_MESSAGES['error'].format(e=str(e)),
|
||||
message_thread_id=message.message_thread_id,
|
||||
)
|
||||
|
||||
# Записываем ошибку в логи
|
||||
logger.error(f"Ошибка размута: {str(e)}")
|
||||
|
||||
# Удаляем сообщения через 5 секунд
|
||||
await delete_messages(bot, message, 5)
|
||||
await delete_messages(bot, message, time_sleep=5, number_message=2)
|
||||
|
||||
except Exception as e:
|
||||
await send_message(chat_id, COMMAND_MESSAGES['general_error'])
|
||||
|
||||
# Отправляем ошибку
|
||||
await bot.send_message(
|
||||
chat_id=message.chat.id,
|
||||
text=COMMAND_MESSAGES['general_error'],
|
||||
message_thread_id=message.message_thread_id,
|
||||
)
|
||||
|
||||
# Записываем ошибку в логи
|
||||
logger.error(f"Общая ошибка в unmute_command: {str(e)}")
|
||||
|
||||
# Удаляем сообщения через 5 секунд
|
||||
await delete_messages(bot, message, 5)
|
||||
await delete_messages(bot, message, time_sleep=5, number_message=2)
|
||||
177
src/utils.py
Normal file
177
src/utils.py
Normal file
@ -0,0 +1,177 @@
|
||||
from telebot.async_telebot import AsyncTeleBot
|
||||
from telebot.types import Message
|
||||
import asyncio
|
||||
import logging
|
||||
|
||||
# Получаем логгер для текущего модуля
|
||||
logger = logging.getLogger(__name__)
|
||||
|
||||
# Удаляет определённое количество сообщения
|
||||
async def delete_messages(bot: AsyncTeleBot, message: Message, time_sleep: int, number_message: int):
|
||||
await asyncio.sleep(time_sleep)
|
||||
for i in range(number_message):
|
||||
await bot.delete_message(message.chat.id, message.message_id+i)
|
||||
|
||||
# Проверяет, является ли отправитель администратором
|
||||
async def check_admin_status(bot: AsyncTeleBot, message: Message):
|
||||
|
||||
if message.reply_to_message and message.is_topic_message is None:
|
||||
message.message_thread_id = None
|
||||
|
||||
try:
|
||||
|
||||
# Получаем статус отправителя
|
||||
admin_status = await bot.get_chat_member(message.chat.id, message.from_user.id)
|
||||
|
||||
# Проверка наличия прав администратора/создателя
|
||||
if admin_status.status not in ('administrator', 'creator'):
|
||||
|
||||
# Отправляем предупреждение
|
||||
await bot.send_message(
|
||||
chat_id=message.chat.id,
|
||||
text=COMMAND_MESSAGES['no_admin_rights'],
|
||||
message_thread_id=message.message_thread_id,
|
||||
)
|
||||
|
||||
# Удаляем сообщения через 5 секунд
|
||||
await delete_messages(bot, message, time_sleep=5, number_message=2)
|
||||
return 1
|
||||
|
||||
# Проверка права на ограничение участников
|
||||
if admin_status.status == 'administrator' and not admin_status.can_restrict_members:
|
||||
|
||||
# Отправляем предупреждение
|
||||
await bot.send_message(
|
||||
chat_id=message.chat.id,
|
||||
text=COMMAND_MESSAGES['no_restrict_rights'],
|
||||
message_thread_id=message.message_thread_id,
|
||||
)
|
||||
|
||||
# Удаляем сообщения через 5 секунд
|
||||
await delete_messages(bot, message, time_sleep=5, number_message=2)
|
||||
return 1
|
||||
|
||||
except Exception as e:
|
||||
|
||||
# Отправляем ошибку
|
||||
await bot.send_message(
|
||||
chat_id=message.chat.id,
|
||||
text=COMMAND_MESSAGES['error'].format(e=str(e)),
|
||||
message_thread_id=message.message_thread_id,
|
||||
)
|
||||
|
||||
# Записываем ошибку в логи
|
||||
logger.error(f"Ошибка при получении статуса администратора: {str(e)}")
|
||||
|
||||
# Удаляем сообщения через 5 секунд
|
||||
await delete_messages(bot, message, time_sleep=5, number_message=2)
|
||||
return 1
|
||||
|
||||
# Проверяет статус целевого пользователя
|
||||
async def check_target_status(bot: AsyncTeleBot, message: Message, target_user):
|
||||
|
||||
if message.reply_to_message and message.is_topic_message is None:
|
||||
message.message_thread_id = None
|
||||
|
||||
try:
|
||||
|
||||
# Получаем статус пользователя
|
||||
target_status = await bot.get_chat_member(
|
||||
chat_id=message.chat.id,
|
||||
user_id=target_user.id,
|
||||
)
|
||||
|
||||
# Проверяем, является ли цель администратором или создателем
|
||||
if target_status.status in ('administrator', 'creator'):
|
||||
|
||||
# Отправляем предупреждение
|
||||
await bot.send_message(
|
||||
chat_id=message.chat.id,
|
||||
text=COMMAND_MESSAGES['cant_mute_admin'],
|
||||
message_thread_id=message.message_thread_id,
|
||||
)
|
||||
|
||||
# Удаляем сообщения через 5 секунд
|
||||
await delete_messages(bot, message, time_sleep=5, number_message=2)
|
||||
return
|
||||
|
||||
except Exception as e:
|
||||
|
||||
# Отправляем ошибку
|
||||
await bot.send_message(
|
||||
chat_id=message.chat.id,
|
||||
text=COMMAND_MESSAGES['error'].format(e=str(e)),
|
||||
message_thread_id=message.message_thread_id,
|
||||
)
|
||||
|
||||
# Записываем ошибку в логи
|
||||
logger.error(f"Ошибка при получении статуса пользователя: {str(e)}")
|
||||
|
||||
# Удаляем сообщения через 5 секунд
|
||||
await delete_messages(bot, message, time_sleep=5, number_message=2)
|
||||
return
|
||||
|
||||
# Возвращает количество секунд
|
||||
def parse_mute_time(time_str: str) -> int | None:
|
||||
|
||||
# Нормализация входной строки
|
||||
normalized = time_str.strip().lower()
|
||||
|
||||
# Словарь для конвертации единиц времени в секунды
|
||||
time_units = {
|
||||
'm': 60, 'м': 60, # минуты
|
||||
'h': 3600, 'ч': 3600, # часы
|
||||
'd': 86400, 'д': 86400 # дни
|
||||
}
|
||||
|
||||
# Проверям, существует ли строка
|
||||
if not normalized:
|
||||
return None
|
||||
|
||||
# Проверяем последний символ (единица измерения)
|
||||
unit = normalized[-1]
|
||||
if unit not in time_units:
|
||||
return None
|
||||
|
||||
# Парсим числовую часть
|
||||
try:
|
||||
value = int(normalized[:-1])
|
||||
if value <= 0:
|
||||
return None
|
||||
|
||||
return value * time_units[unit]
|
||||
|
||||
except (ValueError, TypeError):
|
||||
return None
|
||||
|
||||
# Форматирует время в нормальный вид
|
||||
def format_mute_time(seconds: int) -> str:
|
||||
|
||||
# Обработка минут
|
||||
minutes = seconds // 60
|
||||
if minutes < 60:
|
||||
if minutes % 10 == 1 and minutes % 100 != 11:
|
||||
return f"{minutes} минуту"
|
||||
elif 2 <= minutes % 10 <= 4 and (minutes % 100 < 10 or minutes % 100 >= 20):
|
||||
return f"{minutes} минуты"
|
||||
else:
|
||||
return f"{minutes} минут"
|
||||
|
||||
# Обработка часов
|
||||
hours = minutes // 60
|
||||
if hours < 24:
|
||||
if hours % 10 == 1 and hours % 100 != 11:
|
||||
return f"{hours} час"
|
||||
elif 2 <= hours % 10 <= 4 and (hours % 100 < 10 or hours % 100 >= 20):
|
||||
return f"{hours} часа"
|
||||
else:
|
||||
return f"{hours} часов"
|
||||
|
||||
# Обработка дней
|
||||
days = hours // 24
|
||||
if days % 10 == 1 and days % 100 != 11:
|
||||
return f"{days} день"
|
||||
elif 2 <= days % 10 <= 4 and (days % 100 < 10 or days % 100 >= 20):
|
||||
return f"{days} дня"
|
||||
else:
|
||||
return f"{days} дней"
|
||||
Reference in New Issue
Block a user