Enhance admin handlers with improved logging and error handling

- Added detailed logging for user ban processing in `process_ban_target` and `process_ban_reason` functions, including user data and error messages. - Improved error handling for user input validation and database interactions. - Updated `return_to_admin_menu` function to log user return actions. - Enhanced media group handling in `PostPublishService` with better error logging and author ID retrieval. - Added new button options in voice handlers and updated keyboard layouts for improved user interaction. - Refactored album middleware to better handle media group messages and added documentation for clarity.
2025-09-02 22:20:34 +03:00
parent 1c6a37bc12
commit 1ab427a7ba
12 changed files with 340 additions and 232 deletions
--- a/helper_bot/handlers/callback/services.py
+++ b/helper_bot/handlers/callback/services.py
@@ -1,5 +1,5 @@
-import html
 from datetime import datetime, timedelta
+import html
 from typing import Dict, Any

 from aiogram import Bot
@@ -42,9 +42,14 @@ class PostPublishService:

    async def publish_post(self, call: CallbackQuery) -> None:
        """Основной метод публикации поста"""
+        # Проверяем, является ли сообщение частью медиагруппы
+        if call.message.media_group_id:
+            await self._publish_media_group(call)
+            return
+            
        content_type = call.message.content_type
        
-        if content_type == CONTENT_TYPE_TEXT and call.message.text != CONTENT_TYPE_MEDIA_GROUP:
+        if content_type == CONTENT_TYPE_TEXT:
            await self._publish_text_post(call)
        elif content_type == CONTENT_TYPE_PHOTO:
            await self._publish_photo_post(call)
@@ -56,8 +61,6 @@ class PostPublishService:
            await self._publish_audio_post(call)
        elif content_type == CONTENT_TYPE_VOICE:
            await self._publish_voice_post(call)
-        elif call.message.text == CONTENT_TYPE_MEDIA_GROUP:
-            await self._publish_media_group(call)
        else:
            raise PublishError(f"Неподдерживаемый тип контента: {content_type}")

@@ -115,51 +118,111 @@ class PostPublishService:

    async def _publish_media_group(self, call: CallbackQuery) -> None:
        """Публикация медиагруппы"""
-        post_content = await self.db.get_post_content_from_telegram_by_last_id(call.message.message_id)
-        pre_text = await self.db.get_post_text_from_telegram_by_last_id(call.message.message_id)
-        post_text = html.escape(str(pre_text))
-        author_id = await self._get_author_id_for_media_group(call.message.message_id)
-        
-        await send_media_group_to_channel(bot=self._get_bot(call.message), chat_id=self.main_public, post_content=post_content, post_text=post_text)
-        await self._delete_media_group_and_notify_author(call, author_id)
+        logger.info(f"Начинаю публикацию медиагруппы. Helper message ID: {call.message.message_id}")
+        try:
+            # call.message.message_id - это ID helper сообщения
+            helper_message_id = call.message.message_id
+            
+            # Получаем контент медиагруппы по helper_message_id
+            logger.debug(f"Получаю контент медиагруппы для helper_message_id: {helper_message_id}")
+            post_content = await self.db.get_post_content_by_helper_id(helper_message_id)
+            if not post_content:
+                logger.error(f"Контент медиагруппы не найден в базе данных для helper_message_id: {helper_message_id}")
+                raise PublishError("Контент медиагруппы не найден в базе данных")
+            
+            # Получаем текст поста по helper_message_id
+            logger.debug(f"Получаю текст поста для helper_message_id: {helper_message_id}")
+            pre_text = await self.db.get_post_text_by_helper_id(helper_message_id)
+            post_text = html.escape(str(pre_text)) if pre_text else ""
+            logger.debug(f"Текст поста получен: {'пустой' if not post_text else f'длина: {len(post_text)} символов'}")
+            
+            # Получаем ID автора по helper_message_id
+            logger.debug(f"Получаю ID автора для helper_message_id: {helper_message_id}")
+            author_id = await self.db.get_author_id_by_helper_message_id(helper_message_id)
+            if not author_id:
+                logger.error(f"Автор не найден для медиагруппы {helper_message_id}")
+                raise PostNotFoundError(f"Автор не найден для медиагруппы {helper_message_id}")
+            logger.debug(f"ID автора получен: {author_id}")
+            
+            # Отправляем медиагруппу в канал
+            logger.info(f"Отправляю медиагруппу в канал {self.main_public}")
+            await send_media_group_to_channel(
+                bot=self._get_bot(call.message), 
+                chat_id=self.main_public, 
+                post_content=post_content, 
+                post_text=post_text
+            )
+            
+            logger.debug(f"Удаляю медиагруппу и уведомляю автора {author_id}")
+            await self._delete_media_group_and_notify_author(call, author_id)
+            logger.info(f'Медиагруппа опубликована в канале {self.main_public}.')
+            
+        except Exception as e:
+            logger.error(f"Ошибка при публикации медиагруппы: {e}")
+            raise PublishError(f"Не удалось опубликовать медиагруппу: {str(e)}")

    async def decline_post(self, call: CallbackQuery) -> None:
        """Отклонение поста"""
+        logger.info(f"Начинаю отклонение поста. Message ID: {call.message.message_id}, Content type: {call.message.content_type}")
+        # Проверяем, является ли сообщение частью медиагруппы (осознанный костыль, т.к. сообщение к которому прикреплен коллбек без медиагруппы)
+        if call.message.text == CONTENT_TYPE_MEDIA_GROUP:
+            logger.debug("Сообщение является частью медиагруппы, вызываю _decline_media_group")
+            await self._decline_media_group(call)
+            return
+            
        content_type = call.message.content_type
        
