Update main.py

main.py

@@ -1,7 +1,5 @@
 from fastapi import FastAPI
 import requests
-from datetime import datetime, timezone
-from dateutil import parser
 import os
 
 app = FastAPI()
@@ -9,55 +7,52 @@ app = FastAPI()
 SUPABASE_URL = os.getenv("SUPABASE_URL")
 SUPABASE_API_KEY = os.getenv("SUPABASE_API_KEY")
 
-@app.get("/")
-def root():
-    return {"message": "Plan reset API is running."}
-
 @app.get("/reset-plans")
 def reset_expired_plans():
-    if not SUPABASE_URL or not SUPABASE_API_KEY:
-        return {"status": "error", "detail": "Missing Supabase config."}
-
     headers = {
         "apikey": SUPABASE_API_KEY,
         "Authorization": f"Bearer {SUPABASE_API_KEY}",
         "Content-Type": "application/json"
     }
 
-    # Fetch users whose plans are 'basic' or 'premium'
-    response = requests.get(
-        f"{SUPABASE_URL}/rest/v1/users?plan=in.(basic,premium)&select=id,plan_set_at",
+    # Fetch users on paid plans
+    r = requests.get(
+        f"{SUPABASE_URL}/rest/v1/users?plan=in.(basic,premium)&select=id,days_remaining",
         headers=headers
     )
 
-    if response.status_code != 200:
-        return {"status": "error", "detail": response.text}
+    if r.status_code != 200:
+        return {"status": "error", "detail": r.text}
 
-    users = response.json()
-    now = datetime.now(timezone.utc)
+    users = r.json()
+    reset_count = 0
+    updated_count = 0
 
-    expired_user_ids = []
     for user in users:
-        plan_set_at = user.get("plan_set_at")
-        if plan_set_at:
-            plan_date = parser.parse(plan_set_at)
-            days_passed = (now - plan_date).days
-            if days_passed >= 30:
-                expired_user_ids.append(user["id"])
-
-    # Reset plans
-    reset_count = 0
-    for user_id in expired_user_ids:
-        update = requests.patch(
-            f"{SUPABASE_URL}/rest/v1/users?id=eq.{user_id}",
-            headers=headers,
-            json={"plan": "free", "plan_set_at": None}
-        )
-        if update.status_code == 204:
-            reset_count += 1
+        uid = user["id"]
+        days = user.get("days_remaining", 0)
+
+        if days <= 1:
+            # Reset to free
+            res = requests.patch(
+                f"{SUPABASE_URL}/rest/v1/users?id=eq.{uid}",
+                headers=headers,
+                json={"plan": "free", "days_remaining": 0}
+            )
+            if res.status_code == 204:
+                reset_count += 1
+        else:
+            # Subtract one day
+            res = requests.patch(
+                f"{SUPABASE_URL}/rest/v1/users?id=eq.{uid}",
+                headers=headers,
+                json={"days_remaining": days - 1}
+            )
+            if res.status_code == 204:
+                updated_count += 1
 
     return {
         "status": "success",
-        "total_checked": len(users),
-        "expired_reset": reset_count
+        "reset_to_free": reset_count,
+        "decremented_days": updated_count
     }