Optimize content status count queries for projects

Refactored content statistics queries in dashboard routes to use a single optimized database query via db_service.get_content_counts_by_status. The new implementation reduces multiple queries to a single grouped query, improving performance when retrieving content status counts for projects.

Author: Bentlybro

app/routes/dashboard.py

@@ -425,13 +425,8 @@ async def project_content(project_id):
         page=page, per_page=25, error_out=False
     )
 
-    # Get total project statistics (not just current page)
-    total_stats = {
-        'total': Content.query.filter_by(project_id=project.id).count(),
-        'approved': Content.query.filter_by(project_id=project.id, status='approved').count(),
-        'rejected': Content.query.filter_by(project_id=project.id, status='rejected').count(),
-        'flagged': Content.query.filter_by(project_id=project.id, status='flagged').count(),
-    }
+    # Get total project statistics (not just current page) - optimized single query
+    total_stats = await db_service.get_content_counts_by_status(project.id)
 
     return render_template('dashboard/content.html',
                            project=project,
@@ -537,14 +532,12 @@ async def project_settings(project_id):
         flash('You do not have access to this project', 'error')
         return redirect(url_for('dashboard.projects'))
 
-    # Get comprehensive stats
-    total_content = Content.query.filter_by(project_id=project.id).count()
-    approved_content = Content.query.filter_by(
-        project_id=project.id, status='approved').count()
-    rejected_content = Content.query.filter_by(
-        project_id=project.id, status='rejected').count()
-    flagged_content = Content.query.filter_by(
-        project_id=project.id, status='flagged').count()
+    # Get comprehensive stats - optimized single query for content counts
+    content_counts = await db_service.get_content_counts_by_status(project.id)
+    total_content = content_counts.get('total', 0)
+    approved_content = content_counts.get('approved', 0)
+    rejected_content = content_counts.get('rejected', 0)
+    flagged_content = content_counts.get('flagged', 0)
     rules_count = ModerationRule.query.filter_by(project_id=project.id).count()
     api_keys_count = APIKey.query.filter_by(project_id=project.id).count()
 

app/services/database_service.py

@@ -856,27 +856,29 @@ def _get_user():
 
     # Content Query Operations
     async def get_content_counts_by_status(self, project_id: str) -> Dict[str, int]:
-        """Get content counts by moderation status"""
+        """Get content counts by moderation status - optimized single query"""
         def _get_counts():
-            total = Content.query.filter_by(project_id=project_id).count()
-            approved = Content.query.filter_by(
-                project_id=project_id, status='approved').count()
-            rejected = Content.query.filter_by(
-                project_id=project_id, status='rejected').count()
-            flagged = Content.query.filter_by(
-                project_id=project_id, status='flagged').count()
-            pending = Content.query.filter_by(
-                project_id=project_id, status='pending').count()
-            error = Content.query.filter_by(
-                project_id=project_id, status='error').count()
+            # Use GROUP BY to get all counts in a single query
+            status_counts = db.session.query(
+                Content.status,
+                func.count(Content.id).label('count')
+            ).filter(
+                Content.project_id == project_id
+            ).group_by(Content.status).all()
+
+            # Convert to dictionary
+            counts = {status: count for status, count in status_counts}
+
+            # Calculate total from the status counts
+            total = sum(counts.values())
 
             return {
                 'total': total,
-                'approved': approved,
-                'rejected': rejected,
-                'flagged': flagged,
-                'pending': pending,
-                'error': error
+                'approved': counts.get('approved', 0),
+                'rejected': counts.get('rejected', 0),
+                'flagged': counts.get('flagged', 0),
+                'pending': counts.get('pending', 0),
+                'error': counts.get('error', 0)
             }
 
         return await self._safe_execute(_get_counts) or {