-        if (content_type == CONTENT_TYPE_TEXT and call.message.text != CONTENT_TYPE_MEDIA_GROUP) or \
-           content_type in [CONTENT_TYPE_PHOTO, CONTENT_TYPE_AUDIO, CONTENT_TYPE_VOICE, CONTENT_TYPE_VIDEO, CONTENT_TYPE_VIDEO_NOTE]:
+        if content_type in [CONTENT_TYPE_TEXT, CONTENT_TYPE_PHOTO, CONTENT_TYPE_AUDIO, 
+                           CONTENT_TYPE_VOICE, CONTENT_TYPE_VIDEO, CONTENT_TYPE_VIDEO_NOTE]:
+            logger.debug(f"Отклоняю одиночный пост типа: {content_type}")
            await self._decline_single_post(call)
-        elif call.message.text == CONTENT_TYPE_MEDIA_GROUP:
-            await self._decline_media_group(call)
        else:
+            logger.error(f"Неподдерживаемый тип контента для отклонения: {content_type}")
            raise PublishError(f"Неподдерживаемый тип контента для отклонения: {content_type}")

    async def _decline_single_post(self, call: CallbackQuery) -> None:
        """Отклонение одиночного поста"""
+        logger.debug(f"Отклоняю одиночный пост. Message ID: {call.message.message_id}")
        author_id = await self._get_author_id(call.message.message_id)
+        logger.debug(f"ID автора получен: {author_id}")
+        
+        logger.debug(f"Удаляю сообщение из группы {self.group_for_posts}")
        await self._get_bot(call.message).delete_message(chat_id=self.group_for_posts, message_id=call.message.message_id)
+        
        try:
+            logger.debug(f"Отправляю уведомление об отклонении автору {author_id}")
            await send_text_message(author_id, call.message, MESSAGE_POST_DECLINED)
        except Exception as e:
            if str(e) == ERROR_BOT_BLOCKED:
+                logger.warning(f"Пользователь {author_id} заблокировал бота")
                raise UserBlockedBotError("Пользователь заблокировал бота")
+            logger.error(f"Ошибка при отправке уведомления автору {author_id}: {e}")
            raise
        logger.info(f'Сообщение отклонено админом {call.from_user.full_name} (ID: {call.from_user.id}).')

    async def _decline_media_group(self, call: CallbackQuery) -> None:
        """Отклонение медиагруппы"""
+        logger.debug(f"Отклоняю медиагруппу. Helper message ID: {call.message.message_id}")
+        
        post_ids = await self.db.get_post_ids_from_telegram_by_last_id(call.message.message_id)
-        message_ids = [row[0] for row in post_ids]
+        message_ids = post_ids.copy()
        message_ids.append(call.message.message_id)
+        logger.debug(f"Получены ID сообщений для удаления: {message_ids}")
        
        author_id = await self._get_author_id_for_media_group(call.message.message_id)
+        logger.debug(f"ID автора медиагруппы получен: {author_id}")
+        
+        logger.debug(f"Удаляю {len(message_ids)} сообщений из группы {self.group_for_posts}")
        await self._get_bot(call.message).delete_messages(chat_id=self.group_for_posts, message_ids=message_ids)
+        
        try:
+            logger.debug(f"Отправляю уведомление об отклонении автору медиагруппы {author_id}")
            await send_text_message(author_id, call.message, MESSAGE_POST_DECLINED)
        except Exception as e:
            if str(e) == ERROR_BOT_BLOCKED:
+                logger.warning(f"Пользователь {author_id} заблокировал бота")
                raise UserBlockedBotError("Пользователь заблокировал бота")
+            logger.error(f"Ошибка при отправке уведомления автору медиагруппы {author_id}: {e}")
            raise

    async def _get_author_id(self, message_id: int) -> int:
@@ -171,7 +234,27 @@ class PostPublishService:

    async def _get_author_id_for_media_group(self, message_id: int) -> int:
        """Получение ID автора для медиагруппы"""
+        # Сначала пытаемся найти автора по helper_message_id
        author_id = await self.db.get_author_id_by_helper_message_id(message_id)
+        if author_id:
+            return author_id
+        
+        # Если не найден, ищем по основному message_id медиагруппы
+        # Для этого нужно найти связанные сообщения медиагруппы
+        try:
+            # Получаем все ID сообщений медиагруппы
+            post_ids = await self.db.get_post_ids_from_telegram_by_last_id(message_id)
+            if post_ids:
+                # Берем первый ID (основное сообщение медиагруппы)
+                main_message_id = post_ids[0]
+                author_id = await self.db.get_author_id_by_message_id(main_message_id)
+                if author_id:
+                    return author_id
+        except Exception as e:
+            logger.warning(f"Не удалось найти автора через связанные сообщения: {e}")
+        
+        # Если все способы не сработали, ищем напрямую
+        author_id = await self.db.get_author_id_by_message_id(message_id)
        if not author_id:
            raise PostNotFoundError(f"Автор не найден для медиагруппы {message_id}")
        return author_id
@@ -190,9 +273,10 @@ class PostPublishService:
    async def _delete_media_group_and_notify_author(self, call: CallbackQuery, author_id: int) -> None:
        """Удаление медиагруппы и уведомление автора"""
        post_ids = await self.db.get_post_ids_from_telegram_by_last_id(call.message.message_id)
-        message_ids = [row[0] for row in post_ids]
-        message_ids.append(call.message.message_id)
-        await self._get_bot(call.message).delete_messages(chat_id=self.group_for_posts, message_ids=message_ids)
+
+        #message_ids = post_ids.copy()
+        post_ids.append(call.message.message_id)
+        await self._get_bot(call.message).delete_messages(chat_id=self.group_for_posts, message_ids=post_ids)
        try:
            await send_text_message(author_id, call.message, MESSAGE_POST_PUBLISHED)
        except Exception as e: