Update routes/stylist.py

routes/stylist.py

@@ -58,6 +58,55 @@ def get_account_balance(account_id: str) -> Dict[str, Any]:
         logger.error(f"❌ Error getting account balance: {str(e)}")
         return {"available_balance": 0, "pending_balance": 0, "currency": "BRL"}
 
+def get_connected_account_details(account_id: str) -> Dict[str, Any]:
+    try:
+        account = stripe.Account.retrieve(account_id)
+        external_accounts = stripe.Account.list_external_accounts(
+            account_id,
+            object="bank_account",
+            limit=1
+        )
+        
+        payout_schedule = account.settings.payouts.schedule
+        
+        bank_account = {}
+        if external_accounts and external_accounts.data:
+            bank = external_accounts.data[0]
+            bank_account = {
+                "account_holder_name": bank.account_holder_name,
+                "bank_name": bank.bank_name,
+                "last4": bank.last4,
+                "routing_number": bank.routing_number,
+                "status": bank.status
+            }
+        
+        return {
+            "id": account.id,
+            "business_type": account.business_type,
+            "company": {
+                "name": account.business_profile.name if hasattr(account, 'business_profile') and hasattr(account.business_profile, 'name') else None,
+            },
+            "email": account.email,
+            "payouts_enabled": account.payouts_enabled,
+            "charges_enabled": account.charges_enabled,
+            "payout_schedule": {
+                "interval": payout_schedule.interval,
+                "monthly_anchor": payout_schedule.monthly_anchor if hasattr(payout_schedule, 'monthly_anchor') else None,
+                "weekly_anchor": payout_schedule.weekly_anchor if hasattr(payout_schedule, 'weekly_anchor') else None,
+                "delay_days": payout_schedule.delay_days
+            },
+            "bank_account": bank_account,
+            "verification_status": account.requirements.currently_due or [] 
+        }
+    except Exception as e:
+        logger.error(f"❌ Error getting connected account details: {str(e)}")
+        return {
+            "id": account_id,
+            "payouts_enabled": False,
+            "charges_enabled": False,
+            "error": str(e)
+        }
+
 def get_monthly_revenue(account_id: str) -> Dict[str, Any]:
     ny_timezone = pytz.timezone('America/New_York')
     now_ny = datetime.now(ny_timezone)
@@ -203,6 +252,9 @@ def get_dashboard(user_token: str = Header(None, alias="User-key"), page: int =
         # Obter dados de receita mensal e total dos últimos 6 meses
         revenue_data = get_monthly_revenue(stripe_id)
         
+        # Obter detalhes da conta conectada
+        account_details = get_connected_account_details(stripe_id)
+        
         return {
             "stripe_id": stripe_id,
             "available_balance": get_account_balance(stripe_id),
@@ -210,6 +262,7 @@ def get_dashboard(user_token: str = Header(None, alias="User-key"), page: int =
             "total_revenue_last_6_months": revenue_data["total_last_6_months"],
             "total_followers": get_total_followers(user_id),
             "total_subscribers": get_total_subscribers(user_id),
+            "account_details": account_details,
             **get_active_subscribers(user_id, page)
         }
     except Exception as e: