Update routes/subscription.py

routes/subscription.py

@@ -393,16 +393,19 @@ def check_subscription(
         # Logando a resposta do Stripe para verificar sua estrutura
         logger.info(f"Stripe response data: {subscriptions}")
 
-        if "data" not in subscriptions or not subscriptions["data"]:
-            raise HTTPException(status_code=500, detail="No subscriptions data found in response")
+        if not subscriptions or "data" not in subscriptions or not subscriptions["data"]:
+            return {
+                "status": "inactive",
+                "message": "No active subscription found for this stylist."
+            }
 
         # 🔹 Verificar se existe uma assinatura ativa para o estilista
-        for subscription in subscriptions["data"]:  # Aqui, estamos agora iterando sobre os itens da lista
+        for subscription in subscriptions["data"]:
             # Verifica se o stylist_id está presente no metadata da assinatura
-            if subscription.metadata.get("stylist_id") == data.stylist_id:
+            if subscription["metadata"].get("stylist_id") == data.stylist_id:
                 # Se encontrou a assinatura ativa, vamos registrar a assinatura no Supabase
-                subscription_id = subscription.id  # ID da assinatura
-                price_id = subscription.items.data[0].price.id  # ID do preço da assinatura
+                subscription_id = subscription["id"]  # ID da assinatura
+                price_id = subscription["items"]["data"][0]["price"]["id"]  # ID do preço da assinatura
                 
                 # 🔹 Definir o fuso horário de Nova York usando pytz
                 nyc_tz = pytz.timezone('America/New_York')
@@ -425,7 +428,7 @@ def check_subscription(
                     "Content-Type": "application/json"
                 }
 
-                # Inserir nova linha na tabela Subscriptions (não precisamos definir 'id' explicitamente)
+                # Inserir nova linha na tabela Subscriptions
                 subscription_url = f"{SUPABASE_URL}/rest/v1/Subscriptions"
                 response_subscription = requests.post(
                     subscription_url,
@@ -441,8 +444,8 @@ def check_subscription(
 
                 return {
                     "status": "active",
-                    "subscription_id": subscription.id,
-                    "consultations_per_month": subscription.metadata.get("consultations_per_month")
+                    "subscription_id": subscription["id"],
+                    "consultations_per_month": subscription["metadata"].get("consultations_per_month", None)
                 }
 
         # Caso não tenha encontrado assinatura ativa para o estilista
@@ -453,8 +456,9 @@ def check_subscription(
 
     except stripe.error.StripeError as e:
         # Captura erros específicos do Stripe
+        logger.error(f"Stripe error: {str(e)}")
         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}")
+        logger.error(f"Error checking subscription: {str(e)}")
         raise HTTPException(status_code=500, detail="Error checking subscription.")