Update routes/subscription.py

routes/subscription.py

@@ -837,7 +837,6 @@ async def create_price(data: CreatePriceRequest, user_token: str = Header(None,
         if not user_token:
             raise HTTPException(status_code=401, detail="Missing User-key header")
 
-        # 1. Validar o token e obter user_id
         user_id = verify_token(user_token)
         logger.info(f"🔹 User verified. user_id: {user_id}")
 
@@ -845,13 +844,11 @@ async def create_price(data: CreatePriceRequest, user_token: str = Header(None,
         emergency_price = data.emergency_price
         consultations = data.consultations
 
-        # 2. Verificar se os valores estão dentro dos limites permitidos
         if not (500 <= amount <= 99900):
             raise HTTPException(status_code=400, detail="Amount must be between $5 and $999")
         if not (500 <= emergency_price <= 99900):
             raise HTTPException(status_code=400, detail="Emergency price must be between $5 and $999")
 
-        # 3. Consultations precisa ser 0 obrigatoriamente se não for definido
         if consultations is None:
             consultations = 0
         elif consultations < 0:
@@ -859,7 +856,6 @@ async def create_price(data: CreatePriceRequest, user_token: str = Header(None,
 
         logger.info(f"🔹 Validated amounts: amount = {amount}, emergency_price = {emergency_price}, consultations = {consultations}")
 
-        # 4. Buscar price_id, name, avatar e bio do usuário no Supabase
         supabase_url = f"{SUPABASE_URL}/rest/v1/User?id=eq.{user_id}"
         headers = {
             "Authorization": f"Bearer {user_token}",
@@ -880,22 +876,19 @@ async def create_price(data: CreatePriceRequest, user_token: str = Header(None,
         user_full_name = user.get("name", "Unknown User")
         user_avatar = user.get("avatar", None)
 
-        # Criar nome abreviado do usuário
         name_parts = user_full_name.split()
         if len(name_parts) > 1:
             abbreviated_name = f"{name_parts[0]} {name_parts[1][0]}."
         else:
             abbreviated_name = name_parts[0]
-        
+
         product_name = f"{abbreviated_name} Subscription"
-        
         logger.info(f"🔹 Existing price_id: {existing_price_id}, user_name: {abbreviated_name}")
 
-        # 5. Criar produto no Stripe
         product_data = {
             "name": product_name,
         }
-        
+
         if user_avatar:
             product_data["images"] = [user_avatar]
 
@@ -903,7 +896,6 @@ async def create_price(data: CreatePriceRequest, user_token: str = Header(None,
         product_id = product.id
         logger.info(f"✅ New product created: {product_id}")
 
-        # 6. Criar novo preço no Stripe
         price = stripe.Price.create(
             unit_amount=amount,
             currency="brl",
@@ -913,50 +905,59 @@ async def create_price(data: CreatePriceRequest, user_token: str = Header(None,
         new_price_id = price.id
         logger.info(f"✅ New price created: {new_price_id}")
 
-        # 7. Cancelar as assinaturas existentes e aplicar o novo price_id sem descontos
+        # Coletar e logar todos os clientes com assinatura ativa do preço antigo
+        affected_customers = []
+
         if existing_price_id:
             subscriptions = stripe.Subscription.list(status="active")
             for sub in subscriptions.auto_paging_iter():
                 if sub["items"]["data"][0]["price"]["id"] == existing_price_id:
-                    # Cancelar assinatura no final do período
+                    customer_id = sub["customer"]
+                    affected_customers.append(customer_id)
+
+                    # Cancelar no final do período e atualizar para novo preço
                     stripe.Subscription.modify(sub.id, cancel_at_period_end=True)
-                    # Aplicar o novo preço na próxima fatura
                     stripe.Subscription.modify(
                         sub.id,
                         items=[{
                             "id": sub["items"]["data"][0]["id"],
                             "price": new_price_id
                         }],
-                        proration_behavior="none"  # Garantir que não haja prorrogação ou descontos
+                        proration_behavior="none"
                     )
-                    logger.info(f"🔹 Subscription {sub.id} set to cancel at period end and updated with new price {new_price_id}.")
+                    logger.info(f"🔹 Subscription {sub.id} updated with new price {new_price_id} (customer: {customer_id})")
+
+        logger.info(f"📢 Affected customer IDs with old price: {affected_customers}")
 
-        # 8. Atualizar Supabase com o novo price_id
         update_data = {
             "price_id": new_price_id,
             "price": amount,
             "emergency_price": emergency_price,
             "consultations": consultations
         }
-        
+
         update_headers = {
             "Authorization": f"Bearer {user_token}",
             "apikey": SUPABASE_KEY,
             "Content-Type": "application/json"
         }
-        
+
         update_response = requests.patch(supabase_url, headers=update_headers, json=update_data)
-        
+
         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, price, emergency_price, and consultations")
-        return {"message": "Price created and user updated successfully!", "price_id": new_price_id}
+
+        logger.info(f"✅ Successfully updated user {user_id} with new price_id and prices")
+        return {
+            "message": "Price created and user updated successfully!",
+            "price_id": new_price_id,
+            "affected_customers": affected_customers
+        }
 
     except Exception as e:
         logger.error(f"❌ Error creating price: {e}")
         raise HTTPException(status_code=500, detail=f"Error creating price: {str(e)}")
-
+        
 @router.post("/emergency_checkout_session")
 def emergency_checkout_session(
     data: EmergencyPaymentRequest,