Update routes/subscription.py

routes/subscription.py

@@ -85,6 +85,20 @@ def get_new_price_id(stylist_id: str):
     if response.status_code == 200 and response.json():
         return response.json()[0].get("price_id")
     return None
+
+def update_subscription_price_in_db(subscription_id: str, new_price_id: str):
+    """
+    Atualiza a coluna 'price_id' na tabela 'Subscriptions' para o novo price_id correspondente.
+    """
+    update_data = {"price_id": new_price_id}
+
+    response = requests.patch(
+        f"{SUPABASE_URL}/rest/v1/Subscriptions?sub_id=eq.{subscription_id}",
+        headers=SUPABASE_ROLE_HEADERS,
+        json=update_data
+    )
+
+    return response.status_code in [200, 204]  # Retorna True se a atualização for bem-sucedida
     
 def verify_token(user_token: str) -> str:
     """
@@ -169,6 +183,12 @@ async def update_subscription(
 
         logger.info(f"✅ Subscription {subscription_id} updated to new price {new_price_id}")
 
+        # 🔹 8. Atualizar o price_id na tabela Subscriptions
+        if update_subscription_price_in_db(subscription_id, new_price_id):
+            logger.info(f"✅ Subscription {subscription_id} price_id updated in database.")
+        else:
+            raise HTTPException(status_code=500, detail="Failed to update subscription price_id in database.")
+
         return {
             "message": "Subscription reactivated and updated successfully!",
             "subscription_id": subscription_id,