From f0e53c49538539903238cbb857769258fbc93b09 Mon Sep 17 00:00:00 2001 From: kosh Date: Mon, 2 Sep 2024 04:49:59 +0530 Subject: [PATCH] It's 5 O clock --- backend/data/sqlite-database.db | Bin 32768 -> 32768 bytes backend/src/rest_api.py | 33 +++++++++++++------------------- backend/src/utils/team.py | 7 +++++-- backend/src/utils/user.py | 2 +- 4 files changed, 19 insertions(+), 23 deletions(-) diff --git a/backend/data/sqlite-database.db b/backend/data/sqlite-database.db index 7bbe1816e2c6b909c54899c1c20f665a32333e1b..129bd784b3bd7e17d77a895c03654e5d4999702c 100644 GIT binary patch delta 441 zcmZo@U}|V!njpG_sg72FprIic2!H zh;lNj7$vC~8mS~EnWvRz7&}H(lohy#o0b=P6q)7f6+}3@MrK3?mSp?ng_flj2PEg` zrWj|075k*5T6mULCKhMpBv#}m<~gM2CT8a70i`)T@^dN`f>KjbQ*%>`6kPMtb25uF z6ns-tGVM!q6@rs9^7C_yj8Pym$I!^w(1`JogXc7+OOZhHg}E3Q7#R7VGw?s(ESPYY zfAV2{2}a3H2Cw)PB!My@Ss>tKVHRgBNlnZxhN}P0!2cbn{w==%2Qv#J$S_G>pgLhb z@5$@|O04;Ov3%aZK#}FE7ZB!Q5ar+n0U#NYT3jL^gf8MH2o&Lzm6Vp2W|5Q!sVOW^ s0V5z$UYG)5!$^<_P@G}+y{`}!OmT{!01tyW+^&q&oSgj4r{c^M0ddNC8UO$Q delta 688 zcmZo@U}|V!njpAr1km0LuG!z$PXk;;EbXHVU z78Pe^5#?l5F-lS~G*U_R$P6^i%heBcDe%hEj|$5UEhq|44au#ra4RYFOAbl*3#};h zwa8C$H}y9z4fXb|$kI!#Of1gGNvz0C%yUT3P0Y;EOU}>b^h(UkQwS(ZRS3-X&6>wVO$a# zm|GrF?`v7)*>mAX9*VlLe@$ zBsDR&SP)|FV+Q{3{EzwH0L{J1ufoC1!U&2SaV}<7Mo8F*fy6*=5#a^u7vx(ynLR*> faVap-Rr3i62=Xw9a`1uxkPJyJF4??2&QlQp8GF)q diff --git a/backend/src/rest_api.py b/backend/src/rest_api.py index 2ba7640..ce6bb0d 100644 --- a/backend/src/rest_api.py +++ b/backend/src/rest_api.py @@ -1,5 +1,4 @@ import os -from os.path import isdir import secrets import sqlite3 import flask @@ -20,8 +19,7 @@ def register_mail() -> flask.Response: return flask.Response(f"mail or password is missing", 422) try: utils.MailHandler.send_code(mail) - except Exception as e: - print(e) + except Exception: return flask.Response("Something went wrong, check mail or try again later", 404) response = flask.Response() response.set_cookie("mail", mail) @@ -31,11 +29,9 @@ def register_mail() -> flask.Response: @app.post("/verify-mail-code") def confirm_mail_code() -> flask.Response: try: - print(flask.request.form) mail = flask.request.cookies["mail"] token = flask.request.form["token"] - except KeyError as e: - print("\n\n\n", e) + except KeyError: return flask.Response("Token missing or /register_mail not called", 422) try: success = utils.MailHandler.verify_code(mail, token) @@ -43,9 +39,10 @@ def confirm_mail_code() -> flask.Response: return flask.Response("Register mail not called. Try again", 404) if success: flask.session["mail"] = mail - response = flask.Response(str(success)) - response.delete_cookie("mail") - return response + response = flask.Response(str(success)) + response.delete_cookie("mail") + return response + return flask.Response("Wrong code", 400) @app.post("/register-leader") @@ -97,7 +94,7 @@ def login() -> flask.Response: except KeyError: return flask.Response("Mail doesn't exist", 404) except AssertionError: - return flask.Response("Wrong password", 422) + return flask.Response("Wrong password", 400) @app.post("/get-mail") @@ -111,7 +108,6 @@ def get_id() -> flask.Response: @app.post("/logout") def logout() -> flask.Response: try: - print(flask.session) flask.session.pop("user_id") except KeyError: return flask.Response("Not logged in", 428) @@ -120,13 +116,11 @@ def logout() -> flask.Response: @app.post("/register-teammate") def register() -> flask.Response: - print(flask.request.form) try: user_id = flask.session["user_id"] except KeyError: return flask.Response("Not logged in", 428) try: - print(utils.UserHandler.get_user_by_id(user_id)) team_id = utils.UserHandler.get_user_by_id(user_id).team_id utils.UserHandler.create_user(utils.User( -1, @@ -144,7 +138,7 @@ def register() -> flask.Response: if error_message == "UNIQUE constraint failed: users.phone_number": return flask.Response("Phone number taken", 400) if error_message == "UNIQUE constraint failed: users.mail": - return flask.Response("Mail taken") + return flask.Response("Mail taken", 400) return flask.Response("Phone/ pin code wrong", 400) return flask.Response() @@ -193,9 +187,9 @@ def upload_project_files() -> flask.Response: except KeyError: return flask.Response("file missing", 422) file_name = file.filename if file.filename else secrets.token_urlsafe(20) - if not os.path.isdir(f"./data/project-files/{team_id}"): - os.mkdir(f"./data/project-files/{team_id}") - file.save(f"./data/project-files/{team_id}/{file_name}") + if not os.path.isdir(f"../data/project-files/{team_id}"): + os.mkdir(f"../data/project-files/{team_id}") + file.save(f"../data/project-files/{team_id}/{file_name}") return flask.Response() @@ -212,10 +206,9 @@ def get_info() -> flask.Response: return flask.Response("Not logged in", 428) team_id = utils.UserHandler.get_user_by_id(user_id).team_id team_members = utils.TeamHandler.get_team_members(team_id) - print(team_members) for team_member in team_members: info["team_members"].append(utils.UserHandler.get_user_by_id(team_member).to_dict()) info["project_details"] = vars(utils.TeamHandler.get_project(team_id)) - if os.path.isdir(f"./data/project-files/{team_id}"): - info["project_files"] = os.listdir(f"./data/project-files/{team_id}") + if os.path.isdir(f"../data/project-files/{team_id}"): + info["project_files"] = os.listdir(f"../data/project-files/{team_id}") return flask.jsonify(info) diff --git a/backend/src/utils/team.py b/backend/src/utils/team.py index d58bb65..394557b 100644 --- a/backend/src/utils/team.py +++ b/backend/src/utils/team.py @@ -33,7 +33,10 @@ class Team: ) -> None: self.team_id = int(team_id) self.team_name = team_name - self.is_related_to_special_topics = bool(int(is_related_to_special_topics)) + try: + self.is_related_to_special_topics = bool(int(is_related_to_special_topics)) + except ValueError: + self.is_related_to_special_topics = is_related_to_special_topics == "true" self.problem_statement = problem_statement self.title = title self.solution = solution @@ -45,7 +48,7 @@ class Team: class TeamHandler: - __connection: sqlite3.Connection = sqlite3.connect("./data/sqlite-database.db", check_same_thread=False) + __connection: sqlite3.Connection = sqlite3.connect("../data/sqlite-database.db", check_same_thread=False) __cursor: sqlite3.Cursor = __connection.cursor() @classmethod diff --git a/backend/src/utils/user.py b/backend/src/utils/user.py index cdca8a8..4c68fa7 100644 --- a/backend/src/utils/user.py +++ b/backend/src/utils/user.py @@ -62,7 +62,7 @@ class User: return the_dict class UserHandler: - __connection: sqlite3.Connection = sqlite3.connect("./data/sqlite-database.db", check_same_thread=False) + __connection: sqlite3.Connection = sqlite3.connect("../data/sqlite-database.db", check_same_thread=False) __cursor: sqlite3.Cursor = __connection.cursor() @classmethod