Update main.py

main.py

@@ -9,42 +9,55 @@ 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"
     }
 
-    # Step 1: Get users with non-free plans
-    r = requests.get(
+    # 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",
         headers=headers
     )
 
-    if r.status_code != 200:
-        return {"status": "error", "detail": r.text}
-
-    users = r.json()
+    if response.status_code != 200:
+        return {"status": "error", "detail": response.text}
 
+    users = response.json()
     now = datetime.now(timezone.utc)
-    expired_ids = [
-        user["id"]
-        for user in users
-        if user.get("plan_set_at") and (now - parser.parse(user["plan_set_at"])).days >= 30
-    ]
-
-    # Step 2: Reset plans
-    for uid in expired_ids:
-        requests.patch(
-            f"{SUPABASE_URL}/rest/v1/users?id=eq.{uid}",
+
+    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
 
     return {
         "status": "success",
-        "reset_count": len(expired_ids),
-        "reset_users": expired_ids
+        "total_checked": len(users),
+        "expired_reset": reset_count
     }