def test_exists_no_pkey_fail(app, mocker):
# mock db query
query_mock = mocker.patch('flask_sqlalchemy._QueryProperty.__get__')
query_mock.return_value \
.get.return_value = None
errors, models = exists({}, SomeModel, 'field', None)
assert errors == {'field': ["Missing data for required field."]}
assert models is None
def post_user():
"""Creates a new user.
:returns: JSON string of the new user's data; status code
:rtype: (str, int)
"""
# pre-validate data
errors = unique({}, User, User.username,
request.json.get('username', None))
errors = unique_email(errors, User, User.email,
request.json.get('email', None))
errors, roles = exists(errors, Role, 'roles',
request.json.get('roles', []))
# validate data
try:
data = UserAdminSchema().load(request.json)
except ValidationError as err:
errors = dict(list(errors.items()) + list(err.messages.items()))
# return any errors
if errors:
return jsonify({"error": errors}), 400
# save user
user = User(username=data['username'].lower().strip(),
email=data['email'].lower().strip(),
password=data['password'],
is_verified=data['is_verified'],
status=data['status'],
status_changed_at=datetime.now())
for role in roles:
user.roles.append(role)
# save user profile
if 'profile' in data:
user_profile = UserProfile(
user=user,
first_name=data['profile']['first_name'].strip(),
last_name=data['profile']['last_name'].strip(),
joined_at=data['profile']['joined_at'],
status=data['status'],
status_changed_at=datetime.now())
db.session.add(user_profile)
db.session.add(user)
db.session.commit()
# response
return jsonify({'user': UserAdminSchema().dump(user)}), 201
def test_exists_no_pkey_custom_message_fail(app, mocker):
# mock db query
query_mock = mocker.patch('flask_sqlalchemy._QueryProperty.__get__')
query_mock.return_value \
.get.return_value = None
errors, models = exists({}, SomeModel, 'field', None, missing_error="bar")
assert errors == {'field': ["bar"]}
assert models is None
def test_exists_multiple_fail(app, mocker):
# mock db query
query_mock = mocker.patch('flask_sqlalchemy._QueryProperty.__get__')
query_mock.return_value \
.get.return_value = None
errors, models = exists({}, SomeModel, 'field', [1, 2])
assert errors == {'field': ["Invalid value."]}
assert models == []
def test_exists_multiple_pass(app, mocker):
test_model = SomeModel()
test_model.id = 1
# mock db query
query_mock = mocker.patch('flask_sqlalchemy._QueryProperty.__get__')
query_mock.return_value \
.get.return_value = test_model
errors, models = exists({}, SomeModel, 'field', [1, 2])
assert errors == {}
assert models == [test_model, test_model]
def post_administrator():
"""Creates a new administrator
:returns: JSON string of the new administrator's data; status code
:rtype: (str, int)
"""
# pre-validate data
errors = unique({}, Administrator, Administrator.username,
request.json.get('username', None))
errors = unique_email(errors, Administrator, Administrator.email,
request.json.get('email', None))
errors, roles = exists(errors, Role, 'roles',
request.json.get('roles', []))
# validate data
try:
data = AdministratorAdminSchema().load(request.json)
except ValidationError as err:
errors = dict(list(errors.items()) + list(err.messages.items()))
# return any errors
if errors:
return jsonify({"error": errors}), 400
# save admin
admin = Administrator(username=data['username'].lower().strip(),
email=data['email'].lower().strip(),
first_name=data['first_name'],
last_name=data['last_name'],
password=data['password'],
joined_at=data['joined_at'],
status=data['status'],
status_changed_at=datetime.now())
for role in roles:
admin.roles.append(role)
db.session.add(admin)
db.session.commit()
# response
return jsonify({'administrator':
AdministratorAdminSchema().dump(admin)}), 201
def put_user(user_id):
"""Updates an existing user.
:param user_id: ID of user
:type user_id: int
:returns: JSON string of the user's data; status code
:rtype: (str, int)
"""
# get user
user = User.query.get(user_id)
if user is None:
abort(404)
# pre-validate data
errors = unique({},
User,
User.username,
request.json.get('username', None),
update=user)
errors = unique_email(errors,
User,
User.email,
request.json.get('email', None),
update=user)
errors, roles = exists(errors, Role, 'roles',
request.json.get('roles', []))
# validate data
try:
if request.json.get('password', None):
data = UserAdminSchema().load(request.json)
else:
data = UserAdminSchema(exclude=('password', )).load(request.json)
except ValidationError as err:
errors = dict(list(errors.items()) + list(err.messages.items()))
# return any errors
if errors:
return jsonify({"error": errors}), 400
# save user
user.username = data['username'].lower().strip()
user.email = data['email'].lower().strip()
user.is_verified = data['is_verified']
if 'password' in data:
user.password = data['password']
user.roles[:] = []
for role in roles:
user.roles.append(role)
if user.status != data['status']:
user.status = data['status']
user.status_changed_at = datetime.now()
# save user profile
if 'profile' in data:
user_profile = user.profile if user.profile else None
if user_profile:
user_profile.first_name = data['profile']['first_name'].strip()
user_profile.last_name = data['profile']['last_name'].strip()
user_profile.joined_at = data['profile']['joined_at']
if user_profile.status != data['status']:
user_profile.status = data['status']
user_profile.status_changed_at = datetime.now()
else:
user_profile = UserProfile(
user_id=user.id,
first_name=data['profile']['first_name'].strip(),
last_name=data['profile']['last_name'].strip(),
joined_at=data['profile']['joined_at'],
status=data['status'],
status_changed_at=datetime.now())
db.session.add(user_profile)
user.profile = user_profile
db.session.commit()
# response
return jsonify({'user': UserAdminSchema().dump(user)}), 200
def put_administrator(administrator_id):
"""Updates an existing administrator
:param administrator_id: ID of administrator
:type administrator_id: int
:returns: JSON string of the administrator's data; status code
:rtype: (str, int)
"""
# get administrator
administrator = Administrator.query.get(administrator_id)
if administrator is None:
abort(404)
# pre-validate data
errors = unique({},
Administrator,
Administrator.username,
request.json.get('username', None),
update=administrator)
errors = unique_email(errors,
Administrator,
Administrator.email,
request.json.get('email', None),
update=administrator)
errors, roles = exists(errors, Role, 'roles',
request.json.get('roles', []))
# validate data
try:
if request.json.get('password', None):
data = AdministratorAdminSchema().load(request.json)
else:
data = AdministratorAdminSchema(exclude=('password', )).load(
request.json)
except ValidationError as err:
errors = dict(list(errors.items()) + list(err.messages.items()))
# return any errors
if errors:
return jsonify({"error": errors}), 400
# save administrator
administrator.username = data['username'].lower().strip()
administrator.email = data['email'].lower().strip()
administrator.first_name = data['first_name']
administrator.last_name = data['last_name']
administrator.joined_at = data['joined_at']
if 'password' in data:
administrator.password = data['password']
administrator.roles[:] = []
for role in roles:
administrator.roles.append(role)
if administrator.status != data['status']:
administrator.status = data['status']
administrator.status_changed_at = datetime.now()
db.session.commit()
# response
return jsonify(
{'administrator': AdministratorAdminSchema().dump(administrator)}), 200
def post_user_account_step1():
"""User registration step 1.
:returns: JSON string of the user's account information; status code
:rtype: (str, int)
"""
# pre-validate data
errors = unique({}, User, User.username,
str(request.json.get('username')).lower().strip()
if request.json.get('username', None) else None)
errors = unique_email(
errors, User, User.email,
str(request.json.get('email')).lower().strip() if request.json.get(
'email', None) else None)
errors, tos = exists(errors,
TermsOfService,
'tos_id',
request.json.get('tos_id', None),
missing_error="Please agree to the terms of service.")
if (request.json.get('password', None)
and request.json.get('password2', None)):
if request.json.get('password') != request.json.get('password2'):
errors['password2'] = ["Passwords must match."]
# validate data
try:
data = UserAccountSchema(exclude=(
'first_name',
'last_name',
)).load(request.json)
except ValidationError as err:
errors = dict(list(errors.items()) + list(err.messages.items()))
# return any errors
if errors:
return jsonify({"error": errors}), 400
# save user
user = User(username=data['username'].lower().strip(),
email=data['email'].lower().strip(),
password=data['password'],
is_verified=False,
status=User.STATUS_ENABLED,
status_changed_at=datetime.now())
user_role = Role.query.filter(Role.name == 'USER').first()
if user_role:
user.roles.append(user_role)
db.session.add(user)
# save user terms of service
user_tos = UserTermsOfService(user=user,
terms_of_service=tos,
accept_date=datetime.now(),
ip_address=request.environ.get(
'HTTP_X_REAL_IP', request.remote_addr))
db.session.add(user_tos)
# save password history
pass_history = UserPasswordHistory(user=user,
password=user.password,
set_date=datetime.now())
db.session.add(pass_history)
db.session.commit()
# prep output
output = {
'id': user.id,
'username': user.username,
'email': user.email,
'password_changed_at': user.password_changed_at,
'is_verified': user.is_verified,
'first_name': None,
'last_name': None,
'joined_at': None,
}
# response
return jsonify({'user_account': UserAccountSchema().dump(output)}), 201