diff --git a/backend/data/sqlite-database.db b/backend/data/sqlite-database.db index 42aca8a..e06b1f6 100644 Binary files a/backend/data/sqlite-database.db and b/backend/data/sqlite-database.db differ diff --git a/backend/src/rest_api.py b/backend/src/rest_api.py index 546c49d..593a22b 100644 --- a/backend/src/rest_api.py +++ b/backend/src/rest_api.py @@ -56,6 +56,7 @@ def register_leader() -> flask.Response: -1, mail=mail, team_id=team_id, + team_leader=1, **request_data, )) user_id = utils.UserHandler.get_user_by_mail(mail).user_id @@ -117,11 +118,13 @@ def register() -> flask.Response: 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, secrets.token_urlsafe(20), team_id=team_id, + team_leader=0, **flask.request.form, )) except TypeError: diff --git a/backend/src/utils/user.py b/backend/src/utils/user.py index 70f1f89..f1fecc4 100644 --- a/backend/src/utils/user.py +++ b/backend/src/utils/user.py @@ -20,6 +20,7 @@ class User: phone_number: int name: str team_id: int + team_leader: int def __init__( self, @@ -36,6 +37,7 @@ class User: pin_code: int|str, phone_number: int|str, name: str, + team_leader: bool|int|str, team_id: int|str, ) -> None: self.user_id: int = int(user_id) @@ -52,6 +54,7 @@ class User: self.phone_number: int = int(phone_number) self.name = name self.team_id = int(team_id) + self.team_leader = int(team_leader) def to_dict(self) -> dict: the_dict = vars(self) @@ -67,7 +70,7 @@ class UserHandler: cls, user: User ) -> None: - assert len(utils.TeamHandler.get_team_members(user.team_id)) <= 4 + assert len(utils.TeamHandler.get_team_members(user.team_id)) < 4 salt = bcrypt.gensalt() encrypted_password = bcrypt.hashpw(user.password.encode("utf-8"), salt) cls.__cursor.execute( @@ -88,7 +91,7 @@ class UserHandler: team_id, team_leader ) - VALUES(?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, 1) + VALUES(?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?) """, ( encrypted_password, @@ -104,6 +107,7 @@ class UserHandler: user.phone_number, user.name, user.team_id, + user.team_leader, ) ) cls.__connection.commit() @@ -134,7 +138,8 @@ class UserHandler: user[10], user[11], user[12], - user[13] + user[13], + user[14], ) @classmethod @@ -163,7 +168,8 @@ class UserHandler: user[10], user[11], user[12], - user[13] + user[13], + user[14], ) @classmethod