From dc1a5d63ca318befa962c0d748a12dd428d7fe68 Mon Sep 17 00:00:00 2001 From: "berry :3" Date: Fri, 25 Apr 2025 20:45:07 +0300 Subject: fix deleted accs --- db.py | 8 ++++++++ main.py | 2 +- 2 files changed, 9 insertions(+), 1 deletion(-) diff --git a/db.py b/db.py index cdc84dc..1d164ee 100644 --- a/db.py +++ b/db.py @@ -28,6 +28,14 @@ inboxd = database[os.getenv("INBOX_COLLECTION")] class acc: def get(username): + user = usersd.find_one({"username": username, "$nor": [{"deleted": True}]}) + # user = usersd.find_one({"username": username}) + if not user: + return "notExists" + else: + return user + + def get_if_deleted(username): user = usersd.find_one({"username": username}) if not user: return "notExists" diff --git a/main.py b/main.py index f5a0209..4d2f85f 100644 --- a/main.py +++ b/main.py @@ -246,7 +246,7 @@ async def handler(websocket): # if r["invite_code"] not in invite_codes: # await websocket.send(util.error("invalidInvite", listener)) # continue - if db.acc.get(r["username"]) != "notExists": + if db.acc.get_if_deleted(r["username"]) != "notExists": await websocket.send(util.error("usernameTaken", listener)) continue ips = [] -- cgit 1.4.1-2-gfad0