Update routes/subscription.py

routes/subscription.py

@@ -57,8 +57,23 @@ def create_checkout_session(data: SubscriptionRequest):
         if not stylist_price or not consultations or not stylist_stripe_id:
             raise HTTPException(status_code=400, detail="Stylist profile is incomplete")
 
-        # 🔹 3. Criar Checkout Session no Stripe
+        # 🔹 3. Buscar o customer_id do usuário no banco de dados (Supabase)
+        user_response = requests.get(
+            f"{SUPABASE_URL}/rest/v1/User?id=eq.{data.user_id}",
+            headers=SUPABASE_HEADERS
+        )
+        
+        user_data = user_response.json()
+        if not user_data:
+            raise HTTPException(status_code=404, detail="User not found")
+        
+        customer_id = user_data[0].get("customer_id")
+        if not customer_id:
+            raise HTTPException(status_code=400, detail="Customer ID not found for user")
+
+        # 🔹 4. Criar Checkout Session no Stripe com o customer_id
         session = stripe.checkout.Session.create(
+            customer=customer_id,  # Passando o customer_id recuperado
             success_url="https://yourdomain.com/success",
             cancel_url="https://yourdomain.com/cancel",
             payment_method_types=["card"],
@@ -98,7 +113,7 @@ def create_checkout_session(data: SubscriptionRequest):
     except Exception as e:
         logger.error(f"Error creating checkout session: {e}")
         raise HTTPException(status_code=500, detail="Error creating checkout session.")
-
+        
 ### **WEBHOOK PARA PROCESSAR PAGAMENTOS**
 @router.post("/webhook")
 async def stripe_webhook(request: Request):