Добавлен функционал для работы с S3 хранилищем и обновление контента опубликованных постов

- В `env.example` добавлены настройки для S3 хранилища. - Обновлен файл зависимостей `requirements.txt`, добавлена библиотека `aioboto3` для работы с S3. - В `PostRepository` и `AsyncBotDB` реализованы методы для обновления и получения контента опубликованных постов. - Обновлены обработчики публикации постов для сохранения идентификаторов опубликованных сообщений и их контента. - Реализована логика сохранения медиафайлов в S3 или на локальный диск в зависимости от конфигурации. - Обновлены тесты для проверки нового функционала.
2026-01-23 23:19:16 +03:00
parent 42f168f329
commit fecac6091e
14 changed files with 992 additions and 143 deletions
--- a/database/repositories/post_repository.py
+++ b/database/repositories/post_repository.py
@@ -19,11 +19,19 @@ class PostRepository(DatabaseConnection):
                created_at INTEGER NOT NULL,
                status TEXT NOT NULL DEFAULT 'suggest',
                is_anonymous INTEGER,
+                published_message_id INTEGER,
                FOREIGN KEY (author_id) REFERENCES our_users (user_id) ON DELETE CASCADE
            )
        '''
        await self._execute_query(post_query)
        
+        # Добавляем поле published_message_id если его нет (для существующих БД)
+        try:
+            await self._execute_query('ALTER TABLE post_from_telegram_suggest ADD COLUMN published_message_id INTEGER')
+        except Exception:
+            # Поле уже существует, игнорируем ошибку
+            pass
+        
        # Таблица контента постов
        content_query = '''
            CREATE TABLE IF NOT EXISTS content_post_from_telegram (
@@ -47,6 +55,26 @@ class PostRepository(DatabaseConnection):
        '''
        await self._execute_query(link_query)
        
+        # Таблица контента опубликованных постов
+        published_content_query = '''
+            CREATE TABLE IF NOT EXISTS published_post_content (
+                published_message_id INTEGER NOT NULL,
+                content_name TEXT NOT NULL,
+                content_type TEXT,
+                published_at INTEGER NOT NULL,
+                PRIMARY KEY (published_message_id, content_name)
+            )
+        '''
+        await self._execute_query(published_content_query)
+        
+        # Создаем индексы
+        try:
+            await self._execute_query('CREATE INDEX IF NOT EXISTS idx_published_post_content_message_id ON published_post_content(published_message_id)')
+            await self._execute_query('CREATE INDEX IF NOT EXISTS idx_post_from_telegram_suggest_published ON post_from_telegram_suggest(published_message_id)')
+        except Exception:
+            # Индексы уже существуют, игнорируем ошибку
+            pass
+        
        self.logger.info("Таблицы для постов созданы")
    
    async def add_post(self, post: TelegramPost) -> None:
@@ -174,6 +202,20 @@ class PostRepository(DatabaseConnection):
        self.logger.info(f"Получен контент поста: {len(post_content)} элементов")
        return post_content
    
+    async def get_post_content_by_message_id(self, message_id: int) -> List[Tuple[str, str]]:
+        """Получает контент одиночного поста по message_id."""
+        query = """
+            SELECT cpft.content_name, cpft.content_type
+            FROM post_from_telegram_suggest pft
+            JOIN message_link_to_content mltc ON pft.message_id = mltc.post_id
+            JOIN content_post_from_telegram cpft ON cpft.message_id = mltc.message_id
+            WHERE pft.message_id = ? AND pft.helper_text_message_id IS NULL
+        """
+        post_content = await self._execute_query_with_result(query, (message_id,))
+        
+        self.logger.info(f"Получен контент одиночного поста: {len(post_content)} элементов для message_id={message_id}")
+        return post_content
+    
    async def get_post_text_by_helper_id(self, helper_message_id: int) -> Optional[str]:
        """Получает текст поста по helper_text_message_id."""
        query = "SELECT text FROM post_from_telegram_suggest WHERE helper_text_message_id = ?"
@@ -252,3 +294,40 @@ class PostRepository(DatabaseConnection):
            self.logger.info(f"Получены текст и is_anonymous для helper_message_id={helper_message_id}")
            return text, is_anonymous
        return None, None
+    
+    async def update_published_message_id(self, original_message_id: int, published_message_id: int) -> None:
+        """Обновляет published_message_id для опубликованного поста."""
+        query = "UPDATE post_from_telegram_suggest SET published_message_id = ? WHERE message_id = ?"
+        await self._execute_query(query, (published_message_id, original_message_id))
+        self.logger.info(f"Обновлен published_message_id: {original_message_id} -> {published_message_id}")
+    
+    async def add_published_post_content(
+        self, published_message_id: int, content_path: str, content_type: str
+    ) -> bool:
+        """Добавляет контент опубликованного поста."""
+        try:
+            from datetime import datetime
+            published_at = int(datetime.now().timestamp())
+            
+            query = """
+                INSERT OR IGNORE INTO published_post_content 
+                (published_message_id, content_name, content_type, published_at) 
+                VALUES (?, ?, ?, ?)
+            """
+            await self._execute_query(query, (published_message_id, content_path, content_type, published_at))
+            self.logger.info(f"Добавлен контент опубликованного поста: published_message_id={published_message_id}, type={content_type}")
+            return True
+        except Exception as e:
+            self.logger.error(f"Ошибка при добавлении контента опубликованного поста: {e}")
+            return False
+    
+    async def get_published_post_content(self, published_message_id: int) -> List[Tuple[str, str]]:
+        """Получает контент опубликованного поста."""
+        query = """
+            SELECT content_name, content_type
+            FROM published_post_content
+            WHERE published_message_id = ?
+        """
+        post_content = await self._execute_query_with_result(query, (published_message_id,))
+        self.logger.info(f"Получен контент опубликованного поста: {len(post_content)} элементов для published_message_id={published_message_id}")
+        return post_content