Update routes/subscription.py

routes/subscription.py

@@ -71,10 +71,12 @@ def create_checkout_session(data: SubscriptionRequest):
                     "quantity": 1
                 }
             ],
-            metadata={
-                "user_id": data.user_id,
-                "stylist_id": data.stylist_id,
-            },
+            subscription_data={
+                "metadata": {
+                    "user_id": data.user_id,
+                    "stylist_id": data.stylist_id,
+                }
+            }
         )
 
         return {
@@ -93,9 +95,7 @@ async def stripe_webhook(request: Request):
     event = None
     
     try:
-        event = stripe.Event.construct_from(
-            json.loads(payload), stripe.api_key
-        )
+        event = stripe.Event.construct_from(json.loads(payload), stripe.api_key)
     except Exception as e:
         logger.error(f"⚠️ Error processing webhook: {e}")
         raise HTTPException(status_code=400, detail="Webhook error")
@@ -104,10 +104,14 @@ async def stripe_webhook(request: Request):
 
     if event["type"] == "invoice.payment_succeeded":
         invoice = event["data"]["object"]
-        user_id = invoice.get("metadata", {}).get("user_id")
-        stylist_id = invoice.get("metadata", {}).get("stylist_id")
+        subscription_id = invoice.get("subscription")  # Get subscription ID
+
+        # Retrieve subscription to get metadata
+        subscription = stripe.Subscription.retrieve(subscription_id)
+        user_id = subscription.metadata.get("user_id")
+        stylist_id = subscription.metadata.get("stylist_id")
+
         amount = invoice["amount_paid"]
-        
         stylist_share = int(amount * 0.8)  # 80% for the stylist
         platform_share = amount - stylist_share  # 20% for the platform