auth

1 files changed, 5 insertions, 6 deletions

diff --git a/db.py b/db.py
index 3feef6c..88dc79b 100644
--- a/db.py
+++ b/db.py
@@ -4,6 +4,7 @@ from passlib.hash import scrypt
 from pymongo.mongo_client import MongoClient
 from pymongo.server_api import ServerApi
 from dotenv import load_dotenv
+import time
 
 load_dotenv()
 
@@ -63,16 +64,14 @@ class acc:
             return "fail"
         return True
 
-    def verify(username, token):
-        user = usersd.find_one({"username": username})
+    def verify(token):
+        user = usersd.find_one({"secure.token": token})
         if not user:
             return "notExists"
         if user["banned_until"] > round(time.time()):
-            return "banned"
-        elif user["secure"]["token"] != token:
-            return "unauthorized"
+            return {"banned": True, "username": user["username"]}
         else:
-            return True
+            return {"banned": False, "username": user["username"]}
 
     def verify_pswd(username, password):
         user = usersd.find_one({"username": username})