Implement audio record management features in AsyncBotDB and AudioRepository

- Added methods to delete audio moderation records and retrieve all audio records in async_db.py. - Enhanced AudioRepository with functionality to delete audio records by file name and retrieve all audio message records. - Improved logging for audio record operations to enhance monitoring and debugging capabilities. - Updated related handlers to ensure proper integration of new audio management features.
2025-09-05 01:31:50 +03:00
parent fc0517c011
commit 5f6882d348
32 changed files with 2661 additions and 214 deletions
--- a/helper_bot/handlers/private/services.py
+++ b/helper_bot/handlers/private/services.py
@@ -33,10 +33,11 @@ from helper_bot.keyboards import get_reply_keyboard_for_post

 # Local imports - metrics
 from helper_bot.utils.metrics import (
-    metrics, 
    track_time, 
    track_errors, 
-    db_query_time
+    db_query_time,
+    track_media_processing,
+    track_file_operations
 )


@@ -74,12 +75,14 @@ class UserService:
    
    @track_time("update_user_activity", "user_service")
    @track_errors("user_service", "update_user_activity")
+    @db_query_time("update_user_activity", "users", "update")
    async def update_user_activity(self, user_id: int) -> None:
        """Update user's last activity timestamp with metrics tracking"""
        await self.db.update_user_date(user_id)
    
    @track_time("ensure_user_exists", "user_service")
    @track_errors("user_service", "ensure_user_exists")
+    @db_query_time("ensure_user_exists", "users", "insert")
    async def ensure_user_exists(self, message: types.Message) -> None:
        """Ensure user exists in database, create if needed with metrics tracking"""
        user_id = message.from_user.id
@@ -89,43 +92,41 @@ class UserService:
        is_bot = message.from_user.is_bot
        language_code = message.from_user.language_code
        
-        if not await self.db.user_exists(user_id):
-            # Create User object with current timestamp
-            current_timestamp = int(datetime.now().timestamp())
-            user = User(
-                user_id=user_id,
-                first_name=first_name,
-                full_name=full_name,
-                username=username,
-                is_bot=is_bot,
-                language_code=language_code,
-                emoji="",
-                has_stickers=False,
-                date_added=current_timestamp,
-                date_changed=current_timestamp,
-                voice_bot_welcome_received=False
-            )
-            await self.db.add_user(user)
-            metrics.record_db_query("add_user", 0.0, "users", "insert")
-        else:
-            is_need_update = await check_username_and_full_name(user_id, username, full_name, self.db)
-            if is_need_update:
-                await self.db.update_user_info(user_id, username, full_name)
-                metrics.record_db_query("update_username_fullname", 0.0, "users", "update")
-                safe_full_name = html.escape(full_name) if full_name else "Неизвестный пользователь"
-                safe_username = html.escape(username) if username else "Без никнейма"
-                
-                await message.answer(
-                    f"Давно не виделись! Вижу что ты изменился;) Теперь буду звать тебя: {safe_full_name} и ник @{safe_username}")
-                await message.bot.send_message(
-                    chat_id=self.settings.group_for_logs,
-                    text=f'Для пользователя: {user_id} обновлены данные в БД.\nНовое имя: {safe_full_name}\nНовый ник:{safe_username}')
+        # Create User object with current timestamp
+        current_timestamp = int(datetime.now().timestamp())
+        user = User(
+            user_id=user_id,
+            first_name=first_name,
+            full_name=full_name,
+            username=username,
+            is_bot=is_bot,
+            language_code=language_code,
+            emoji="",
+            has_stickers=False,
+            date_added=current_timestamp,
+            date_changed=current_timestamp,
+            voice_bot_welcome_received=False
+        )
+        
+        # Пытаемся создать пользователя (если уже существует - игнорируем)
+        # Это устраняет race condition и упрощает логику
+        await self.db.add_user(user)
+        
+        # Проверяем, нужно ли обновить информацию о существующем пользователе
+        is_need_update = await check_username_and_full_name(user_id, username, full_name, self.db)
+        if is_need_update:
+            await self.db.update_user_info(user_id, username, full_name)
+            safe_full_name = html.escape(full_name) if full_name else "Неизвестный пользователь"
+            safe_username = html.escape(username) if username else "Без никнейма"
+            
+            await message.answer(
+                f"Давно не виделись! Вижу что ты изменился;) Теперь буду звать тебя: {safe_full_name} и ник @{safe_username}")
+            await message.bot.send_message(
+                chat_id=self.settings.group_for_logs,
+                text=f'Для пользователя: {user_id} обновлены данные в БД.\nНовое имя: {safe_full_name}\nНовый ник:{safe_username}')
        
        await self.db.update_user_date(user_id)
