Update routes/subscription.py

routes/subscription.py

@@ -507,7 +507,35 @@ def check_subscription(
         logger.info(f"Stripe response data: {subscriptions}")
 
         if not subscriptions or "data" not in subscriptions or not subscriptions["data"]:
-            return {"status": "inactive", "message": "No subscriptions found for this user."}
+            # Se não houver assinaturas para o cliente, desativa as assinaturas associadas ao estilista no banco de dados
+            logger.info(f"No active subscription found for user {user_id}. Deactivating all subscriptions for stylist {data.stylist_id}.")
+
+            # 🔹 Buscar todas as assinaturas do estilista no banco de dados para este cliente
+            response_subscriptions = requests.get(
+                f"{SUPABASE_URL}/rest/v1/Subscriptions?customer_id=eq.{user_id}&stylist_id=eq.{data.stylist_id}",
+                headers=SUPABASE_HEADERS
+            )
+            if response_subscriptions.status_code == 200:
+                subscriptions_data = response_subscriptions.json()
+                for sub in subscriptions_data:
+                    # 🔹 Atualiza para "active = false" para todas as assinaturas do estilista no banco
+                    update_data = {
+                        "active": False
+                    }
+                    subscription_url = f"{SUPABASE_URL}/rest/v1/Subscriptions?id=eq.{sub['id']}"
+                    update_response = requests.patch(
+                        subscription_url,
+                        headers=SUPABASE_HEADERS,
+                        json=update_data
+                    )
+                    if update_response.status_code == 200:
+                        logger.info(f"✅ Subscription {sub['id']} deactivated successfully for stylist {data.stylist_id}.")
+                    else:
+                        logger.error(f"❌ Failed to deactivate subscription {sub['id']} for stylist {data.stylist_id}.")
+            else:
+                logger.error(f"❌ Failed to fetch subscriptions from Supabase for stylist {data.stylist_id} and user {user_id}.")
+            
+            return {"status": "inactive", "message": "No active subscription found for stylist."}
 
         # 🔹 Verifica se existe uma assinatura ativa para o estilista
         stylist_found = False
@@ -564,19 +592,19 @@ def check_subscription(
                     "expiration_date": expiration_date
                 }
 
-        # 🔹 Caso não tenha encontrado assinatura ativa para o estilista, desativa todas as assinaturas no banco de dados
+        # 🔹 Caso não tenha encontrado assinatura ativa para o estilista
         if not stylist_found:
-            logger.info(f"No active subscription found for stylist {data.stylist_id}. Deactivating all subscriptions for this user.")
-            
-            # 🔹 Buscar todas as assinaturas do cliente no banco de dados
+            logger.info(f"No active subscription found for stylist {data.stylist_id}. Deactivating all subscriptions for this stylist.")
+
+            # 🔹 Buscar todas as assinaturas do estilista no banco de dados
             response_subscriptions = requests.get(
-                f"{SUPABASE_URL}/rest/v1/Subscriptions?customer_id=eq.{user_id}",
+                f"{SUPABASE_URL}/rest/v1/Subscriptions?stylist_id=eq.{data.stylist_id}",
                 headers=SUPABASE_HEADERS
             )
             if response_subscriptions.status_code == 200:
                 subscriptions_data = response_subscriptions.json()
                 for sub in subscriptions_data:
-                    # 🔹 Atualiza para "active = false" para todas as assinaturas do cliente no banco
+                    # 🔹 Atualiza para "active = false" para todas as assinaturas do estilista no banco
                     update_data = {
                         "active": False
                     }
@@ -587,11 +615,11 @@ def check_subscription(
                         json=update_data
                     )
                     if update_response.status_code == 200:
-                        logger.info(f"✅ Subscription {sub['id']} deactivated successfully for user {user_id}.")
+                        logger.info(f"✅ Subscription {sub['id']} deactivated successfully for stylist {data.stylist_id}.")
                     else:
-                        logger.error(f"❌ Failed to deactivate subscription {sub['id']} for user {user_id}.")
+                        logger.error(f"❌ Failed to deactivate subscription {sub['id']} for stylist {data.stylist_id}.")
             else:
-                logger.error(f"❌ Failed to fetch subscriptions from Supabase for user {user_id}.")
+                logger.error(f"❌ Failed to fetch subscriptions from Supabase for stylist {data.stylist_id}.")
             
             return {"status": "inactive", "message": "No active subscription found for stylist."}