Добавлен функционал для работы с медиагруппами и улучшена обработка сообщений

- Реализованы методы для добавления связи между постами и сообщениями в `PostRepository` и `AsyncBotDB`.
- Обновлены обработчики публикации постов для корректной работы с медиагруппами, включая удаление и уведомление авторов.
- Улучшена логика обработки сообщений в `AlbumMiddleware` для более эффективного сбора медиагрупп.
- Обновлены тесты для проверки нового функционала и обработки ошибок.

helper_bot/handlers/private/services.py

@@ -323,7 +323,6 @@ class PostService:
     @track_media_processing("media_group")
     async def handle_media_group_post(self, message: types.Message, album: list, first_name: str) -> None:
         """Handle media group post submission"""
-        #TODO: Мне кажется тут какая-то дичь с одинаковыми переменными, в которых post_caption никуда не ведет
         post_caption = " "
         raw_caption = ""
         
@@ -331,12 +330,17 @@ class PostService:
             raw_caption = album[0].caption or ""
             post_caption = get_text_message(album[0].caption.lower(), first_name, message.from_user.username)
         
-        # Определяем анонимность на основе сырого caption
         is_anonymous = determine_anonymity(raw_caption)
+        media_group = await prepare_media_group_from_middlewares(album, post_caption)
+        
+        media_group_message_ids = await send_media_group_message_to_private_chat(
+            self.settings.group_for_posts, message, media_group, self.db, None, self.s3_storage
+        )
+        
+        main_post_id = media_group_message_ids[-1]
         
-        # Создаем основной пост для медиагруппы
         main_post = TelegramPost(
-            message_id=message.message_id,  # ID основного сообщения медиагруппы
+            message_id=main_post_id,
             text=raw_caption,
             author_id=message.from_user.id,
             created_at=int(datetime.now().timestamp()),
@@ -344,32 +348,32 @@ class PostService:
         )
         await self.db.add_post(main_post)
         
-        # Отправляем медиагруппу в группу для модерации
-        media_group = await prepare_media_group_from_middlewares(album, post_caption)
-        media_group_message_id = await send_media_group_message_to_private_chat(
-            self.settings.group_for_posts, message, media_group, self.db, main_post.message_id, self.s3_storage
-        )
+        for msg_id in media_group_message_ids:
+            await self.db.add_message_link(main_post_id, msg_id)
         
         await asyncio.sleep(0.2)
         
-        # Создаем helper сообщение с кнопками
         markup = get_reply_keyboard_for_post()
-        help_message = await send_text_message(self.settings.group_for_posts, message, "ВРУЧНУЮ ВЫКЛАДЫВАТЬ, ПОСЛЕ ВЫКЛАДКИ УДАЛИТЬ ОБА ПОСТА")
+        helper_message = await send_text_message(
+            self.settings.group_for_posts, 
+            message, 
+            "^",
+            markup
+        )
+        helper_message_id = helper_message.message_id
         
-        # Создаем helper пост и связываем его с основным
         helper_post = TelegramPost(
-            message_id=help_message.message_id,  # ID helper сообщения
-            text="^",  # Специальный маркер для медиагруппы
+            message_id=helper_message_id,
+            text="^",
             author_id=message.from_user.id,
-            helper_text_message_id=main_post.message_id,  # Ссылка на основной пост
+            helper_text_message_id=main_post_id,
             created_at=int(datetime.now().timestamp())
         )
         await self.db.add_post(helper_post)
         
-        # Обновляем основной пост, чтобы он ссылался на helper
         await self.db.update_helper_message(
-            message_id=main_post.message_id, 
-            helper_message_id=help_message.message_id
+            message_id=main_post_id, 
+            helper_message_id=helper_message_id
         )
     
     @track_time("process_post", "post_service")
@@ -378,13 +382,8 @@ class PostService:
     async def process_post(self, message: types.Message, album: Union[list, None] = None) -> None:
         """Process post based on content type"""
         first_name = get_first_name(message)
-        # TODO: Бесит меня этот функционал
+        
         if message.media_group_id is not None:
-            safe_username = html.escape(message.from_user.username) if message.from_user.username else "Без никнейма"
-            await send_text_message(
-                self.settings.group_for_logs, message,
-                f'Закинул медиагруппу, пользователь: имя - {first_name}, ник - {safe_username}'
-            )
             await self.handle_media_group_post(message, album, first_name)
             return