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/utils/metrics.py

@@ -52,6 +52,13 @@ class BotMetrics:
             registry=self.registry
         )
         
+        # Total users gauge (отдельная метрика)
+        self.total_users = Gauge(
+            'total_users',
+            'Total number of users in database',
+            registry=self.registry
+        )
+        
         # Database query metrics
         self.db_query_duration_seconds = Histogram(
             'db_query_duration_seconds',
@@ -110,6 +117,46 @@ class BotMetrics:
             ['activity_type', 'user_type', 'chat_type'],
             registry=self.registry
         )
+        
+        # File download metrics
+        self.file_downloads_total = Counter(
+            'file_downloads_total',
+            'Total number of file downloads',
+            ['content_type', 'status'],
+            registry=self.registry
+        )
+        
+        self.file_download_duration_seconds = Histogram(
+            'file_download_duration_seconds',
+            'Time spent downloading files',
+            ['content_type'],
+            buckets=[0.1, 0.5, 1.0, 2.5, 5.0, 10.0, 30.0, 60.0],
+            registry=self.registry
+        )
+        
+        self.file_download_size_bytes = Histogram(
+            'file_download_size_bytes',
+            'Size of downloaded files in bytes',
+            ['content_type'],
+            buckets=[1024, 10240, 102400, 1048576, 10485760, 104857600, 1073741824],
+            registry=self.registry
+        )
+        
+        # Media processing metrics
+        self.media_processing_total = Counter(
+            'media_processing_total',
+            'Total number of media processing operations',
+            ['content_type', 'status'],
+            registry=self.registry
+        )
+        
+        self.media_processing_duration_seconds = Histogram(
+            'media_processing_duration_seconds',
+            'Time spent processing media',
+            ['content_type'],
+            buckets=[0.1, 0.5, 1.0, 2.5, 5.0, 10.0, 30.0],
+            registry=self.registry
+        )
     
     def record_command(self, command_type: str, handler_type: str = "unknown", user_type: str = "unknown", status: str = "success"):
         """Record a bot command execution."""
@@ -140,6 +187,10 @@ class BotMetrics:
         """Set the number of active users for a specific type."""
         self.active_users.labels(user_type=user_type).set(count)
     
+    def set_total_users(self, count: int):
+        """Set the total number of users in database."""
+        self.total_users.set(count)
+    
     def record_db_query(self, query_type: str, duration: float, table_name: str = "unknown", operation: str = "unknown"):
         """Record database query duration."""
         self.db_query_duration_seconds.labels(
@@ -168,6 +219,54 @@ class BotMetrics:
             status=status
         ).observe(duration)
     
+    def record_file_download(self, content_type: str, file_size: int, duration: float):
+        """Record file download metrics."""
+        self.file_downloads_total.labels(
+            content_type=content_type,
+            status="success"
+        ).inc()
+        
+        self.file_download_duration_seconds.labels(
+            content_type=content_type
+        ).observe(duration)
+        
+        self.file_download_size_bytes.labels(
+            content_type=content_type
+        ).observe(file_size)
+    
+    def record_file_download_error(self, content_type: str, error_message: str):
+        """Record file download error metrics."""
+        self.file_downloads_total.labels(
+            content_type=content_type,
+            status="error"
+        ).inc()
+        
+        self.errors_total.labels(
+            error_type="file_download_error",
+            handler_type="media_processing",
+            method_name="download_file"
+        ).inc()
+    
+    def record_media_processing(self, content_type: str, duration: float, success: bool):
+        """Record media processing metrics."""
+        status = "success" if success else "error"
+        
+        self.media_processing_total.labels(
+            content_type=content_type,
+            status=status
+        ).inc()
+        
+        self.media_processing_duration_seconds.labels(
+            content_type=content_type
+        ).observe(duration)
+        
+        if not success:
+            self.errors_total.labels(
+                error_type="media_processing_error",
+                handler_type="media_processing",
+                method_name="add_in_db_media"
+            ).inc()
+    
     def get_metrics(self) -> bytes:
         """Generate metrics in Prometheus format."""
         return generate_latest(self.registry)