Update routes/dashboard_home.py

routes/dashboard_home.py

@@ -251,10 +251,10 @@ async def get_platform_users() -> Dict[str, Any]:
         }
 
 async def get_top_stylists(start_timestamp: Optional[int] = None, end_timestamp: Optional[int] = None, limit: int = 10) -> List[Dict[str, Any]]:
-    """Obtém os estilistas com mais assinantes ativos de forma otimizada"""
+    """Obtém os estilistas com mais assinantes ativos e totais de forma otimizada"""
     try:
-        # Construir a query para obter assinaturas ativas
-        subscriptions_url = f"{SUPABASE_URL}/rest/v1/Subscriptions?active=eq.true&select=stylist_id"
+        # Obter todas as assinaturas
+        base_url = f"{SUPABASE_URL}/rest/v1/Subscriptions?select=stylist_id,active,created_at"
         
         # Adicionar filtros de data se fornecidos
         if start_timestamp or end_timestamp:
@@ -268,33 +268,50 @@ async def get_top_stylists(start_timestamp: Optional[int] = None, end_timestamp:
             
             if date_filters:
                 date_query = "&".join(date_filters)
-                subscriptions_url = f"{subscriptions_url}&{date_query}"
+                base_url = f"{base_url}&{date_query}"
         
-        # Fazer a requisição para as assinaturas
+        # Fazer a requisição para obter todas as assinaturas
         async with aiohttp.ClientSession() as session:
-            async with session.get(subscriptions_url, headers=SUPABASE_HEADERS) as response:
+            async with session.get(base_url, headers=SUPABASE_HEADERS) as response:
                 if response.status != 200:
                     logger.error(f"❌ Erro ao obter assinaturas: {response.status}")
                     return []
                 
                 subscriptions_data = await response.json()
         
-        # Contar assinaturas por estilista
-        stylist_counts = {}
+        # Contadores para assinaturas ativas e totais por estilista
+        stylists_stats = {}
         for subscription in subscriptions_data:
             stylist_id = subscription.get("stylist_id")
-            if stylist_id:
-                stylist_counts[stylist_id] = stylist_counts.get(stylist_id, 0) + 1
+            is_active = subscription.get("active", False)
+            
+            if stylist_id not in stylists_stats:
+                stylists_stats[stylist_id] = {
+                    "active_count": 0, 
+                    "total_count": 0
+                }
+            
+            # Incrementar contadores
+            stylists_stats[stylist_id]["total_count"] += 1
+            if is_active:
+                stylists_stats[stylist_id]["active_count"] += 1
         
-        # Obter IDs dos top estilistas
-        top_stylist_ids = sorted(stylist_counts.keys(), key=lambda x: stylist_counts[x], reverse=True)[:limit]
+        # Se não houver estilistas, retornar lista vazia
+        if not stylists_stats:
+            return []
+        
+        # Ordenar estilistas por número de assinaturas ativas (decrescente)
+        top_stylist_ids = sorted(
+            stylists_stats.keys(), 
+            key=lambda x: stylists_stats[x]["active_count"], 
+            reverse=True
+        )[:limit]
         
         # Sem estilistas encontrados
         if not top_stylist_ids:
             return []
         
-        # Construir a query para obter detalhes dos estilistas
-        stylists_ids_str = ",".join([f"eq.{id}" for id in top_stylist_ids])
+        # Obter detalhes dos estilistas em uma única chamada
         stylists_url = f"{SUPABASE_URL}/rest/v1/User?id=in.({','.join(top_stylist_ids)})&select=id,name,avatar"
         
         # Obter detalhes dos estilistas
@@ -306,21 +323,22 @@ async def get_top_stylists(start_timestamp: Optional[int] = None, end_timestamp:
                 
                 stylists_data = await response.json()
         
-        # Combinar contagem com detalhes e ordenar por número de assinantes
+        # Mapear os IDs para facilitar o acesso
+        stylists_map = {stylist["id"]: stylist for stylist in stylists_data}
+        
+        # Combinar estatísticas com detalhes dos estilistas
         result = []
-        for stylist in stylists_data:
-            stylist_id = stylist.get("id")
-            if stylist_id in stylist_counts:
+        for stylist_id in top_stylist_ids:
+            if stylist_id in stylists_map:
                 result.append({
                     "id": stylist_id,
-                    "name": stylist.get("name", "Nome não disponível"),
-                    "avatar": stylist.get("avatar", ""),
-                    "subscription_count": stylist_counts[stylist_id]
+                    "name": stylists_map[stylist_id].get("name", "Nome não disponível"),
+                    "avatar": stylists_map[stylist_id].get("avatar", ""),
+                    "active_subscriptions": stylists_stats[stylist_id]["active_count"],
+                    "total_subscriptions": stylists_stats[stylist_id]["total_count"]
                 })
         
-        # Ordenar por número de assinantes (decrescente)
-        result.sort(key=lambda x: x["subscription_count"], reverse=True)
-        return result[:limit]  # Limitar ao número solicitado
+        return result
         
     except Exception as e:
         logger.error(f"❌ Erro ao obter top estilistas: {str(e)}")