Update routes/stylist.py

routes/stylist.py

@@ -3,6 +3,8 @@ import stripe
 import requests
 import logging
 from fastapi import APIRouter, HTTPException, Header
+from datetime import datetime, timedelta
+from typing import List, Dict, Any
 
 router = APIRouter()
 
@@ -45,6 +47,55 @@ def verify_token(user_token: str) -> str:
     else:
         raise HTTPException(status_code=401, detail="Invalid or expired token")
 
+def get_weekly_revenue(account_id: str) -> List[Dict[str, Any]]:
+    """
+    Busca os valores recebidos nas últimas 6 semanas para o account_id.
+    Retorna uma lista com número da semana, data e valor.
+    """
+    # Calcular o timestamp para 6 semanas atrás
+    end_date = datetime.now()
+    start_date = end_date - timedelta(weeks=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
+        )
+        
+        # Preparar dicionário para agrupar por semana
+        weekly_data = {}
+        
+        # Processar as transferências
+        for transfer in transfers.data:
+            transfer_date = datetime.fromtimestamp(transfer.created)
+            # Número da semana no ano (1-53)
+            week_number = transfer_date.isocalendar()[1]
+            week_start = transfer_date - timedelta(days=transfer_date.weekday())
+            week_start_str = week_start.strftime("%Y-%m-%d")
+            
+            if week_number not in weekly_data:
+                weekly_data[week_number] = {
+                    "week": week_number,
+                    "date": week_start_str,
+                    "amount": 0
+                }
+            
+            # Adicionar o valor da transferência (em centavos)
+            weekly_data[week_number]["amount"] += transfer.amount
+        
+        # Converter para lista e ordenar por semana
+        result = list(weekly_data.values())
+        result.sort(key=lambda x: x["week"])
+        
+        return result
+        
+    except Exception as e:
+        logger.error(f"❌ Error getting weekly revenue: {str(e)}")
+        return []
+
 @router.get("/dashboard")
 def get_dashboard(user_token: str = Header(None, alias="User-key")):
     try:
@@ -72,38 +123,19 @@ def get_dashboard(user_token: str = Header(None, alias="User-key")):
         user_data = response.json()[0]
         stripe_id = user_data.get("stripe_id")
         
-        # Obter saldo da conta connect do Stripe
-        try:
-            balance = stripe.Balance.retrieve(stripe_account=stripe_id)
-            
-            # Processar o saldo disponível
-            available_balance = 0
-            pending_balance = 0
-            
-            # Soma todos os valores disponíveis em BRL
-            for available in balance.available:
-                if available.currency.lower() == 'brl':
-                    available_balance += available.amount
-            
-            # Soma todos os valores pendentes em BRL
-            for pending in balance.pending:
-                if pending.currency.lower() == 'brl':
-                    pending_balance += pending.amount
-            
-            # Converter de centavos para reais
-            available_balance = available_balance / 100
-            pending_balance = pending_balance / 100
-            
+        if not stripe_id:
             return {
-                "stripe_id": stripe_id,
-                "available_balance": available_balance,
-                "pending_balance": pending_balance,
-                "currency": "BRL"
+                "stripe_id": None,
+                "weekly_revenue": []
             }
         
-        except stripe.error.StripeError as e:
-            logger.error(f"❌ Stripe error: {str(e)}")
-            raise HTTPException(status_code=500, detail=f"Erro ao obter saldo: {str(e)}")
+        # Buscar valores recebidos nas últimas 6 semanas
+        weekly_revenue = get_weekly_revenue(stripe_id)
+        
+        return {
+            "stripe_id": stripe_id,
+            "weekly_revenue": weekly_revenue
+        }
 
     except HTTPException as http_err:
         raise http_err