Update routes/dashboard_home.py

routes/dashboard_home.py

@@ -250,6 +250,82 @@ async def get_platform_users() -> Dict[str, Any]:
             "error": str(e)
         }
 
+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"""
+    try:
+        # Construir a query para obter assinaturas ativas
+        subscriptions_url = f"{SUPABASE_URL}/rest/v1/Subscriptions?active=eq.true&select=stylist_id"
+        
+        # Adicionar filtros de data se fornecidos
+        if start_timestamp or end_timestamp:
+            date_filters = []
+            if start_timestamp:
+                start_date = datetime.fromtimestamp(start_timestamp).strftime("%Y-%m-%d")
+                date_filters.append(f"created_at=gte.{start_date}")
+            if end_timestamp:
+                end_date = datetime.fromtimestamp(end_timestamp).strftime("%Y-%m-%d")
+                date_filters.append(f"created_at=lte.{end_date}")
+            
+            if date_filters:
+                date_query = "&".join(date_filters)
+                subscriptions_url = f"{subscriptions_url}&{date_query}"
+        
+        # Fazer a requisição para as assinaturas
+        async with aiohttp.ClientSession() as session:
+            async with session.get(subscriptions_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 = {}
+        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
+        
+        # Obter IDs dos top estilistas
+        top_stylist_ids = sorted(stylist_counts.keys(), key=lambda x: stylist_counts[x], 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])
+        stylists_url = f"{SUPABASE_URL}/rest/v1/User?id=in.({','.join(top_stylist_ids)})&select=id,name,avatar"
+        
+        # Obter detalhes dos estilistas
+        async with aiohttp.ClientSession() as session:
+            async with session.get(stylists_url, headers=SUPABASE_HEADERS) as response:
+                if response.status != 200:
+                    logger.error(f"❌ Erro ao obter detalhes dos estilistas: {response.status}")
+                    return []
+                
+                stylists_data = await response.json()
+        
+        # Combinar contagem com detalhes e ordenar por número de assinantes
+        result = []
+        for stylist in stylists_data:
+            stylist_id = stylist.get("id")
+            if stylist_id in stylist_counts:
+                result.append({
+                    "id": stylist_id,
+                    "name": stylist.get("name", "Nome não disponível"),
+                    "avatar": stylist.get("avatar", ""),
+                    "subscription_count": stylist_counts[stylist_id]
+                })
+        
+        # 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
+        
+    except Exception as e:
+        logger.error(f"❌ Erro ao obter top estilistas: {str(e)}")
+        return []
+
 async def get_monthly_revenue_data(target_date) -> Dict[str, Any]:
     """Obtém o faturamento de um mês específico de forma otimizada"""
     ny_timezone = pytz.timezone('America/New_York')
@@ -289,85 +365,6 @@ async def get_monthly_revenue_data(target_date) -> Dict[str, Any]:
         "transfer_count": transfer_data["transfer_count"]
     }
 
-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 que mais receberam pagamentos no período"""
-    try:
-        # Obter transferências no período
-        query_params = {"limit": 100}
-        
-        # Adicionar filtros de data se fornecidos
-        if start_timestamp and end_timestamp:
-            query_params["created"] = {"gte": start_timestamp, "lte": end_timestamp}
-        elif start_timestamp:
-            query_params["created"] = {"gte": start_timestamp}
-        elif end_timestamp:
-            query_params["created"] = {"lte": end_timestamp}
-        
-        transfers = []
-        has_more = True
-        last_id = None
-        
-        # Usar batch para reduzir chamadas
-        while has_more:
-            if last_id:
-                query_params["starting_after"] = last_id
-            
-            # Usar thread para não bloquear o evento loop
-            transfer_list = await fetch_stripe_data(stripe.Transfer.list, expand=["data.destination"], **query_params)
-            transfers.extend(transfer_list.data)
-            
-            has_more = transfer_list.has_more
-            if transfer_list.data:
-                last_id = transfer_list.data[-1].id
-            else:
-                has_more = False
-        
-        # Agrupar transferências por estilista (destination account)
-        stylist_totals = {}
-        for transfer in transfers:
-            destination = transfer.destination
-            if destination not in stylist_totals:
-                stylist_totals[destination] = {
-                    "account_id": destination,
-                    "total_received": 0
-                }
-            stylist_totals[destination]["total_received"] += transfer.amount
-        
-        # Ordenar estilistas por valor recebido
-        top_stylists_by_account = sorted(
-            stylist_totals.values(), 
-            key=lambda x: x["total_received"], 
-            reverse=True
-        )[:limit]
-        
-        # Buscar informações adicionais dos estilistas no Supabase
-        async with aiohttp.ClientSession() as session:
-            top_stylists = []
-            for stylist_data in top_stylists_by_account:
-                account_id = stylist_data["account_id"]
-                
-                # Buscar usuário correspondente pelo Stripe account_id
-                query_url = f"{SUPABASE_URL}/rest/v1/User?stripe_account_id=eq.{account_id}&select=id,name,avatar"
-                async with session.get(query_url, headers=SUPABASE_HEADERS) as response:
-                    if response.status == 200:
-                        user_data = await response.json()
-                        if user_data and len(user_data) > 0:
-                            # Combinar dados do Stripe com dados do Supabase
-                            stylist_info = {
-                                "id": user_data[0].get("id"),
-                                "name": user_data[0].get("name", "Estilista"),
-                                "avatar": user_data[0].get("avatar"),
-                                "total_received": stylist_data["total_received"],
-                                "total_transfers": stylist_totals[account_id].get("total_transfers", 1)
-                            }
-                            top_stylists.append(stylist_info)
-        
-        return top_stylists[:limit]
-        
-    except Exception as e:
-        logger.error(f"❌ Erro ao obter top estilistas: {str(e)}")
-        return []
-
 @router.get("/admin/dashboard")
 async def get_admin_dashboard(
     user_token: str = Header(None, alias="User-key"),