Update routes/subscription.py

routes/subscription.py

@@ -117,35 +117,67 @@ async def sync_emergency_payments(request: Request, data: EmergencyPaymentReques
         existing_payments = response_payments.json()
         logger.info(f"Found {len(existing_payments)} emergency payments for stylist {stylist_id} and customer {user_id}")
 
-        # Passo 5: Verificar os pagamentos e inserir os que faltam
-        for payment in existing_payments:
-            payment_id = payment.get("payment_id")
-            if not payment_id:
-                # Inserir pagamento de emergência que não foi encontrado
-                emergency_sub_data = {
-                    "stylist_id": stylist_id,
-                    "client_id": user_id,
-                    "payment_id": payment_id,  # Exemplo de dado de pagamento
-                    "price": 0,  # Valor do pagamento
-                    # Adicione outros campos se necessário
-                }
-                response_insert = requests.post(
-                    f"{SUPABASE_URL}/rest/v1/Emergency_sub",
-                    headers=SUPABASE_HEADERS,
-                    json=emergency_sub_data
-                )
-                
-                if response_insert.status_code == 201:
-                    logger.info(f"Payment {payment_id} added successfully")
-                else:
-                    logger.error(f"Failed to add payment {payment_id}: {response_insert.status_code} - {response_insert.text}")
-        
+        # Passo 5: Buscar pagamentos no Stripe para esse user_id e stylist_id
+        stripe_payments = stripe.PaymentIntent.list(
+            customer=stripe_id,
+            expand=["data.charges"]
+        )
+
+        # Passo 6: Verificar os pagamentos e inserir os que faltam
+        for stripe_payment in stripe_payments["data"]:
+            payment_id = stripe_payment["id"]
+            
+            # Verificar se o payment_id já existe no Supabase
+            existing_payment = next((payment for payment in existing_payments if payment["payment_id"] == payment_id), None)
+            
+            if existing_payment:
+                logger.info(f"Payment {payment_id} already exists in database, skipping.")
+                continue  # Pular se já existe
+
+            # Se não existe, adicionar pagamento de emergência ao Supabase
+            amount = stripe_payment["amount_received"] / 100  # Valor recebido
+            currency = stripe_payment["currency"].upper()  # Moeda
+
+            payment_method = stripe_payment.get("payment_method")
+            payment_type = "N/A"
+            last4 = "N/A"
+            if payment_method:
+                payment_method_details = stripe.PaymentMethod.retrieve(payment_method)
+                payment_type = payment_method_details["type"]
+                last4 = payment_method_details[payment_type].get("last4", "N/A")
+
+            emergency_sub_data = {
+                "stylist_id": stylist_id,
+                "client_id": user_id,
+                "payment_id": payment_id,
+                "amount": amount,
+                "currency": currency,
+                "payment_method": f"{payment_type.upper()} ending in {last4}",
+                "status": stripe_payment["status"],  # Status do pagamento
+                # Adicione outros campos necessários
+            }
+
+            # Inserir o novo pagamento de emergência no Supabase
+            response_insert = requests.post(
+                f"{SUPABASE_URL}/rest/v1/Emergency_sub",
+                headers=SUPABASE_HEADERS,
+                json=emergency_sub_data
+            )
+
+            if response_insert.status_code == 201:
+                logger.info(f"Payment {payment_id} added successfully")
+            else:
+                logger.error(f"Failed to add payment {payment_id}: {response_insert.status_code} - {response_insert.text}")
+
         return {"status": "success", "message": "Emergency payments synchronized successfully"}
 
     except jwt.ExpiredSignatureError:
         raise HTTPException(status_code=401, detail="Token has expired")
     except jwt.InvalidTokenError:
         raise HTTPException(status_code=401, detail="Invalid token")
+    except stripe.error.StripeError as e:
+        logger.error(f"Stripe error: {str(e)}")
+        raise HTTPException(status_code=500, detail=f"Stripe error: {str(e)}")
     except Exception as e:
         logger.error(f"Error in syncing emergency payments: {str(e)}")
         raise HTTPException(status_code=500, detail=f"Error in syncing emergency payments: {str(e)}")