-        metrics.record_db_query("update_user_date", 0.0, "users", "update")
-    

-    @track_errors("user_service", "log_user_message")
    async def log_user_message(self, message: types.Message) -> None:
        """Forward user message to logs group with metrics tracking"""
        await message.forward(chat_id=self.settings.group_for_logs)
@@ -146,6 +147,7 @@ class PostService:
    
    @track_time("handle_text_post", "post_service")
    @track_errors("post_service", "handle_text_post")
+    @db_query_time("handle_text_post", "posts", "insert")
    async def handle_text_post(self, message: types.Message, first_name: str) -> None:
        """Handle text post submission"""
        post_text = get_text_message(message.text.lower(), first_name, message.from_user.username)
@@ -162,6 +164,7 @@ class PostService:
    
    @track_time("handle_photo_post", "post_service")
    @track_errors("post_service", "handle_photo_post")
+    @db_query_time("handle_photo_post", "posts", "insert")
    async def handle_photo_post(self, message: types.Message, first_name: str) -> None:
        """Handle photo post submission"""
        post_caption = ""
@@ -186,6 +189,7 @@ class PostService:
    
    @track_time("handle_video_post", "post_service")
    @track_errors("post_service", "handle_video_post")
+    @db_query_time("handle_video_post", "posts", "insert")
    async def handle_video_post(self, message: types.Message, first_name: str) -> None:
        """Handle video post submission"""
        post_caption = ""
@@ -210,6 +214,7 @@ class PostService:
    
    @track_time("handle_video_note_post", "post_service")
    @track_errors("post_service", "handle_video_note_post")
+    @db_query_time("handle_video_note_post", "posts", "insert")
    async def handle_video_note_post(self, message: types.Message) -> None:
        """Handle video note post submission"""
        markup = get_reply_keyboard_for_post()
@@ -230,6 +235,7 @@ class PostService:
    
    @track_time("handle_audio_post", "post_service")
    @track_errors("post_service", "handle_audio_post")
+    @db_query_time("handle_audio_post", "posts", "insert")
    async def handle_audio_post(self, message: types.Message, first_name: str) -> None:
        """Handle audio post submission"""
        post_caption = ""
@@ -254,6 +260,7 @@ class PostService:
    
    @track_time("handle_voice_post", "post_service")
    @track_errors("post_service", "handle_voice_post")
+    @db_query_time("handle_voice_post", "posts", "insert")
    async def handle_voice_post(self, message: types.Message) -> None:
        """Handle voice post submission"""
        markup = get_reply_keyboard_for_post()
@@ -273,7 +280,9 @@ class PostService:
            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")    
+    @track_errors("post_service", "handle_media_group_post") 
+    @db_query_time("handle_media_group_post", "posts", "insert")
+    @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"""
        post_caption = " "
@@ -320,6 +329,7 @@ class PostService:
    
    @track_time("process_post", "post_service")
    @track_errors("post_service", "process_post")
+    @track_media_processing("media_group")
    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)
@@ -360,6 +370,7 @@ class StickerService:
    
    @track_time("send_random_hello_sticker", "sticker_service")
    @track_errors("sticker_service", "send_random_hello_sticker")
+    @track_file_operations("sticker")
    async def send_random_hello_sticker(self, message: types.Message) -> None:
        """Send random hello sticker with metrics tracking"""
        name_stick_hello = list(Path('Stick').rglob('Hello_*'))
@@ -372,6 +383,7 @@ class StickerService:
    
    @track_time("send_random_goodbye_sticker", "sticker_service")
    @track_errors("sticker_service", "send_random_goodbye_sticker")
+    @track_file_operations("sticker")
    async def send_random_goodbye_sticker(self, message: types.Message) -> None:
        """Send random goodbye sticker with metrics tracking"""
        name_stick_bye = list(Path('Stick').rglob('Universal_*'))