diff options
| -rw-r--r-- | db.py | 8 | ||||
| -rw-r--r-- | main.py | 2 |
2 files changed, 9 insertions, 1 deletions
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 = [] |