Update routes/stylist.py

routes/stylist.py

@@ -96,6 +96,45 @@ def get_weekly_revenue(account_id: str) -> List[Dict[str, Any]]:
         logger.error(f"❌ Error getting weekly revenue: {str(e)}")
         return []
 
+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
+        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
+        }
+        
+    except Exception as e:
+        logger.error(f"❌ Error getting account balance: {str(e)}")
+        return {
+            "available_balance": 0,
+            "pending_balance": 0,
+            "currency": "BRL"
+        }
+
 @router.get("/dashboard")
 def get_dashboard(user_token: str = Header(None, alias="User-key")):
     try:
@@ -126,14 +165,23 @@ def get_dashboard(user_token: str = Header(None, alias="User-key")):
         if not stripe_id:
             return {
                 "stripe_id": None,
+                "available_balance": 0,
+                "pending_balance": 0,
+                "currency": "BRL",
                 "weekly_revenue": []
             }
         
+        # Buscar saldo da conta
+        balance_info = get_account_balance(stripe_id)
+        
         # Buscar valores recebidos nas últimas 6 semanas
         weekly_revenue = get_weekly_revenue(stripe_id)
         
         return {
             "stripe_id": stripe_id,
+            "available_balance": balance_info["available_balance"],
+            "pending_balance": balance_info["pending_balance"],
+            "currency": balance_info["currency"],
             "weekly_revenue": weekly_revenue
         }