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 = []
|
