From 9aa6bcd67544d1172f0e930a9877ebbf4e436eed Mon Sep 17 00:00:00 2001
From: JJ <nicetry@noemail.com>
Date: Tue, 1 Apr 2025 09:44:32 +0100
Subject: testing hook

---
 .Constants.py.swp             | Bin 12288 -> 0 bytes
 scripts/users/handle_users.py |   5 +++--
 server.py                     |   3 ++-
 3 files changed, 5 insertions(+), 3 deletions(-)
 delete mode 100644 .Constants.py.swp

diff --git a/.Constants.py.swp b/.Constants.py.swp
deleted file mode 100644
index ffb1f96..0000000
Binary files a/.Constants.py.swp and /dev/null differ
diff --git a/scripts/users/handle_users.py b/scripts/users/handle_users.py
index 7a315f1..e2fb7af 100644
--- a/scripts/users/handle_users.py
+++ b/scripts/users/handle_users.py
@@ -3,11 +3,12 @@ mongo_collection = mongo_database.get_collection("users")
 
 def user_exists(user):
     try:
-        res = mongo_collection.find_one({"name": user}, {"password": 1})
+        res = mongo_collection.find_one({"name": user}, {"password": 1, "isAiSubscriber": 1})
         if res is None:
             return {"success": True, "user": None}
         else:
-            return {"success": True, "user": user, "password": res["password"]}
+            # Fix this - there is a better way
+            return {"success": True, "user": user, "password": res["password"], "isAiSubscriber": res["isAiSubscriber"]}
     except:
         return {"success": False, "error": "Request error, try again"}
     
diff --git a/server.py b/server.py
index 66c2d43..b6de869 100644
--- a/server.py
+++ b/server.py
@@ -126,6 +126,7 @@ def login_user():
         pw_matches = bcrypt.check_password_hash(res["password"], pw)
         if pw_matches:
             session["username"] = user
+            session["isAiSubscriber"] = res["isAiSubscriber"]
             return Response(headers={"HX-Redirect": "/home"})
         else:
             return "Incorrect password, try again"
@@ -186,7 +187,7 @@ def update_account():
         session["isAiSubscriber"] = True
         return Response(headers={"HX-Redirect": "/account"})
     else:
-        return f"<span class='error'>{response["message"]}</span>", 422
+        return f"<span class='error'>{response['message']}</span>", 422
 
     
 if __name__ == "__main__":
-- 
cgit v1.2.3