habulaj commited on
Commit
a51c943
·
verified ·
1 Parent(s): 8f8ef0c

Update routes/stylist.py

Browse files
Files changed (1) hide show
  1. routes/stylist.py +11 -1
routes/stylist.py CHANGED
@@ -335,9 +335,19 @@ def get_dashboard(user_token: str = Header(None, alias="User-key"), page: int =
335
  user_id = verify_token(user_token)
336
  user_data_url = f"{SUPABASE_URL}/rest/v1/User?id=eq.{user_id}"
337
  response = requests.get(user_data_url, headers=SUPABASE_HEADERS)
338
- user_data = response.json()[0]
 
 
 
 
 
 
339
  stripe_id = user_data.get("stripe_id")
340
 
 
 
 
 
341
  # Obter dados de receita mensal e total dos últimos 6 meses
342
  revenue_data = get_monthly_revenue(stripe_id)
343
 
 
335
  user_id = verify_token(user_token)
336
  user_data_url = f"{SUPABASE_URL}/rest/v1/User?id=eq.{user_id}"
337
  response = requests.get(user_data_url, headers=SUPABASE_HEADERS)
338
+
339
+ # Verificar se a resposta contém dados
340
+ user_data_response = response.json()
341
+ if not user_data_response:
342
+ raise HTTPException(status_code=404, detail="User not found")
343
+
344
+ user_data = user_data_response[0]
345
  stripe_id = user_data.get("stripe_id")
346
 
347
+ # Verificar se o usuário tem um stripe_id
348
+ if not stripe_id:
349
+ raise HTTPException(status_code=400, detail="User does not have a Stripe account")
350
+
351
  # Obter dados de receita mensal e total dos últimos 6 meses
352
  revenue_data = get_monthly_revenue(stripe_id)
353