Implement max 4 team-members not 5 members

This commit is contained in:
Koshin S Hegde 2024-09-01 13:47:11 +05:30
parent 85ec0e1a4e
commit f7d2285e74
3 changed files with 13 additions and 4 deletions

Binary file not shown.

View File

@ -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:

View File

@ -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