Update routes/subscription.py

routes/subscription.py

@@ -13,54 +13,22 @@ stripe.api_key = "sk_test_51N6K5JB9VMe0qzbOjlJvMEsfdQyrFgV49vRaeErtmhrzHV3Cu3f5j
 stripe.api_version = "2023-10-16"
 
 ### **DATA MODELS** ###
-class PriceRequest(BaseModel):
-    amount: int  # Value in cents (e.g., R$25.00 -> 2500)
-    stylist_name: str  # Name of the stylist
-
 class SubscriptionRequest(BaseModel):
     user_id: str
     stylist_id: str  # This must be the Stripe connected account ID (acct_xxxxx)
-    price_id: str  # Now using a pre-created price
+    amount: int  # Value in cents (e.g., R$25.00 -> 2500)
+    stylist_name: str  # Name of the stylist
 
 class CancelSubscriptionRequest(BaseModel):
     subscription_id: str
 
-### **1️⃣ CREATE DYNAMIC PRICE** ###
-@router.post("/create_price")
-def create_price(data: PriceRequest):
+### **CREATE CHECKOUT SESSION** ###
+@router.post("/create_checkout_session")
+def create_checkout_session(data: SubscriptionRequest):
     try:
         product_name = f"{data.stylist_name}'s Subscription"
         product_description = f"Monthly plan to access exclusive services from {data.stylist_name}."
 
-        product = stripe.Product.create(
-            name=product_name,
-            description=product_description,
-        )
-
-        price = stripe.Price.create(
-            unit_amount=data.amount,
-            currency="brl",  # Changed to BRL
-            recurring={"interval": "month"},
-            product=product.id,
-        )
-
-        return {
-            "message": "Price created successfully!",
-            "price_id": price.id,
-            "product_id": product.id,
-            "product_name": product_name,
-            "product_description": product_description,
-            "amount": data.amount / 100,  # Converting cents to BRL
-            "currency": "BRL",
-        }
-    except Exception as e:
-        logger.error(f"Error creating price: {e}")
-        raise HTTPException(status_code=500, detail="Error creating price.")
-
-### **2️⃣ CREATE CHECKOUT SESSION** ###
-@router.post("/create_checkout_session")
-def create_checkout_session(data: SubscriptionRequest):
-    try:
         session = stripe.checkout.Session.create(
             success_url="https://yourdomain.com/success",
             cancel_url="https://yourdomain.com/cancel",
@@ -68,7 +36,15 @@ def create_checkout_session(data: SubscriptionRequest):
             mode="subscription",
             line_items=[
                 {
-                    "price": data.price_id,
+                    "price_data": {
+                        "currency": "brl",
+                        "product_data": {
+                            "name": product_name,
+                            "description": product_description
+                        },
+                        "unit_amount": data.amount,
+                        "recurring": {"interval": "month"}
+                    },
                     "quantity": 1
                 }
             ],
@@ -88,7 +64,7 @@ def create_checkout_session(data: SubscriptionRequest):
         logger.error(f"Error creating checkout session: {e}")
         raise HTTPException(status_code=500, detail="Error creating checkout session.")
 
-### **3️⃣ WEBHOOK TO PROCESS PAYMENTS & TRANSFER FUNDS** ###
+### **WEBHOOK TO PROCESS PAYMENTS & TRANSFER FUNDS** ###
 @router.post("/webhook")
 async def stripe_webhook(request: Request):
     payload = await request.body()
@@ -105,12 +81,11 @@ async def stripe_webhook(request: Request):
 
     if event["type"] == "invoice.payment_succeeded":
         invoice = event["data"]["object"]
-        subscription_id = invoice.get("subscription")  # Get subscription ID
+        subscription_id = invoice.get("subscription")
 
-        # 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")  # Must be a Stripe connected account ID
+        stylist_id = subscription.metadata.get("stylist_id")
 
         amount = invoice["amount_paid"]
         stylist_share = int(amount * 0.8)  # 80% for stylist
@@ -119,9 +94,9 @@ async def stripe_webhook(request: Request):
         # 🔹 Transfer the funds to the stylist
         try:
             transfer = stripe.Transfer.create(
-                amount=stylist_share,  # Value in CENTS
-                currency="brl",  # Changed to BRL
-                destination=stylist_id,  # The Stripe Connected Account ID (acct_xxxx)
+                amount=stylist_share,
+                currency="brl",
+                destination=stylist_id,
                 description=f"Payment for stylist {stylist_id} - Subscription",
             )
             logger.info(f"💸 Transfer successful: R${stylist_share / 100:.2f} BRL to {stylist_id}")
@@ -134,7 +109,7 @@ async def stripe_webhook(request: Request):
             "status": "Payment processed successfully!",
             "user_id": user_id,
             "stylist_id": stylist_id,
-            "total_paid": amount / 100,  # Convert cents to BRL
+            "total_paid": amount / 100,
             "stylist_share": stylist_share / 100,
             "platform_share": platform_share / 100,
             "transfer_status": transfer_status
@@ -142,13 +117,13 @@ async def stripe_webhook(request: Request):
 
     return {"status": "Event received, no action needed."}
 
-### **4️⃣ CANCEL SUBSCRIPTION** ###
+### **CANCEL SUBSCRIPTION** ###
 @router.post("/cancel_subscription")
 def cancel_subscription(data: CancelSubscriptionRequest):
     try:
         subscription = stripe.Subscription.modify(
             data.subscription_id,
-            cancel_at_period_end=True,  # Cancels at the end of the current cycle
+            cancel_at_period_end=True,
         )
         return {"status": "Subscription will be canceled at period end"}
     except Exception as e: