Merge pull request 'feature' (#1) from feature into master

Reviewed-on: #1

src/config.py

@@ -1,9 +1,24 @@
 # Директория, где хранятся модули
-MODULES_DIR = "modules"
+MODULES_DIR = 'modules'
 
 # Название файла db sqlite
-DATABASE_NAME = "users.db"
+DATABASE_NAME = 'users.db'
 
-# Текст для команд 
+# Сообщения команд
-MESSAGE_FOR_START = "Бот-модератор для чата @linux_gaming_ru"
+COMMAND_MESSAGES = {
-MESSAGE_FOR_HELP = "пусто"
+    'start': 'Бот-модератор для чата @linux_gaming_ru',
+    'help': 'пусто',
+    'manual_ban': (
+        'ℹ️ Использование бана:\n'
+        '1️⃣ Ответьте на сообщение: <code>/ban</code>\n'
+        '2️⃣ Укажите тэг: <code>/ban @username</code>\n'
+        '3️⃣ Укажите ID: <code>/ban 123456789</code>'
+    ),
+    'no_admin_rights': '❌ Только администраторы могут использовать эту команду.',
+    'no_restrict_rights': '❌ У вас недостаточно прав.',
+    'user_not_found': '❌ Пользователь не найден.',
+    'cant_ban_admin': '❌ Невозможно забанить администратора.',
+    'banned': '✅ Пользователь успешно забанен.',
+    'general_error': '⚠️ Произошла непредвиденная ошибка.'
+
+}

src/database.py

@@ -59,7 +59,25 @@ class Database: # Инициализация класса
     def get_user(self, user_id: int) -> Optional[Tuple]:
         with self._get_connection() as connect:
             cursor = connect.cursor()
-            cursor.execute('''SELECT id, nickname, tag FROM users WHERE id = ?''', (user_id,))
+            cursor.execute('''
+                SELECT id, nickname, tag 
+                FROM users 
+                WHERE id = ?
+                ''', (user_id,))
+            return cursor.fetchone()
+
+    # Находит пользователя по username (без @)
+    def get_user_by_username(self, username: str) -> Optional[Tuple]:
+        if not username:
+            return None
+            
+        with self._get_connection() as connect:
+            cursor = connect.cursor()
+            cursor.execute('''
+                SELECT id, nickname, tag 
+                FROM users 
+                WHERE LOWER(tag) = LOWER(?)
+                ''', (username,))
             return cursor.fetchone()
 
 # Создаем экземпляр базы данных для импорта в других модулях

src/modules/ban.py

@@ -0,0 +1,143 @@
+from telebot.async_telebot import AsyncTeleBot
+from telebot.types import Message, User
+import asyncio
+import logging
+from database import db
+
+from config import COMMAND_MESSAGES
+
+logger = logging.getLogger(__name__) # Получаем логгер для текущего модуля
+
+def register_handlers(bot: AsyncTeleBot): # Регистрирует все обработчики команд
+
+    @bot.message_handler(commands=['ban']) # Обработчик команды /ban
+    async def ban_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()
+
+        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'])
+                    return
+                    
+                # Проверяем право администратора на бан
+                if admin_status.status == 'administrator' and not admin_status.can_restrict_members:
+                    await send_message(chat_id, COMMAND_MESSAGES['no_restrict_rights'])
+                    return
+            
+            except Exception as e:
+                await send_message(chat_id, f"⚠️ Ошибка: {str(e)}")
+                logger.error(f"Ошибка при получении статуса администратора: {str(e)}")
+                return
+
+            # Случай №1 - Команда используется в ответ на сообщение
+            if len(parts_msg) == 1:
+
+                # Если банят в теме
+                if message.is_topic_message:
+
+                    # Если без ответа на сообщение
+                    if message.message_thread_id == message.reply_to_message.message_id:
+                        return
+
+                    # Если команда правильная
+                    else:
+                        target_user = message.reply_to_message.from_user
+
+                # Если банят в обычном чате
+                elif message.reply_to_message and message.is_topic_message is None:
+                    target_user = message.reply_to_message.from_user
+                
+                # Не выводим сообщение, что команда неправильная
+                else:
+                    return
+            
+            # Случай №2 - Команда с аргументом (/ban @username или /ban 12345)
+            elif len(parts_msg) == 2:
+
+                # Выводим помощь (/ban help)
+                if parts_msg[1].strip() in ['help', 'помощь']:
+                    await send_message(chat_id, COMMAND_MESSAGES['manual_ban'])
+                    return
+                
+                identifier = parts_msg[1].strip()
+                
+                # Поиск по 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],
+                            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
+                        )
+
+            # Если пользователь не найден
+            if not target_user:
+                await send_message(chat_id, COMMAND_MESSAGES['user_not_found'])
+                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'])
+                    return
+                    
+            except Exception as e:
+                await send_message(chat_id, f"⚠️ Ошибка: {str(e)}")
+                logger.error(f"Ошибка при получении статуса пользователя: {str(e)}")
+                return
+
+            # Выполняем бан
+            try:
+                await bot.ban_chat_member(message.chat.id, target_user.id)
+                await send_message(chat_id, COMMAND_MESSAGES['banned'])
+                
+                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)
+                logger.info(f"Пользователь {target_user.id} забанен администратором {message.from_user.id}.")
+                
+            except Exception as e:
+                await send_message(chat_id, f"⚠️ Ошибка: {str(e)}")
+                logger.error(f"Ошибка бана: {str(e)}")
+                
+        except Exception as e:
+            await send_message(chat_id, COMMAND_MESSAGES['general_error'])
+            logger.error(f"Общая ошибка в ban_command: {str(e)}")

src/modules/chat_events.py

@@ -0,0 +1,21 @@
+from telebot.async_telebot import AsyncTeleBot
+from telebot.types import Message
+import asyncio
+
+import logging
+
+logger = logging.getLogger(__name__) # Получаем логгер для текущего модуля
+
+def register_handlers(bot: AsyncTeleBot): # Регистрирует все обработчики событий
+
+    @bot.message_handler(content_types=['new_chat_members']) # Обработчик захода
+    async def handle_new_members(message: Message):
+        await asyncio.sleep(10)
+        await bot.delete_message(message.chat.id, message.message_id)
+        logger.info(f"Пользователь {message.from_user.id} зашёл в чат.")
+
+    @bot.message_handler(content_types=['left_chat_member']) # Обработчик выхода
+    async def handle_left_members(message: Message):
+        await asyncio.sleep(10)
+        await bot.delete_message(message.chat.id, message.message_id)
+        logger.info(f"Пользователь {message.from_user.id} вышел из чата.")

src/modules/start.py

@@ -1,7 +1,7 @@
 from telebot.async_telebot import AsyncTeleBot
 import logging
 
-from config import MESSAGE_FOR_START
+from config import COMMAND_MESSAGES
 
 logger = logging.getLogger(__name__) # Получаем логгер для текущего модуля
 
@@ -9,8 +9,15 @@ def register_handlers(bot: AsyncTeleBot): # Регистрирует все об
 
     @bot.message_handler(commands=['start']) # Обработчик команды /start
     async def start_command(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:
-            logger.info(f"Команда START ({message.from_user.id})")
+
-            await bot.send_message(message.chat.id, MESSAGE_FOR_START)
+            await send_message(chat_id, COMMAND_MESSAGES['start'])
+            logger.info(f"Пользователь {message.from_user.id} запустил /start")
+
         except Exception as e:
-            logger.error(f"Команда START ({message.from_user.id}) {str(e)}")
+            logger.error(f"Пользователь {message.from_user.id} запустил /start: {str(e)}")