Update routes/subscription.py

routes/subscription.py

@@ -5,6 +5,7 @@ from datetime import datetime
 import pytz
 import os
 import requests
+import asyncio
 import jwt
 from fastapi import APIRouter, HTTPException, Request, Header
 from pydantic import BaseModel
@@ -474,7 +475,7 @@ def cancel_subscription(data: CancelSubscriptionRequest):
         raise HTTPException(status_code=500, detail=str(e))
 
 @router.post("/check_subscription")
-def check_subscription(
+async def check_subscription(
     data: CheckSubscriptionRequest,
     user_token: str = Header(None, alias="User-key")
 ):
@@ -486,9 +487,9 @@ def check_subscription(
         user_id = verify_token(user_token)
 
         # 🔹 Buscar o stripe_id do usuário no Supabase
-        response_user = requests.get(
+        response_user = await async_request(
             f"{SUPABASE_URL}/rest/v1/User?id=eq.{user_id}",
-            headers=SUPABASE_HEADERS
+            SUPABASE_HEADERS
         )
         user_data = response_user.json()
         if not user_data:
@@ -499,46 +500,28 @@ def check_subscription(
             raise HTTPException(status_code=404, detail="Stripe customer not found for user")
 
         # 🔹 Buscar todas as assinaturas do cliente (ativas, canceladas, expiradas, etc.)
-        subscriptions = stripe.Subscription.list(
+        subscriptions = await async_stripe_request(
+            stripe.Subscription.list,
             customer=user_stripe_id,
             expand=["data.items"]
         )
 
         logger.info(f"Stripe response data: {subscriptions}")
 
-        if not subscriptions or "data" not in subscriptions or not subscriptions["data"]:
+        if not subscriptions or not subscriptions["data"]:
             # 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(
+            response_subscriptions = await async_request(
                 f"{SUPABASE_URL}/rest/v1/Subscriptions?customer_id=eq.{user_id}&stylist_id=eq.{data.stylist_id}",
-                headers=SUPABASE_HEADERS
+                SUPABASE_HEADERS
             )
             if response_subscriptions.status_code == 200:
                 subscriptions_data = response_subscriptions.json()
-
-                # 🔹 Preparar a atualização em lote
-                update_data = [{"id": sub["id"], "active": False} for sub in subscriptions_data]
-                
-                # Atualiza todas as assinaturas do estilista no banco de dados de uma vez
-                if update_data:
-                    update_response = requests.patch(
-                        f"{SUPABASE_URL}/rest/v1/Subscriptions",
-                        headers=SUPABASE_HEADERS,
-                        json=update_data
-                    )
-                    if update_response.status_code == 200:
-                        logger.info(f"✅ All subscriptions for stylist {data.stylist_id} deactivated successfully.")
-                    else:
-                        logger.error(f"❌ Failed to deactivate subscriptions for stylist {data.stylist_id}.")
-                
-                # Retorna as assinaturas desativadas no banco
-                return {
-                    "status": "inactive",
-                    "message": "No active subscription found for stylist.",
-                    "subscriptions_updated": subscriptions_data
-                }
+                await asyncio.gather(
+                    *(update_subscription_status(sub['id'], False) for sub in subscriptions_data)
+                )
             else:
                 logger.error(f"❌ Failed to fetch subscriptions from Supabase for stylist {data.stylist_id} and user {user_id}.")
             
@@ -578,17 +561,13 @@ def check_subscription(
                     "canceled_date": canceled_at_date
                 }
 
-                response_subscription = requests.post(
+                await async_request(
                     f"{SUPABASE_URL}/rest/v1/Subscriptions",
-                    headers=SUPABASE_HEADERS,
-                    json=subscription_data
+                    SUPABASE_HEADERS,
+                    json=subscription_data,
+                    method='POST'
                 )
 
-                if response_subscription.status_code == 201:
-                    logger.info(f"✅ Subscription updated successfully for user {user_id}")
-                else:
-                    logger.error(f"❌ Failed to update subscription: {response_subscription.status_code} - {response_subscription.text}")
-
                 # Retorna informações sobre a assinatura sincronizada
                 return {
                     "status": status,
@@ -604,38 +583,53 @@ def check_subscription(
             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(
+            response_subscriptions = await async_request(
                 f"{SUPABASE_URL}/rest/v1/Subscriptions?stylist_id=eq.{data.stylist_id}",
-                headers=SUPABASE_HEADERS
+                SUPABASE_HEADERS
             )
             if response_subscriptions.status_code == 200:
                 subscriptions_data = response_subscriptions.json()
-
-                # 🔹 Preparar a atualização em lote
-                update_data = [{"id": sub["id"], "active": False} for sub in subscriptions_data]
-                
-                # Atualiza todas as assinaturas do estilista no banco de dados de uma vez
-                if update_data:
-                    update_response = requests.patch(
-                        f"{SUPABASE_URL}/rest/v1/Subscriptions",
-                        headers=SUPABASE_HEADERS,
-                        json=update_data
-                    )
-                    if update_response.status_code == 200:
-                        logger.info(f"✅ All subscriptions for stylist {data.stylist_id} deactivated successfully.")
-                    else:
-                        logger.error(f"❌ Failed to deactivate subscriptions for stylist {data.stylist_id}.")
-                
-                # Retorna as assinaturas desativadas no banco
-                return {
-                    "status": "inactive",
-                    "message": "No active subscription found for stylist.",
-                    "subscriptions_updated": subscriptions_data
-                }
+                await asyncio.gather(
+                    *(update_subscription_status(sub['id'], False) for sub in subscriptions_data)
+                )
+            else:
+                logger.error(f"❌ Failed to fetch subscriptions from Supabase for stylist {data.stylist_id}.")
+            
+            return {"status": "inactive", "message": "No active subscription found for stylist."}
 
     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 checking subscription: {str(e)}")
-        raise HTTPException(status_code=500, detail="Error checking subscription.")
+        raise HTTPException(status_code=500, detail="Error checking subscription.")
+
+
+# Função assíncrona para fazer requisições de rede
+async def async_request(url, headers, json=None, method='GET'):
+    method_func = requests.post if method == 'POST' else requests.get
+    response = method_func(url, headers=headers, json=json)
+    return response
+
+# Função assíncrona para chamar a Stripe API
+async def async_stripe_request(func, **kwargs):
+    loop = asyncio.get_event_loop()
+    response = await loop.run_in_executor(None, lambda: func(**kwargs))
+    return response
+
+# Função assíncrona para atualizar o status de uma assinatura
+async def update_subscription_status(subscription_id, status):
+    update_data = {
+        "active": status
+    }
+    subscription_url = f"{SUPABASE_URL}/rest/v1/Subscriptions?id=eq.{subscription_id}"
+    update_response = await async_request(
+        subscription_url,
+        SUPABASE_HEADERS,
+        json=update_data,
+        method='PATCH'
+    )
+    if update_response.status_code == 200:
+        logger.info(f"✅ Subscription {subscription_id} deactivated successfully.")
+    else:
+        logger.error(f"❌ Failed to deactivate subscription {subscription_id}.")