Update routes/subscription.py

routes/subscription.py

@@ -12,6 +12,111 @@ logger = logging.getLogger(__name__)
 stripe.api_key = "sk_test_51QtyIUPwiMKKxiyaqRrDh4e3JdpE2Ket5p7idABq7dBYhpImMftf4N1IXTWB9yPxoTfb9CxhoBX7a96Sds7CQ9As00E2yf3dyW"
 stripe.api_version = "2023-10-16"
 
+### **DATA MODELS** ###
+class PriceRequest(BaseModel):
+    amount: int  # Value in cents (e.g., $25.00 -> 2500)
+    stylist_name: str  # Name of the stylist
+
+class SubscriptionRequest(BaseModel):
+    user_id: str
+    stylist_id: str
+    price_id: str  # Now using a pre-created price
+
+class CancelSubscriptionRequest(BaseModel):
+    subscription_id: str
+
+### **1️⃣ CREATE DYNAMIC PRICE** ###
+@router.post("/create_price")
+def create_price(data: PriceRequest):
+    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="usd",
+            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 dollars
+            "currency": "USD",
+        }
+    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",
+            payment_method_types=["card"],
+            mode="subscription",
+            line_items=[
+                {
+                    "price": data.price_id,
+                    "quantity": 1
+                }
+            ],
+            subscription_data={
+                "metadata": {
+                    "user_id": data.user_id,
+                    "stylist_id": data.stylist_id,
+                }
+            }
+        )
+
+        return {
+            "message": "Checkout session created successfully!",
+            "checkout_url": session.url
+        }
+    except Exception as e:
+        logger.error(f"Error creating checkout session: {e}")
+        raise HTTPException(status_code=500, detail="Error creating checkout session.")
+
+### **3️⃣ WEBHOOK TO PROCESS PAYMENTS** ###
+@router.post("/webhook")
+async def stripe_webhook(request: Request):
+    payload = await request.body()
+    headers = dict(request.headers)
+    event = None
+    
+    try:
+        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")
+
+    logger.info(f"🔹 Event received: {event['type']}")
+
+    import stripe
+from fastapi import APIRouter, HTTPException, Request
+from pydantic import BaseModel
+import json
+import logging
+
+router = APIRouter()
+
+logging.basicConfig(level=logging.INFO, format="%(asctime)s - %(levelname)s - %(message)s")
+logger = logging.getLogger(__name__)
+
+stripe.api_key = "sk_test_51QtyIUPwiMKKxiyaqRrDh4e3JdpE2Ket5p7idABq7dBYhpImMftf4N1IXTWB9yPxoTfb9CxhoBX7a96Sds7CQ9As00E2yf3dyW"
+stripe.api_version = "2023-10-16"
+
 ### **DATA MODELS** ###
 class PriceRequest(BaseModel):
     amount: int  # Value in cents (e.g., $25.00 -> 2500)
@@ -133,6 +238,18 @@ async def stripe_webhook(request: Request):
 
     return {"status": "Event received, no action needed."}
 
+### **4️⃣ 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
+        )
+        return {"status": "Subscription will be canceled at period end"}
+    except Exception as e:
+        raise HTTPException(status_code=500, detail=str(e))
+
 ### **4️⃣ CANCEL SUBSCRIPTION** ###
 @router.post("/cancel_subscription")
 def cancel_subscription(data: CancelSubscriptionRequest):