Update routes/subscription.py

routes/subscription.py

@@ -152,7 +152,7 @@ def create_checkout_session(
         # 🔹 1. Validar o token e obter user_id
         user_id = verify_token(user_token)
 
-        # 🔹 2. Buscar estilista no Supabase
+        # 🔹 2. Buscar estilista no Supabase usando o ID enviado na solicitação
         response = requests.get(
             f"{SUPABASE_URL}/rest/v1/User?id=eq.{data.id}",
             headers=SUPABASE_HEADERS
@@ -165,11 +165,12 @@ def create_checkout_session(
         stylist = stylist_data[0]
         stylist_stripe_id = stylist.get("stripe_id")
         consultations = stylist.get("consultations")
+        price_id = stylist.get("price_id")  # Buscar o price_id do estilista
 
-        if not consultations or not stylist_stripe_id:
-            raise HTTPException(status_code=400, detail="Stylist profile is incomplete")
+        if not consultations or not stylist_stripe_id or not price_id:
+            raise HTTPException(status_code=400, detail="Stylist profile is incomplete or missing price_id")
 
-        # 🔹 3. Buscar stripe_id e price_id do usuário autenticado
+        # 🔹 3. Buscar stripe_id do usuário autenticado (cliente)
         response_user = requests.get(
             f"{SUPABASE_URL}/rest/v1/User?id=eq.{user_id}",
             headers=SUPABASE_HEADERS
@@ -185,15 +186,12 @@ def create_checkout_session(
 
         user = user_data[0]
         user_stripe_id = user.get("stripe_id")
-        price_id = user.get("price_id")  
 
-        # Depuração: Verificando o price_id
-        logger.info(f"📌 Retrieved price_id: {price_id}")
+        # Depuração: Verificando o stripe_id do usuário
+        logger.info(f"📌 Retrieved user stripe_id: {user_stripe_id}")
 
         if not user_stripe_id:
             raise HTTPException(status_code=400, detail="User does not have a Stripe ID")
-        if not price_id:
-            raise HTTPException(status_code=400, detail="User does not have a valid price ID")
 
         # 🔹 4. Criar Checkout Session no Stripe
         session = stripe.checkout.Session.create(
@@ -204,7 +202,7 @@ def create_checkout_session(
             customer=user.get("stripe_customer_id"),  # Certifique-se de que isso retorna 'cus_...'
             line_items=[
                 {
-                    "price": price_id,  # ✅ Mantém o price_id, sem "description"
+                    "price": price_id,  # Agora estamos usando o price_id do estilista
                     "quantity": 1
                 }
             ],