Update routes/stylist.py

routes/stylist.py

@@ -52,7 +52,8 @@ def verify_token(user_token: str) -> str:
 def get_monthly_revenue(account_id: str) -> List[Dict[str, Any]]:
     """
     Busca os valores recebidos nos últimos 6 meses para o account_id.
-    Retorna uma lista com número do mês, nome do mês, indicador se é o mês atual e valor.
+    Retorna uma lista com número do mês, nome do mês, indicador se é o mês atual, valor
+    e porcentagem de crescimento/queda em relação ao mês anterior.
     """
     # Obter data atual no fuso horário de Nova York
     ny_timezone = pytz.timezone('America/New_York')
@@ -81,7 +82,8 @@ def get_monthly_revenue(account_id: str) -> List[Dict[str, Any]]:
             "month": month_num,
             "name": month_name,
             "current": is_current,
-            "amount": 0
+            "amount": 0,
+            "growth_percentage": None  # Será calculado depois
         }
     
     # Calcular o timestamp para 6 meses atrás
@@ -112,6 +114,20 @@ def get_monthly_revenue(account_id: str) -> List[Dict[str, Any]]:
         result = list(monthly_data.values())
         result.sort(key=lambda x: (current_month - x["month"]) % 12)
         
+        # Calcular a porcentagem de crescimento/queda para cada mês
+        for i in range(1, len(result)):
+            current_amount = result[i]["amount"]
+            previous_amount = result[i-1]["amount"]
+            
+            if previous_amount > 0:
+                # Calcular a porcentagem de crescimento e arredondar para 1 casa decimal
+                growth = ((current_amount - previous_amount) / previous_amount) * 100
+                direction = "up" if growth >= 0 else "down"
+                result[i]["growth_percentage"] = {
+                    "direction": direction,
+                    "value": round(abs(growth), 1)
+                }
+        
         return result
         
     except Exception as e:
@@ -158,6 +174,66 @@ def get_account_balance(account_id: str) -> Dict[str, Any]:
             "currency": "BRL"
         }
 
+def get_total_followers(user_id: str, user_token: str) -> int:
+    """
+    Busca o total de seguidores do usuário na tabela 'followers'.
+    """
+    try:
+        # Consulta para contar os seguidores onde following_id é o user_id
+        followers_url = f"{SUPABASE_URL}/rest/v1/followers?select=count&following_id=eq.{user_id}"
+        
+        response = requests.get(
+            followers_url,
+            headers={
+                "Authorization": f"Bearer {user_token}",
+                "apikey": SUPABASE_KEY,
+                "Content-Type": "application/json",
+                "Prefer": "count=exact"
+            }
+        )
+        
+        if response.status_code != 200:
+            logger.error(f"❌ Error getting followers count: {response.text}")
+            return 0
+            
+        # Extrair contagem do cabeçalho de resposta
+        count = int(response.headers.get("Content-Range", "0-0/0").split("/")[-1])
+        return count
+        
+    except Exception as e:
+        logger.error(f"❌ Error getting total followers: {str(e)}")
+        return 0
+
+def get_total_subscribers(user_id: str, user_token: str) -> int:
+    """
+    Busca o total de inscritos ativos do usuário na tabela 'Subscriptions'.
+    """
+    try:
+        # Consulta para contar as inscrições ativas onde stylist_id é o user_id
+        subscribers_url = f"{SUPABASE_URL}/rest/v1/Subscriptions?select=count&stylist_id=eq.{user_id}&active=eq.true"
+        
+        response = requests.get(
+            subscribers_url,
+            headers={
+                "Authorization": f"Bearer {user_token}",
+                "apikey": SUPABASE_KEY,
+                "Content-Type": "application/json",
+                "Prefer": "count=exact"
+            }
+        )
+        
+        if response.status_code != 200:
+            logger.error(f"❌ Error getting subscribers count: {response.text}")
+            return 0
+            
+        # Extrair contagem do cabeçalho de resposta
+        count = int(response.headers.get("Content-Range", "0-0/0").split("/")[-1])
+        return count
+        
+    except Exception as e:
+        logger.error(f"❌ Error getting total subscribers: {str(e)}")
+        return 0
+
 @router.get("/dashboard")
 def get_dashboard(user_token: str = Header(None, alias="User-key")):
     try:
@@ -185,13 +261,22 @@ def get_dashboard(user_token: str = Header(None, alias="User-key")):
         user_data = response.json()[0]
         stripe_id = user_data.get("stripe_id")
         
+        # Buscar total de seguidores
+        total_followers = get_total_followers(user_id, user_token)
+        
+        # Buscar total de inscritos ativos
+        total_subscribers = get_total_subscribers(user_id, user_token)
+        
+        # Resposta padrão se não houver stripe_id
         if not stripe_id:
             return {
                 "stripe_id": None,
                 "available_balance": 0,
                 "pending_balance": 0,
                 "currency": "BRL",
-                "monthly_revenue": []
+                "monthly_revenue": [],
+                "total_followers": total_followers,
+                "total_subscribers": total_subscribers
             }
         
         # Buscar saldo da conta
@@ -205,7 +290,9 @@ def get_dashboard(user_token: str = Header(None, alias="User-key")):
             "available_balance": balance_info["available_balance"],
             "pending_balance": balance_info["pending_balance"],
             "currency": balance_info["currency"],
-            "monthly_revenue": monthly_revenue
+            "monthly_revenue": monthly_revenue,
+            "total_followers": total_followers,
+            "total_subscribers": total_subscribers
         }
 
     except HTTPException as http_err: