Update main.py

main.py

@@ -1,46 +1,50 @@
-from fastapi import FastAPI, Request
-import requests
-from datetime import datetime, timezone
-from dateutil import parser
-import os
-
-app = FastAPI()
-
-SUPABASE_URL = os.getenv("SUPABASE_URL")
-SUPABASE_API_KEY = os.getenv("SUPABASE_API_KEY")
-
-@app.get("/reset-plans")
-def reset_expired_plans():
-    headers = {
-        "apikey": SUPABASE_API_KEY,
-        "Authorization": f"Bearer {SUPABASE_API_KEY}",
-        "Content-Type": "application/json"
-    }
-
-    # Step 1: Fetch users with basic or premium
-    r = 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()
-
-    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 their plan to free
-    for uid in expired_ids:
-        requests.patch(
-            f"{SUPABASE_URL}/rest/v1/users?id=eq.{uid}",
-            headers=headers,
-            json={"plan": "free", "plan_set_at": None}
-        )
-
-    return {"status": "success", "reset_count": len(expired_ids)}
+from fastapi import FastAPI
+import requests
+from datetime import datetime, timezone
+from dateutil import parser
+import os
+
+app = FastAPI()
+
+SUPABASE_URL = os.getenv("SUPABASE_URL")
+SUPABASE_API_KEY = os.getenv("SUPABASE_API_KEY")
+
+@app.get("/reset-plans")
+def reset_expired_plans():
+    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(
+        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()
+
+    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}",
+            headers=headers,
+            json={"plan": "free", "plan_set_at": None}
+        )
+
+    return {
+        "status": "success",
+        "reset_count": len(expired_ids),
+        "reset_users": expired_ids
+    }