Update routes/stylist.py

routes/stylist.py

@@ -29,17 +29,12 @@ logging.basicConfig(level=logging.INFO)
 logger = logging.getLogger(__name__)
 
 def verify_token(user_token: str) -> str:
-    """
-    Valida o token JWT no Supabase e retorna o user_id se for válido.
-    """
     headers = {
         "Authorization": f"Bearer {user_token}",
         "apikey": SUPABASE_KEY,
         "Content-Type": "application/json"
     }
-
     response = requests.get(f"{SUPABASE_URL}/auth/v1/user", headers=headers)
-
     if response.status_code == 200:
         user_data = response.json()
         user_id = user_data.get("id")
@@ -49,286 +44,95 @@ def verify_token(user_token: str) -> str:
     else:
         raise HTTPException(status_code=401, detail="Invalid or expired token")
 
+def get_total_followers(user_id: str) -> int:
+    response = requests.get(
+        f"{SUPABASE_URL}/rest/v1/followers?following_id=eq.{user_id}&select=id",
+        headers=SUPABASE_HEADERS
+    )
+    if response.status_code == 200:
+        return len(response.json())
+    return 0
+
+def get_total_subscribers(user_id: str) -> int:
+    response = requests.get(
+        f"{SUPABASE_URL}/rest/v1/Subscriptions?stylist_id=eq.{user_id}&active=eq.true&select=id",
+        headers=SUPABASE_HEADERS
+    )
+    if response.status_code == 200:
+        return len(response.json())
+    return 0
+
 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, 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')
     now_ny = datetime.now(ny_timezone)
     current_month = now_ny.month
     current_year = now_ny.year
-    
-    # Preparar dicionário com os últimos 6 meses (incluindo o atual)
     monthly_data = {}
-    
-    # Criar entradas para os últimos 6 meses, mesmo sem dados
     for i in range(6):
-        # Calcular o mês e ano para i meses atrás
         target_date = now_ny - relativedelta(months=i)
         month_num = target_date.month
-        month_name = target_date.strftime('%b')  # Formato abreviado do mês (Jan, Feb, etc)
+        month_name = target_date.strftime('%b')
         year = target_date.year
-        
-        # Chave composta para garantir unicidade (ano-mês)
         month_key = f"{year}-{month_num}"
-        
-        # Verificar se é o mês atual
         is_current = (month_num == current_month and year == current_year)
-        
-        monthly_data[month_key] = {
-            "month": month_num,
-            "name": month_name,
-            "current": is_current,
-            "amount": 0,
-            "growth_percentage": None  # Será calculado depois
-        }
-    
-    # Calcular o timestamp para 6 meses atrás
+        monthly_data[month_key] = {"month": month_num, "name": month_name, "current": is_current, "amount": 0}
     start_date = now_ny - relativedelta(months=6)
     start_timestamp = int(start_date.timestamp())
-    
-    # Buscar transferências para a conta
     try:
-        transfers = stripe.Transfer.list(
-            destination=account_id,
-            created={"gte": start_timestamp},
-            limit=100  # Limitando a 100 transferências mais recentes
-        )
-        
-        # Processar as transferências
+        transfers = stripe.Transfer.list(destination=account_id, created={"gte": start_timestamp}, limit=100)
         for transfer in transfers.data:
             transfer_date = datetime.fromtimestamp(transfer.created, ny_timezone)
             month_num = transfer_date.month
             year = transfer_date.year
             month_key = f"{year}-{month_num}"
-            
-            # Verificar se o mês está dentro dos últimos 6 meses
             if month_key in monthly_data:
-                # Adicionar o valor da transferência (em centavos)
                 monthly_data[month_key]["amount"] += transfer.amount
-        
-        # Converter para lista e ordenar por data (mais antigo primeiro)
         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"]
-            
-            # Lógica corrigida para calcular o crescimento
-            if previous_amount == 0 and current_amount > 0:
-                # Se o mês anterior foi zero e o atual tem valor, é crescimento de 100%
-                result[i]["growth_percentage"] = {
-                    "direction": "up",
-                    "value": 100.0
-                }
-            elif previous_amount == 0 and current_amount == 0:
-                # Se ambos são zero, não há mudança
-                result[i]["growth_percentage"] = None
-            elif previous_amount > 0 and current_amount == 0:
-                # Se o anterior tinha valor e o atual é zero, é queda de 100%
-                result[i]["growth_percentage"] = {
-                    "direction": "down",
-                    "value": 100.0
-                }
-            elif previous_amount > 0:
-                # Cálculo normal quando ambos têm valores
-                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:
         logger.error(f"❌ Error getting monthly revenue: {str(e)}")
-        # Retornar os meses mesmo sem transferências
         return list(monthly_data.values())
 
 def get_account_balance(account_id: str) -> Dict[str, Any]:
