def add_user():
request_data = request.get_json()
if not request_data:
return bad_request("No input data provided")
try:
data = UserSchema().load(request_data)
except ValidationError as err:
return error_response(422, err.messages)
email = data.get('email')
password = data.get('password')
firstname = data.get('firstname')
lastname = data.get('lastname')
username = data.get('username') or \
f'{firstname}{lastname}_{int(random()*1e6)}'
try:
# check for existing user
user_email = User.find_by_identity(data.get('email'))
user_username = User.find_by_identity(data.get('username'))
if user_email is None and user_username is None:
# add new user to db
user = User()
user.firstname = firstname
user.lastname = lastname
user.username = username
user.email = email
user.password = password
user.bio = data.get('bio')
user.is_active = data.get('is_active') or False
user.is_admin = data.get('is_admin') or False
user.save()
response = jsonify({
'message': 'Successfully added new user.',
})
response.status_code = 201
response.headers['Location'] = url_for(
'users.get_user', id=user.id)
return response
else:
return bad_request('Sorry. That user already exists.')
# handle errors
except (exc.IntegrityError, ValueError):
db.session.rollback()
return error_response(500, 'Something went wrong, please try again.')
def update_user(id):
request_data = request.get_json()
if not request_data:
return bad_request("No input data provided")
try:
data = UserSchema().load(request_data)
except ValidationError as err:
return error_response(422, err.messages)
try:
user = User.find_by_id(id)
existing_user = User.find_by_identity(data.get('username'))
if existing_user is not None:
print(existing_user)
if existing_user.id != user.id:
print(existing_user.id != user.id)
return bad_request(f'Username already exists.')
# update user
user.firstname = data.get('firstname')
user.lastname = data.get('lastname')
user.username = data.get('username')
user.bio = data.get('bio')
user.is_active = data.get('is_active')
user.is_admin = data.get('is_admin')
user.save()
response = jsonify({
'message': f'Successfully updated {user.firstname}.',
})
response.status_code = 200
return response
# handle errors
except (exc.IntegrityError, ValueError):
db.session.rollback()
return bad_request('Something went wrong, please try again.')
