Update routes/subscription.py

routes/subscription.py

@@ -242,75 +242,77 @@ async def stripe_webhook(request: Request):
     logger.info(f"🔹 Event received: {event['type']}")
 
     if event["type"] == "checkout.session.completed":
-        session = event["data"]["object"]
-        payment_intent_id = session.get("payment_intent")
-
-        # Verifica se o PaymentIntent ID está presente
-        if not payment_intent_id:
-            logger.error("⚠️ No PaymentIntent ID found in session.")
-            # Verifica se a sessão de checkout tem um PaymentIntent associado
-            if "subscription" in session:
-                subscription_id = session["subscription"]
-                subscription = stripe.Subscription.retrieve(subscription_id)
-                payment_intent_id = subscription.latest_invoice.payment_intent
-                logger.info(f"🔹 Found PaymentIntent from subscription: {payment_intent_id}")
-            else:
-                raise HTTPException(status_code=400, detail="PaymentIntent ID missing.")
-
-        try:
-            # Recupera o PaymentIntent associado à sessão
-            payment_intent = stripe.PaymentIntent.retrieve(payment_intent_id)
-        except Exception as e:
-            logger.error(f"⚠️ Error retrieving PaymentIntent: {e}")
-            raise HTTPException(status_code=400, detail="Error retrieving PaymentIntent")
-
-        # Extrai os dados do metadata
-        user_id = session.metadata.get("user_id")  # ID do cliente
-        stylist_id = session.metadata.get("stylist_id")  # ID do estilista no Stripe
-        consultations_per_month = session.metadata.get("consultations_per_month")  # Consultas por mês
-
-        # Verifica o valor pago
-        amount = payment_intent["amount_received"]
-
-        # Define a divisão do pagamento entre a plataforma e o estilista
-        stylist_share = int(amount * 0.8)  # 80% para o estilista
-        platform_share = amount - stylist_share  # 20% para a plataforma
-
-        # Realizando a transferência para o estilista
-        try:
-            transfer = stripe.Transfer.create(
-                amount=stylist_share,  # Montante para o estilista (80%)
-                currency="brl",  # Moeda
-                destination=stylist_id,  # Conta do estilista
-                description="Payment for styling service"
-            )
-            logger.info(f"💸 Transfer to stylist {stylist_id} successful: R${stylist_share / 100:.2f} BRL.")
-        except Exception as e:
-            logger.error(f"⚠️ Error transferring funds to stylist: {e}")
-            raise HTTPException(status_code=500, detail="Error transferring funds to stylist.")
-
-        # Transferência para a plataforma
-        try:
-            transfer_platform = stripe.Transfer.create(
-                amount=platform_share,  # Montante para a plataforma (20%)
-                currency="brl",  # Moeda
-                destination="your_platform_stripe_account_id",  # Substitua com sua conta Stripe da plataforma
-                description="Platform share"
-            )
-            logger.info(f"💸 Transfer to platform successful: R${platform_share / 100:.2f} BRL.")
-        except Exception as e:
-            logger.error(f"⚠️ Error transferring funds to platform: {e}")
-            raise HTTPException(status_code=500, detail="Error transferring funds to platform.")
+    session = event["data"]["object"]
+    payment_intent_id = session.get("payment_intent")
+
+    # Verifica se o PaymentIntent ID está presente
+    if not payment_intent_id:
+        logger.error("⚠️ No PaymentIntent ID found in session.")
+        # Verifica se a sessão de checkout tem um PaymentIntent associado
+        if "subscription" in session:
+            subscription_id = session["subscription"]
+            subscription = stripe.Subscription.retrieve(subscription_id)
+            invoice_id = subscription.latest_invoice  # Aqui obtemos o ID da fatura
+            invoice = stripe.Invoice.retrieve(invoice_id)  # Recupera a fatura
+            payment_intent_id = invoice.payment_intent  # Agora conseguimos pegar o PaymentIntent
+            logger.info(f"🔹 Found PaymentIntent from subscription: {payment_intent_id}")
+        else:
+            raise HTTPException(status_code=400, detail="PaymentIntent ID missing.")
 
-        return {
-            "status": "Payment processed successfully!",
-            "user_id": user_id,
-            "stylist_id": stylist_id,
-            "consultations_per_month": consultations_per_month,
-            "total_paid": amount / 100,
-            "stylist_share": stylist_share / 100,
-            "platform_share": platform_share / 100
-        }
+    try:
+        # Recupera o PaymentIntent associado à sessão
+        payment_intent = stripe.PaymentIntent.retrieve(payment_intent_id)
+    except Exception as e:
+        logger.error(f"⚠️ Error retrieving PaymentIntent: {e}")
+        raise HTTPException(status_code=400, detail="Error retrieving PaymentIntent")
+
+    # Extrai os dados do metadata
+    user_id = session.metadata.get("user_id")  # ID do cliente
+    stylist_id = session.metadata.get("stylist_id")  # ID do estilista no Stripe
+    consultations_per_month = session.metadata.get("consultations_per_month")  # Consultas por mês
+
+    # Verifica o valor pago
+    amount = payment_intent["amount_received"]
+
+    # Define a divisão do pagamento entre a plataforma e o estilista
+    stylist_share = int(amount * 0.8)  # 80% para o estilista
+    platform_share = amount - stylist_share  # 20% para a plataforma
+
+    # Realizando a transferência para o estilista
+    try:
+        transfer = stripe.Transfer.create(
+            amount=stylist_share,  # Montante para o estilista (80%)
+            currency="brl",  # Moeda
+            destination=stylist_id,  # Conta do estilista
+            description="Payment for styling service"
+        )
+        logger.info(f"💸 Transfer to stylist {stylist_id} successful: R${stylist_share / 100:.2f} BRL.")
+    except Exception as e:
+        logger.error(f"⚠️ Error transferring funds to stylist: {e}")
+        raise HTTPException(status_code=500, detail="Error transferring funds to stylist.")
+
+    # Transferência para a plataforma
+    try:
+        transfer_platform = stripe.Transfer.create(
+            amount=platform_share,  # Montante para a plataforma (20%)
+            currency="brl",  # Moeda
+            destination="your_platform_stripe_account_id",  # Substitua com sua conta Stripe da plataforma
+            description="Platform share"
+        )
+        logger.info(f"💸 Transfer to platform successful: R${platform_share / 100:.2f} BRL.")
+    except Exception as e:
+        logger.error(f"⚠️ Error transferring funds to platform: {e}")
+        raise HTTPException(status_code=500, detail="Error transferring funds to platform.")
+
+    return {
+        "status": "Payment processed successfully!",
+        "user_id": user_id,
+        "stylist_id": stylist_id,
+        "consultations_per_month": consultations_per_month,
+        "total_paid": amount / 100,
+        "stylist_share": stylist_share / 100,
+        "platform_share": platform_share / 100
+    }
 
     return {"status": "Event received, no action needed."}