Enhance bot functionality with new features and improvements

- Added a new `/status` endpoint in `server_prometheus.py` to provide process status information, including uptime and resource usage metrics.
- Implemented a PID manager in `run_helper.py` to track the bot's process, improving monitoring capabilities.
- Introduced a method to delete audio moderation records in `audio_repository.py`, enhancing database management.
- Updated voice message handling in callback handlers to ensure proper deletion of audio moderation records.
- Improved error handling and logging in various services, ensuring better tracking of media processing and file downloads.
- Refactored media handling functions to streamline operations and improve code readability.
- Enhanced metrics tracking for file downloads and media processing, providing better insights into bot performance.

helper_bot/handlers/callback/callback_handlers.py

@@ -273,6 +273,9 @@ async def save_voice_message(
             message_id=call.message.message_id
         )
         
+        # Удаляем запись из таблицы audio_moderate
+        await bot_db.delete_audio_moderate_record(call.message.message_id)
+        
         await call.answer(text='Сохранено!', cache_time=3)
         
     except Exception as e:
@@ -296,6 +299,9 @@ async def delete_voice_message(
             message_id=call.message.message_id
         )
         
+        # Удаляем запись из таблицы audio_moderate
+        await bot_db.delete_audio_moderate_record(call.message.message_id)
+        
         await call.answer(text='Удалено!', cache_time=3)
         
     except Exception as e:

helper_bot/handlers/callback/services.py

@@ -331,6 +331,12 @@ class BanService:
         self.group_for_posts = settings['Telegram']['group_for_posts']
         self.important_logs = settings['Telegram']['important_logs']
 
+    def _get_bot(self, message) -> Bot:
+        """Получает бота из контекста сообщения или использует переданного"""
+        if self.bot:
+            return self.bot
+        return message.bot
+
     @track_time("ban_user_from_post", "ban_service")
     @track_errors("ban_service", "ban_user_from_post")
     async def ban_user_from_post(self, call: CallbackQuery) -> None:

helper_bot/handlers/private/services.py

@@ -180,7 +180,9 @@ class PostService:
             created_at=int(datetime.now().timestamp())
         )
         await self.db.add_post(post)
-        await add_in_db_media(sent_message, self.db)
+        success = await add_in_db_media(sent_message, self.db)
+        if not success:
+            logger.warning(f"handle_photo_post: Не удалось сохранить медиа для поста {sent_message.message_id}")
     
     @track_time("handle_video_post", "post_service")
     @track_errors("post_service", "handle_video_post")
@@ -202,7 +204,9 @@ class PostService:
             created_at=int(datetime.now().timestamp())
         )
         await self.db.add_post(post)
-        await add_in_db_media(sent_message, self.db)
+        success = await add_in_db_media(sent_message, self.db)
+        if not success:
+            logger.warning(f"handle_photo_post: Не удалось сохранить медиа для поста {sent_message.message_id}")
     
     @track_time("handle_video_note_post", "post_service")
     @track_errors("post_service", "handle_video_note_post")
@@ -220,7 +224,9 @@ class PostService:
             created_at=int(datetime.now().timestamp())
         )
         await self.db.add_post(post)
-        await add_in_db_media(sent_message, self.db)
+        success = await add_in_db_media(sent_message, self.db)
+        if not success:
+            logger.warning(f"handle_photo_post: Не удалось сохранить медиа для поста {sent_message.message_id}")
     
     @track_time("handle_audio_post", "post_service")
     @track_errors("post_service", "handle_audio_post")
@@ -242,7 +248,9 @@ class PostService:
             created_at=int(datetime.now().timestamp())
         )
         await self.db.add_post(post)
-        await add_in_db_media(sent_message, self.db)
+        success = await add_in_db_media(sent_message, self.db)
+        if not success:
+            logger.warning(f"handle_photo_post: Не удалось сохранить медиа для поста {sent_message.message_id}")
     
     @track_time("handle_voice_post", "post_service")
     @track_errors("post_service", "handle_voice_post")
@@ -260,7 +268,9 @@ class PostService:
             created_at=int(datetime.now().timestamp())
         )
         await self.db.add_post(post)
-        await add_in_db_media(sent_message, self.db)
+        success = await add_in_db_media(sent_message, self.db)
+        if not success:
+            logger.warning(f"handle_photo_post: Не удалось сохранить медиа для поста {sent_message.message_id}")
     
     @track_time("handle_media_group_post", "post_service")
     @track_errors("post_service", "handle_media_group_post")    
@@ -283,7 +293,7 @@ class PostService:
         # Отправляем медиагруппу в группу для модерации
         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
+            self.settings.group_for_posts, message, media_group, self.db, main_post.message_id
         )
         
         await asyncio.sleep(0.2)

helper_bot/handlers/voice/services.py

@@ -299,10 +299,19 @@ class AudioFileService:
             logger.info(f"Получена информация о файле: {file_info.file_path}")
             
             downloaded_file = await bot.download_file(file_path=file_info.file_path)
-            logger.info(f"Файл скачан, размер: {len(downloaded_file.read()) if downloaded_file else 'None'} bytes")
             
-            # Сбрасываем позицию в файле
-            downloaded_file.seek(0)
+            # Проверяем что файл успешно скачан
+            if not downloaded_file:
+                logger.error("Не удалось скачать файл")
+                raise FileOperationError("Не удалось скачать файл")
+            
+            # Получаем размер файла без изменения позиции
+            current_pos = downloaded_file.tell()
+            downloaded_file.seek(0, 2)  # Переходим в конец файла
+            file_size = downloaded_file.tell()
+            downloaded_file.seek(current_pos)  # Возвращаемся в исходную позицию
+            
+            logger.info(f"Файл скачан, размер: {file_size} bytes")
             
             # Создаем директорию если она не существует
             import os
@@ -312,6 +321,9 @@ class AudioFileService:
             file_path = f'{VOICE_USERS_DIR}/{file_name}.ogg'
             logger.info(f"Сохраняем файл по пути: {file_path}")
             
+            # Сбрасываем позицию в файле перед сохранением
+            downloaded_file.seek(0)
+            
             # Сохраняем файл
             with open(file_path, 'wb') as new_file:
                 new_file.write(downloaded_file.read())