Update routes/subscription.py

routes/subscription.py

@@ -8,23 +8,24 @@ router = APIRouter()
 
 logging.basicConfig(level=logging.INFO, format="%(asctime)s - %(levelname)s - %(message)s")
 logger = logging.getLogger(__name__)
+
 stripe.api_key = "sk_test_51N6K5JB9VMe0qzbOjlJvMEsfdQyrFgV49vRaeErtmhrzHV3Cu3f5jMDJmrhKdI5uqvpHubjkmwDQgMOtCEmz19t800AouH7W6g"
 stripe.api_version = "2023-10-16"
 
-### MODELOS DE DADOS ###
+### **DATA MODELS** ###
 class PriceRequest(BaseModel):
-    amount: int  # Value in cents (e.g., R$25.00 -> 2500)
-    stylist_name: str  # Added stylist name to the request bod
-    
+    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  # Agora usamos um preço já criado
+    price_id: str  # Now using a pre-created price
 
 class CancelSubscriptionRequest(BaseModel):
     subscription_id: str
 
-### 1. CREATE DYNAMIC PRICE ###
+### **1️⃣ CREATE DYNAMIC PRICE** ###
 @router.post("/create_price")
 def create_price(data: PriceRequest):
     try:
@@ -38,7 +39,7 @@ def create_price(data: PriceRequest):
 
         price = stripe.Price.create(
             unit_amount=data.amount,
-            currency="brl",
+            currency="usd",
             recurring={"interval": "month"},
             product=product.id,
         )
@@ -49,14 +50,14 @@ def create_price(data: PriceRequest):
             "product_id": product.id,
             "product_name": product_name,
             "product_description": product_description,
-            "amount": data.amount / 100,  # Converting to real currency
-            "currency": "BRL",
+            "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 ###
+### **2️⃣ CREATE CHECKOUT SESSION** ###
 @router.post("/create_checkout_session")
 def create_checkout_session(data: SubscriptionRequest):
     try:
@@ -87,6 +88,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** ###
 @router.post("/webhook")
 async def stripe_webhook(request: Request):
     payload = await request.body()
@@ -114,65 +116,30 @@ async def stripe_webhook(request: Request):
         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} BRL")
+        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} BRL")
-        logger.info(f"🏢 Platform will receive: {platform_share / 100:.2f} BRL")
-
-        # Verifique se o payment_intent está presente
-        payment_intent = invoice.get("payment_intent")
-        if payment_intent:
-            logger.info(f"Payment Intent: {payment_intent}")
-            try:
-                transfer = stripe.Transfer.create(
-                    amount=stylist_share,  # Amount in cents
-                    currency="brl",
-                    destination=stylist_id,  # The stylist's connected Stripe account ID
-                    description="Subscription payment",
-                    source_transaction=payment_intent  # Use the payment_intent as source_transaction
-                )
-                logger.info(f"✅ Transfer successful! Transfer ID: {transfer.id}")
-            except Exception as e:
-                logger.error(f"⚠️ Error transferring funds to stylist using payment_intent: {e}")
-        else:
-            # Tente usar charge ID, caso payment_intent não esteja disponível
-            charge_id = invoice.get("charge")
-            if charge_id:
-                logger.info(f"Charge ID: {charge_id}")
-                try:
-                    transfer = stripe.Transfer.create(
-                        amount=stylist_share,  # Amount in cents
-                        currency="brl",
-                        destination=stylist_id,  # The stylist's connected Stripe account ID
-                        description="Subscription payment",
-                        source_transaction=charge_id  # Use the charge ID if payment_intent is missing
-                    )
-                    logger.info(f"✅ Transfer successful! Transfer ID: {transfer.id}")
-                except Exception as e:
-                    logger.error(f"⚠️ Error transferring funds to stylist using charge: {e}")
-            else:
-                logger.error("⚠️ No payment intent or charge found, unable to transfer funds.")
-
+        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 and transfer initiated!",
+            "status": "Payment processed successfully!",
             "user_id": user_id,
             "stylist_id": stylist_id,
-            "total_paid": amount / 100,  # Convert cents to real currency
+            "total_paid": amount / 100,  # Convert cents to dollars
             "stylist_share": stylist_share / 100,
             "platform_share": platform_share / 100,
-            "transfer_id": transfer.id if 'transfer' in locals() else None
         }
 
     return {"status": "Event received, no action needed."}
 
-### 4. CANCELAR ASSINATURA ###
+### **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,  # Cancela no final do ciclo atual
+            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: