diff options
| -rw-r--r-- | main.py | 15 |
1 files changed, 8 insertions, 7 deletions
diff --git a/main.py b/main.py index d1e321c..dc467df 100644 --- a/main.py +++ b/main.py @@ -227,7 +227,8 @@ async def handler(websocket): await websocket.send(util.error("malformedJson", listener)) continue if r["command"] == "register": - fc = util.field_check({"username": {"range": range(1,21), "types": [str]}, "password": {"range": range(8,256), "types": [str]}, "invite_code": {"range": range(16,17), "types": [str]}}, r) + # "invite_code": {"range": range(16,17), "types": [str]} + fc = util.field_check({"username": {"range": range(1,21), "types": [str]}, "password": {"range": range(8,256), "types": [str]}, "invite_code": {"range": range(0, 199),"types": [str, None]}}, r) if fc != True: await websocket.send(util.error(fc, listener)) continue @@ -238,13 +239,13 @@ async def handler(websocket): await websocket.send(util.error("lockdown", listener)) continue r["username"] = r["username"].lower() - r["invite_code"] = r["invite_code"].upper() + # r["invite_code"] = r["invite_code"].upper() if not re.fullmatch("[a-z0-9-_.]{1,20}", r["username"]): await websocket.send(util.error("invalidUsername", listener)) continue - if r["invite_code"] not in invite_codes: - await websocket.send(util.error("invalidInvite", listener)) - continue + # if r["invite_code"] not in invite_codes: + # await websocket.send(util.error("invalidInvite", listener)) + # continue if db.acc.get(r["username"]) != "notExists": await websocket.send(util.error("usernameTaken", listener)) continue @@ -279,7 +280,7 @@ async def handler(websocket): if result != True: await websocket.send(util.error(result, listener)) continue - invite_codes.remove(r["invite_code"]) + # invite_codes.remove(r["invite_code"]) await websocket.send(json.dumps({"error": False, "token": data["secure"]["token"], "listener": listener})) elif r["command"] == "login_pswd": fc = util.field_check({"username": {"range": range(1,21), "types": [str]}, "password": {"range": range(8,256), "types": [str]}}, r) @@ -678,4 +679,4 @@ async def main(): async with serve(handler, addr, port) as server: await server.serve_forever() -asyncio.run(main()) \ No newline at end of file +asyncio.run(main()) |