Реализован функцоинал хранения сырых текстов поста в базе данных. Оформление поста происходит непосредственно перед его отправкой в канал.

- Реализованы методы `get_post_text_and_anonymity_by_message_id` и `get_post_text_and_anonymity_by_helper_id` в `PostRepository` для получения текста поста и флага анонимности. - Обновлена модель `TelegramPost`, добавлено поле `is_anonymous`. - Изменена схема базы данных для включения поля `is_anonymous` в таблицу `post_from_telegram_suggest`. - Обновлены функции публикации постов в `PostPublishService` для учета анонимности. - Добавлены тесты для проверки новых функций и корректности работы с анонимностью.
2026-01-23 12:12:21 +03:00
parent c6ba90552d
commit 89022aedaf
12 changed files with 1064 additions and 48 deletions
--- a/helper_bot/handlers/private/services.py
+++ b/helper_bot/handlers/private/services.py
@@ -13,11 +13,13 @@ from dataclasses import dataclass
 from aiogram import types
 from aiogram.types import FSInputFile
 from database.models import TelegramPost, User
+from logs.custom_logger import logger

 # Local imports - utilities
 from helper_bot.utils.helper_func import (
    get_first_name, 
    get_text_message, 
+    determine_anonymity,
    send_text_message, 
    send_photo_message,
    send_media_group_message_to_private_chat, 
@@ -154,11 +156,17 @@ class PostService:
        markup = get_reply_keyboard_for_post()
        
        sent_message_id = await send_text_message(self.settings.group_for_posts, message, post_text, markup)
+        
+        # Сохраняем сырой текст и определяем анонимность
+        raw_text = message.text or ""
+        is_anonymous = determine_anonymity(raw_text)
+        
        post = TelegramPost(
            message_id=sent_message_id,
-            text=message.text,
+            text=raw_text,
            author_id=message.from_user.id,
-            created_at=int(datetime.now().timestamp())
+            created_at=int(datetime.now().timestamp()),
+            is_anonymous=is_anonymous
        )
        await self.db.add_post(post)
    
@@ -176,11 +184,16 @@ class PostService:
            self.settings.group_for_posts, message, message.photo[-1].file_id, post_caption, markup
        )
        
+        # Сохраняем сырой caption и определяем анонимность
+        raw_caption = message.caption or ""
+        is_anonymous = determine_anonymity(raw_caption)
+        
        post = TelegramPost(
            message_id=sent_message.message_id,
-            text=sent_message.caption or "",
+            text=raw_caption,
            author_id=message.from_user.id,
-            created_at=int(datetime.now().timestamp())
+            created_at=int(datetime.now().timestamp()),
+            is_anonymous=is_anonymous
        )
        await self.db.add_post(post)
        success = await add_in_db_media(sent_message, self.db)
@@ -201,11 +214,16 @@ class PostService:
            self.settings.group_for_posts, message, message.video.file_id, post_caption, markup
        )
        
+        # Сохраняем сырой caption и определяем анонимность
+        raw_caption = message.caption or ""
+        is_anonymous = determine_anonymity(raw_caption)
+        
        post = TelegramPost(
            message_id=sent_message.message_id,
-            text=sent_message.caption or "",
+            text=raw_caption,
            author_id=message.from_user.id,
-            created_at=int(datetime.now().timestamp())
+            created_at=int(datetime.now().timestamp()),
+            is_anonymous=is_anonymous
        )
        await self.db.add_post(post)
        success = await add_in_db_media(sent_message, self.db)
@@ -222,11 +240,16 @@ class PostService:
            self.settings.group_for_posts, message, message.video_note.file_id, markup
        )
        
+        # Сохраняем пустую строку, так как video_note не имеет caption
+        raw_caption = ""
+        is_anonymous = determine_anonymity(raw_caption)
+        
        post = TelegramPost(
            message_id=sent_message.message_id,
-            text=sent_message.caption or "",
+            text=raw_caption,
            author_id=message.from_user.id,
-            created_at=int(datetime.now().timestamp())
+            created_at=int(datetime.now().timestamp()),
+            is_anonymous=is_anonymous
        )
        await self.db.add_post(post)
        success = await add_in_db_media(sent_message, self.db)
@@ -247,11 +270,16 @@ class PostService:
            self.settings.group_for_posts, message, message.audio.file_id, post_caption, markup
        )
        
+        # Сохраняем сырой caption и определяем анонимность
+        raw_caption = message.caption or ""
+        is_anonymous = determine_anonymity(raw_caption)
+        
        post = TelegramPost(
            message_id=sent_message.message_id,
-            text=sent_message.caption or "",
+            text=raw_caption,
            author_id=message.from_user.id,
-            created_at=int(datetime.now().timestamp())
+            created_at=int(datetime.now().timestamp()),
+            is_anonymous=is_anonymous
        )
        await self.db.add_post(post)
        success = await add_in_db_media(sent_message, self.db)
@@ -268,11 +296,16 @@ class PostService:
            self.settings.group_for_posts, message, message.voice.file_id, markup
        )
        
+        # Сохраняем пустую строку, так как voice не имеет caption
+        raw_caption = ""
+        is_anonymous = determine_anonymity(raw_caption)
+        
        post = TelegramPost(
            message_id=sent_message.message_id,
-            text=sent_message.caption or "",
+            text=raw_caption,
            author_id=message.from_user.id,
-            created_at=int(datetime.now().timestamp())
+            created_at=int(datetime.now().timestamp()),
+            is_anonymous=is_anonymous
        )
        await self.db.add_post(post)
        success = await add_in_db_media(sent_message, self.db)
@@ -285,17 +318,24 @@ 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 = ""
        
        if album and album[0].caption:
+            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)
+        
        # Создаем основной пост для медиагруппы
        main_post = TelegramPost(
            message_id=message.message_id,  # ID основного сообщения медиагруппы
-            text=post_caption,
+            text=raw_caption,
            author_id=message.from_user.id,
-            created_at=int(datetime.now().timestamp())
+            created_at=int(datetime.now().timestamp()),
+            is_anonymous=is_anonymous
        )
        await self.db.add_post(main_post)