Update routes/subscription.py

routes/subscription.py

@@ -73,15 +73,26 @@ async def generate_dashboard_link(data: UserIDRequest):
     try:
         user_id = data.user_id  # O ID do estilista ou cliente
 
+        # Verifica se é um estilista (conta conectada)
         if user_id.startswith("acct_"):
-            # O usuário é um estilista (Stripe Connect)
-            dashboard_url = f"https://dashboard.stripe.com/connect/accounts/{user_id}/overview"
-            return {"status": "success", "dashboard_link": dashboard_url}
-
+            try:
+                # Cria o login link para o painel do estilista (Stripe Express)
+                login_link = stripe.Account.create_login_link(user_id)
+                
+                return {
+                    "status": "success",
+                    "dashboard_link": login_link.url  # Retorna o link gerado para o painel do estilista
+                }
+            except stripe.error.StripeError as e:
+                logger.error(f"Error creating login link for stylist account: {str(e)}")
+                raise HTTPException(status_code=500, detail="Error creating login link for stylist account.")
+        
+        # Caso seja um cliente, não podemos criar um login link
         elif user_id.startswith("cus_"):
-            # O usuário é um cliente
-            dashboard_url = f"https://billing.stripe.com/p/{user_id}"
-            return {"status": "success", "dashboard_link": dashboard_url}
+            return {
+                "status": "success",
+                "dashboard_link": f"https://billing.stripe.com/p/{user_id}"
+            }
 
         else:
             raise HTTPException(status_code=400, detail="Invalid user ID format. Must start with 'acct_' for stylist or 'cus_' for customer.")