Update routes/subscription.py

routes/subscription.py

@@ -58,9 +58,8 @@ async def create_price(
         if not amount or not user_id:
             raise HTTPException(status_code=400, detail="Amount and user_id are required")
 
-        # 🔹 1. Buscar o price_id atual do usuário no Supabase
+        # 🔹 Buscar `price_id` salvo no Supabase
         supabase_url = f"{SUPABASE_URL}/rest/v1/User?id=eq.{user_id}"
-        
         supabase_headers = {
             "apikey": SUPABASE_KEY,
             "Authorization": f"Bearer {user_token}",  # 🔥 Passando token do usuário
@@ -74,22 +73,17 @@ async def create_price(
             raise HTTPException(status_code=404, detail="User not found")
 
         user = user_data[0]
-        current_price_id = user.get("price_id")  # 🔍 Pegamos o price_id atual do usuário
+        existing_price_id = user.get("price_id")
 
-        # 🔹 2. Se o price_id existir, tentar atualizar
-        if current_price_id:
+        # 🔥 Se já existir um price_id, desativar o preço antigo e criar um novo
+        if existing_price_id:
             try:
-                stripe.Price.modify(
-                    current_price_id,
-                    active=True,  # Mantém ativo
-                    metadata={"updated_at": "now"}  # Apenas um metadado de controle
-                )
-                logger.info(f"🔄 Price updated successfully: {current_price_id}")
-                return {"message": "Price updated successfully!", "price_id": current_price_id}
-            except stripe.error.InvalidRequestError:
-                logger.warning(f"⚠️ Invalid price_id: {current_price_id}, creating a new one...")
-
-        # 🔹 3. Se o price_id não existir ou for inválido, criar um novo
+                stripe.Price.modify(existing_price_id, active=False)  # Desativando preço antigo
+                logger.info(f"🔹 Disabled old price: {existing_price_id}")
+            except Exception as e:
+                logger.warning(f"⚠️ Could not disable old price: {e}")
+
+        # 🔹 Criar novo preço no Stripe
         price = stripe.Price.create(
             unit_amount=amount,
             currency="brl",
@@ -97,25 +91,23 @@ async def create_price(
             product_data={"name": "Custom Subscription Price"}
         )
 
-        logger.info(f"✅ New price created: {price.id} for user {user_id}")
+        logger.info(f"✅ Price created successfully: {price.id} for user {user_id}")
 
-        # 🔹 4. Atualizar Supabase com o novo price_id
-        update_response = requests.patch(
+        # 🔹 Atualizar Supabase com novo `price_id`
+        response_update = requests.patch(
             supabase_url,
             headers=supabase_headers,
             json={"price_id": price.id}
         )
 
-        logger.info(f"🔹 Supabase response: {update_response.status_code}, {update_response.text}")
-
-        if update_response.status_code not in [200, 204]:
-            raise HTTPException(status_code=500, detail=f"Failed to update Supabase: {update_response.text}")
+        if response_update.status_code not in [200, 204]:
+            raise HTTPException(status_code=500, detail=f"Failed to update Supabase: {response_update.text}")
 
-        return {"message": "New price created successfully!", "price_id": price.id}
+        return {"message": "Price created successfully!", "price_id": price.id}
 
     except Exception as e:
-        logger.error(f"❌ Error creating/updating price: {e}")
-        raise HTTPException(status_code=500, detail="Error creating/updating price.")
+        logger.error(f"❌ Error creating price: {e}")
+        raise HTTPException(status_code=500, detail="Error creating price.")
         
 @router.post("/create_checkout_session")
 def create_checkout_session(data: SubscriptionRequest):