def add_user_work(user_id, args):
user = User.get_user(user_id)
if user is None:
raise Error(StatusCode.NOT_FOUND, 'Cannot found user')
if 'work_place_link' not in args:
args['work_place_link'] = None
work = User.add_user_work(user_id, args['work_place_name'],
args['work_place_link'])
return jsonify({'status': True, 'work': work}), StatusCode.OK
def leave_group(user_id, args):
user = User.get_user(user_id)
if user is None:
raise Error(StatusCode.NOT_FOUND, 'Cannot found user')
User.leave_group(user_id, args['group_link'])
return jsonify({
'status': True,
'message': 'Leave group successfully'
}), StatusCode.OK
def delete_user_work(user_id, args):
user = User.get_user(user_id)
if user is None:
raise Error(StatusCode.NOT_FOUND, 'Cannot found user')
User.delete_user_work(user_id, args['work_place_link'])
return jsonify({
'status': True,
'message': 'Delete work successfully'
}), StatusCode.OK
def update_user(user_id, args):
user = User.get_user(user_id)
if user is None:
raise Error(StatusCode.NOT_FOUND, 'Cannot find user')
if 'user_id' not in args:
args['user_id'] = user['user_id']
if 'username' not in args:
args['username'] = user['username']
if 'user_basic_info' not in args:
args['user_basic_info'] = user['user_basic_info']
if 'user_contact_info' not in args:
args['user_contact_info'] = user['user_contact_info']
if 'user_relationship_info' not in args:
args['user_relationship_info'] = user['user_relationship_info']
new_user_id = args['user_id']
user = User.get_user(new_user_id)
if user_id != new_user_id and user is not None:
raise Error(StatusCode.BAD_REQUEST, 'User ID is already existed')
user = User.update_user(user_id, new_user_id, args['username'],
args['user_basic_info'], args['user_contact_info'],
args['user_relationship_info'])
return jsonify({'status': True, 'user': user}), StatusCode.OK
def add_user(args):
user = User.get_user(args['user_id'])
if user is not None:
raise Error(StatusCode.BAD_REQUEST, 'User is already existed')
if 'user_basic_info' not in args:
args['user_basic_info'] = None
if 'user_contact_info' not in args:
args['user_contact_info'] = None
if 'user_relationship_info' not in args:
args['user_relationship_info'] = None
user = User.add_user(args['user_id'], args['username'],
args['user_basic_info'], args['user_contact_info'],
args['user_relationship_info'])
return jsonify({'status': True, 'user': user}), StatusCode.OK
def login(cls):
""" Get user info of terminal """
fields = ("username", "password")
username, password = list(input(f"{field} :") for field in fields)
if not username and password:
print(f"username: {username}, password: {password} is empty")
""" Check user exists in database """
user = User.get_user(username)
if user:
if user.get("password", "") == password:
if not Session.exists_with_username(username):
""" User is valid and add user to session """
if Session.save(username, user.get("role")):
print("User successfully login")
print(Session.get_first_session())
return
print(
"User not found or username and password is incorrect or already login(logout for login another user)"
)
def get_user_works(user_id):
user = User.get_user(user_id)
if user is None:
raise Error(StatusCode.NOT_FOUND, 'Cannot find user')
works = User.get_user_works(user_id)
return jsonify({'status': True, 'works': works}), StatusCode.OK
def delete_user(user_id):
user = User.get_user(user_id)
if user is None:
raise Error(StatusCode.NOT_FOUND, 'Cannot find user')
User.delete_user(user_id)
return jsonify({'status': True, 'message': 'Delete user successfully'})
def get_user_groups(user_id):
user = User.get_user(user_id)
if user is None:
raise Error(StatusCode.NOT_FOUND, 'Cannot found user')
groups = User.get_user_groups(user_id)
return jsonify({'status': True, 'groups': groups}), StatusCode.OK