-    """
-    Busca o saldo disponível e pendente da conta Stripe.
-    """
     try:
         balance = stripe.Balance.retrieve(stripe_account=account_id)
-        
-        # Inicializar valores
         available_balance = 0
         pending_balance = 0
-        currency = "BRL"  # Default para BRL
-        
-        # Calcular saldo disponível
+        currency = "BRL"
         for balance_item in balance.available:
             if balance_item.currency.upper() == "BRL":
                 available_balance = balance_item.amount
-                currency = "BRL"
                 break
-                
-        # Calcular saldo pendente
         for balance_item in balance.pending:
             if balance_item.currency.upper() == "BRL":
                 pending_balance = balance_item.amount
                 break
-                
-        return {
-            "available_balance": available_balance,
-            "pending_balance": pending_balance,
-            "currency": currency
-        }
-        
+        return {"available_balance": available_balance, "pending_balance": pending_balance, "currency": currency}
     except Exception as e:
         logger.error(f"❌ Error getting account balance: {str(e)}")
-        return {
-            "available_balance": 0,
-            "pending_balance": 0,
-            "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: Status code {response.status_code}")
-            return 0
-        
-        # Tentar obter o formato de resposta do corpo - formato diferente do esperado
-        try:
-            # Se a resposta for um array com um objeto que tem o campo count
-            response_data = response.json()
-            if isinstance(response_data, list) and len(response_data) > 0 and "count" in response_data[0]:
-                return int(response_data[0]["count"])
-        except Exception as e:
-            logger.error(f"❌ Error parsing followers count: {str(e)}")
-        
-        # Tentar o método anterior como fallback
-        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: Status code {response.status_code}")
-            return 0
-        
-        # Tentar obter o formato de resposta do corpo - formato diferente do esperado
-        try:
-            # Se a resposta for um array com um objeto que tem o campo count
-            response_data = response.json()
-            if isinstance(response_data, list) and len(response_data) > 0 and "count" in response_data[0]:
-                return int(response_data[0]["count"])
-        except Exception as e:
-            logger.error(f"❌ Error parsing subscribers count: {str(e)}")
-        
-        # Tentar o método anterior como fallback
-        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
+        return {"available_balance": 0, "pending_balance": 0, "currency": "BRL"}
 
 @router.get("/dashboard")
 def get_dashboard(user_token: str = Header(None, alias="User-key")):
     try:
         if not user_token:
             raise HTTPException(status_code=401, detail="Missing User-key header")
-        
-        # Validar o token e obter user_id
         user_id = verify_token(user_token)
         logger.info(f"🔹 User verified. user_id: {user_id}")
-
-        # Buscar stripe_id do usuário no Supabase
         user_data_url = f"{SUPABASE_URL}/rest/v1/User?id=eq.{user_id}"
-        response = requests.get(
-            user_data_url, 
-            headers={
-                "Authorization": f"Bearer {user_token}",
-                "apikey": SUPABASE_KEY,
-                "Content-Type": "application/json"
-            }
-        )
-        
+        response = requests.get(user_data_url, headers={"Authorization": f"Bearer {user_token}", "apikey": SUPABASE_KEY, "Content-Type": "application/json"})
         if response.status_code != 200 or not response.json():
             raise HTTPException(status_code=404, detail="User not found")
-            
         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
+        total_followers = get_total_followers(user_id)
+        total_subscribers = get_total_subscribers(user_id)
         if not stripe_id:
-            return {
-                "stripe_id": None,
-                "available_balance": 0,
-                "pending_balance": 0,
-                "currency": "BRL",
-                "monthly_revenue": [],
-                "total_followers": total_followers,
-                "total_subscribers": total_subscribers
-            }
-        
-        # Buscar saldo da conta
+            return {"stripe_id": None, "available_balance": 0, "pending_balance": 0, "currency": "BRL", "monthly_revenue": [], "total_followers": total_followers, "total_subscribers": total_subscribers}
         balance_info = get_account_balance(stripe_id)
-        
-        # Buscar valores recebidos nos últimos 6 meses
         monthly_revenue = get_monthly_revenue(stripe_id)
-        
-        return {
-            "stripe_id": stripe_id,
-            "available_balance": balance_info["available_balance"],
-            "pending_balance": balance_info["pending_balance"],
-            "currency": balance_info["currency"],
-            "monthly_revenue": monthly_revenue,
-            "total_followers": total_followers,
-            "total_subscribers": total_subscribers
-        }
-
+        return {"stripe_id": stripe_id, "available_balance": balance_info["available_balance"], "pending_balance": balance_info["pending_balance"], "currency": balance_info["currency"], "monthly_revenue": monthly_revenue, "total_followers": total_followers, "total_subscribers": total_subscribers}
     except HTTPException as http_err:
         raise http_err
     except Exception as e: