Update routes/subscription.py

routes/subscription.py

@@ -12,111 +12,6 @@ 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)
@@ -209,46 +104,39 @@ async def stripe_webhook(request: Request):
     logger.info(f"🔹 Event received: {event['type']}")
 
     if event["type"] == "invoice.payment_succeeded":
-        invoice = event["data"]["object"]
-        subscription_id = invoice.get("subscription")  # Get subscription ID
+    invoice = event["data"]["object"]
+    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")
+    # Retrieve subscription 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
-        
-        logger.info(f"✅ Payment confirmed: ${amount / 100:.2f} USD")
-        logger.info(f"👤 User ID: {user_id}")
-        logger.info(f"✂️ Stylist ID: {stylist_id}")
-        logger.info(f"💰 Stylist will receive: ${stylist_share / 100:.2f} USD")
-        logger.info(f"🏢 Platform will receive: ${platform_share / 100:.2f} USD")
-        
-        return {
-            "status": "Payment processed successfully!",
-            "user_id": user_id,
-            "stylist_id": stylist_id,
-            "total_paid": amount / 100,  # Convert cents to dollars
-            "stylist_share": stylist_share / 100,
-            "platform_share": platform_share / 100,
-        }
-
-    return {"status": "Event received, no action needed."}
+    amount = invoice["amount_paid"]
+    stylist_share = int(amount * 0.8)  # 80% for stylist
+    platform_share = amount - stylist_share  # 20% for platform
 
-### **4️⃣ CANCEL SUBSCRIPTION** ###
-@router.post("/cancel_subscription")
-def cancel_subscription(data: CancelSubscriptionRequest):
+    # 🔹 Adicionar transferência para o estilista
     try:
-        subscription = stripe.Subscription.modify(
-            data.subscription_id,
-            cancel_at_period_end=True,  # Cancels at the end of the current cycle
+        transfer = stripe.Transfer.create(
+            amount=stylist_share,  # Valor em CENTAVOS
+            currency="usd",
+            destination=stylist_id,  # Conta Stripe do estilista (Connected Account ID)
+            description=f"Payment for stylist {stylist_id} - Subscription",
         )
-        return {"status": "Subscription will be canceled at period end"}
+        logger.info(f"💸 Transferência realizada: {stylist_share / 100:.2f} USD para {stylist_id}")
     except Exception as e:
-        raise HTTPException(status_code=500, detail=str(e))
+        logger.error(f"🚨 Erro ao transferir para o estilista {stylist_id}: {e}")
+
+    return {
+        "status": "Payment processed and transferred successfully!",
+        "user_id": user_id,
+        "stylist_id": stylist_id,
+        "total_paid": amount / 100,
+        "stylist_share": stylist_share / 100,
+        "platform_share": platform_share / 100,
+        "transfer_status": "Completed" if "transfer" in locals() else "Failed"
+    }
 
 ### **4️⃣ CANCEL SUBSCRIPTION** ###
 @router.post("/cancel_subscription")