Update routes/subscription.py

routes/subscription.py

@@ -241,15 +241,21 @@ async def stripe_webhook(request: Request):
 
     logger.info(f"🔹 Event received: {event['type']}")
 
-    # Verifica se o evento é de uma sessão de checkout completada
     if event["type"] == "checkout.session.completed":
         session = event["data"]["object"]
         payment_intent_id = session.get("payment_intent")
 
-        # Verifica se o payment_intent existe
+        # Verifica se o PaymentIntent ID está presente
         if not payment_intent_id:
             logger.error("⚠️ No PaymentIntent ID found in session.")
-            raise HTTPException(status_code=400, detail="PaymentIntent ID missing 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
@@ -270,13 +276,8 @@ async def stripe_webhook(request: Request):
         stylist_share = int(amount * 0.8)  # 80% para o estilista
         platform_share = amount - stylist_share  # 20% para a plataforma
 
-        # Logando os valores
-        logger.info(f"💸 Payment of R${amount / 100:.2f} BRL received for stylist {stylist_id}.")
-        logger.info(f"💸 Stylist share: R${stylist_share / 100:.2f} BRL, Platform share: R${platform_share / 100:.2f} BRL.")
-
         # Realizando a transferência para o estilista
         try:
-            # Transfere os fundos para o estilista (Stripe Connect)
             transfer = stripe.Transfer.create(
                 amount=stylist_share,  # Montante para o estilista (80%)
                 currency="brl",  # Moeda
@@ -288,7 +289,7 @@ async def stripe_webhook(request: Request):
             logger.error(f"⚠️ Error transferring funds to stylist: {e}")
             raise HTTPException(status_code=500, detail="Error transferring funds to stylist.")
 
-        # Você também pode transferir a parte da plataforma, se necessário
+        # Transferência para a plataforma
         try:
             transfer_platform = stripe.Transfer.create(
                 amount=platform_share,  # Montante para a plataforma (20%)
@@ -301,14 +302,10 @@ async def stripe_webhook(request: Request):
             logger.error(f"⚠️ Error transferring funds to platform: {e}")
             raise HTTPException(status_code=500, detail="Error transferring funds to platform.")
 
-        # Aqui você pode atualizar seu banco de dados com as informações do pagamento
-        # Exemplo de atualização do banco com o valor e o status do pagamento
-        # update_payment_status(user_id, stylist_id, amount, stylist_share, platform_share)
-
         return {
             "status": "Payment processed successfully!",
-            "user_id": user_id,  # Retorna o ID do cliente
-            "stylist_id": stylist_id,  # Retorna o ID do estilista
+            "user_id": user_id,
+            "stylist_id": stylist_id,
             "consultations_per_month": consultations_per_month,
             "total_paid": amount / 100,
             "stylist_share": stylist_share / 100,