Update routes/subscription.py

routes/subscription.py

@@ -38,6 +38,8 @@ class SubscriptionRequest(BaseModel):
 
 class CreatePriceRequest(BaseModel):
     amount: int  # Valor em centavos (ex: 2500 para R$25,00)
+    emergency_price: int  # Valor de emergência (ex: 500 para R$5,00)
+    consultations: int  # Número de consultas (ex: 3)
 
 def verify_token(user_token: str) -> str:
     """
@@ -125,8 +127,11 @@ async def create_price(
         logger.info(f"🔹 User verified. user_id: {user_id}")
 
         amount = data.amount
-        if not amount:
-            raise HTTPException(status_code=400, detail="Amount is required")
+        emergency_price = data.emergency_price
+        consultations = data.consultations
+
+        if not amount or not emergency_price or not consultations:
+            raise HTTPException(status_code=400, detail="Amount, emergency_price, and consultations are required")
 
         # 🔹 2. Buscar price_id do usuário no Supabase
         supabase_url = f"{SUPABASE_URL}/rest/v1/User?id=eq.{user_id}"
@@ -166,14 +171,13 @@ async def create_price(
                     stripe.Subscription.modify(sub.id, cancel_at_period_end=True)
                     logger.info(f"🔹 Subscription {sub.id} set to cancel at period end.")
 
-        # 🔹 5. Atualizar Supabase com o novo price_id
-        update_data = {"price_id": new_price_id}  # Corrigido para 'new_price_id'
-        headers = {
-            "Authorization": f"Bearer {user_token}",  # Adicionando o token no header
-            "apikey": SUPABASE_KEY,  # Corrigindo para garantir que a chave da API é incluída
-            "Content-Type": "application/json"
+        # 🔹 5. Atualizar Supabase com o novo price_id e valores adicionais
+        update_data = {
+            "price_id": new_price_id,  # Novo price_id
+            "price": amount,  # Atualizando o valor de 'price'
+            "emergency_price": emergency_price,  # Atualizando o valor de 'emergency_price'
+            "consultations": consultations  # Atualizando o valor de 'consultations'
         }
-        
         update_response = requests.patch(supabase_url, headers=headers, json=update_data)
         
         # Log detalhado para verificar a resposta
@@ -182,8 +186,8 @@ async def create_price(
         if update_response.status_code not in [200, 204]:
             raise HTTPException(status_code=500, detail=f"Failed to update Supabase: {update_response.text}")
         
-        logger.info(f"✅ Successfully updated user {user_id} with new price_id {new_price_id}")
-        return {"message": "Price created successfully!", "price_id": new_price_id}
+        logger.info(f"✅ Successfully updated user {user_id} with new price_id, price, emergency_price, and consultations")
+        return {"message": "Price created and user updated successfully!", "price_id": new_price_id}
 
     except Exception as e:
         logger.error(f"❌ Error creating price: {e}")