Update routes/subscription.py

routes/subscription.py

@@ -23,6 +23,10 @@ SUPABASE_HEADERS = {
     "Content-Type": "application/json"
 }
 
+class CheckSubscriptionRequest(BaseModel):
+    user_id: str
+    stylist_id: str
+
 # 📌 Agora recebemos `user_id` (ID do cliente que está comprando)
 class SubscriptionRequest(BaseModel):
     id: str  # ID do estilista
@@ -172,19 +176,19 @@ def cancel_subscription(data: CancelSubscriptionRequest):
 
 # 🚀 Verificação de Assinatura
 @router.post("/check_subscription")
-def check_subscription(user_id: str, stylist_id: str):
+def check_subscription(data: CheckSubscriptionRequest):
     try:
         # Buscar todas as assinaturas ativas para o usuário (cliente)
         subscriptions = stripe.Subscription.list(
-            customer=user_id,  # ID do cliente
-            status="active",    # Assinaturas ativas
-            expand=["data.items"]  # Expande os dados para pegar os itens e o metadata
+            customer=data.user_id,  # ID do cliente
+            status="active",         # Assinaturas ativas
+            expand=["data.items"]    # Expande os dados para pegar os itens e o metadata
         )
 
         # Verificar se existe uma assinatura ativa para o estilista
         for subscription in subscriptions["data"]:
             # Verifica se o stylist_id está presente no metadata da assinatura
-            if subscription.metadata.get("stylist_id") == stylist_id:
+            if subscription.metadata.get("stylist_id") == data.stylist_id:
                 return {
                     "status": "active",
                     "subscription_id": subscription.id,