Update routes/subscription.py

routes/subscription.py

@@ -168,4 +168,39 @@ def cancel_subscription(data: CancelSubscriptionRequest):
         )
         return {"status": "Subscription will be canceled at period end"}
     except Exception as e:
-        raise HTTPException(status_code=500, detail=str(e))
+        raise HTTPException(status_code=500, detail=str(e))
+
+# 🚀 Verificação de Assinatura
+@router.post("/check_subscription")
+def check_subscription(user_id: str, stylist_id: str):
+    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
+        )
+
+        # 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:
+                return {
+                    "status": "active",
+                    "subscription_id": subscription.id,
+                    "consultations_per_month": subscription.metadata.get("consultations_per_month")
+                }
+
+        # Caso não tenha encontrado assinatura ativa para o estilista
+        return {
+            "status": "inactive",
+            "message": "No active subscription found for this stylist."
+        }
+
+    except stripe.error.StripeError as e:
+        # Captura erros específicos do Stripe
+        raise HTTPException(status_code=500, detail=f"Stripe error: {str(e)}")
+    except Exception as e:
+        # Captura outros erros genéricos
+        logger.error(f"Error checking subscription: {e}")
+        raise HTTPException(status_code=500, detail="Error checking subscription.")