def post(self):
data = UserRegister.parser.parse_args()
if UserModel.find_by_username(data['username']):
return {
'message': 'User "{}" already exists'.format(data['username'])
}, 400
UserModel(**data).save_to_db()
return {'message': 'User "{}" created'.format(data['username'])}, 201
def factory_user_model_with_contact():
"""Produce a user model."""
user_info = {
'username': '******',
'firstname': 'bar',
'lastname': 'User',
'keycloak_guid': uuid.uuid4()
}
user = UserModel(
username=user_info['username'],
firstname=user_info['firstname'],
lastname=user_info['lastname'],
keycloak_guid=user_info.get('keycloak_guid', None),
type=user_info.get('access_type', None),
)
user.save()
contact = factory_contact_model()
contact.save()
contact_link = ContactLinkModel()
contact_link.contact = contact
contact_link.user = user
contact_link.save()
return user
def factory_user_model_with_contact(user_info: dict = TestUserInfo.user1,
keycloak_guid=None):
"""Produce a user model."""
user_type = Role.ANONYMOUS_USER.name if user_info.get(
'access_type', None) == AccessType.ANONYMOUS.value else None
user = UserModel(username=user_info.get(
'username', user_info.get('preferred_username')),
firstname=user_info['firstname'],
lastname=user_info['lastname'],
keycloak_guid=user_info.get('keycloak_guid',
keycloak_guid),
type=user_type,
email='*****@*****.**',
login_source=user_info.get('loginSource'))
user.save()
contact = factory_contact_model()
contact.save()
contact_link = ContactLinkModel()
contact_link.contact = contact
contact_link.user = user
contact_link.save()
return user
def decorated_function(*args, **kwargs):
api_key = request.headers.get('api_key')
user_retrieved = UserModel.find_by_apikey(api_key)
if user_retrieved:
return view_function(*args, **kwargs)
else:
abort(401, 'Access denided. Invalid API key')
def decorated_function(*args, **kwargs):
api_key = request.headers.get('api_key')
user_retrieved = UserModel.find_by_apikey(api_key)
if user_retrieved.role == "admin":
return view_function(*args, **kwargs)
else:
abort(401, 'Access denided. Must have admin privileges')
def factory_user_model(user_info: dict = TestUserInfo.user1):
"""Produce a user model."""
user = UserModel(username=user_info['username'],
firstname=user_info['firstname'],
lastname=user_info['lastname'],
roles=user_info['roles'],
keycloak_guid=user_info['keycloak_guid'])
user.save()
return user
def post(self):
data = UserAuth.parser.parse_args()
u = UserModel.find_by_username(data['username'])
if u:
if u.check_password(data['password']):
return u.json()
return {'message': 'Incorrect password'}, 401
else:
return {
'message': 'User "{}" does not exists'.format(data['username'])
}, 400
def factory_user_model(user_info: dict = TestUserInfo.user1):
"""Produce a user model."""
user = UserModel(username=user_info['username'],
firstname=user_info['firstname'],
lastname=user_info['lastname'],
roles=user_info['roles'],
keycloak_guid=user_info.get('keycloak_guid', None),
type=user_info.get('access_type', None),
email='*****@*****.**',
login_source=user_info.get('login_source', None),
)
user.save()
return user
def factory_user_model(user_info: dict = TestUserInfo.user1):
"""Produce a user model."""
user_type = Role.ANONYMOUS_USER.name if user_info.get('access_type', None) == AccessType.ANONYMOUS.value else None
user = UserModel(username=user_info['username'],
firstname=user_info['firstname'],
lastname=user_info['lastname'],
keycloak_guid=user_info.get('keycloak_guid', None),
type=user_type,
email='*****@*****.**',
login_source=user_info.get('login_source', None),
)
user.save()
return user
def factory_user_model_with_contact(user_info: dict = TestUserInfo.user1):
"""Produce a user model."""
user = UserModel(username=user_info['username'],
firstname=user_info['firstname'],
lastname=user_info['lastname'],
roles=user_info['roles'],
keycloak_guid=user_info.get('keycloak_guid', None),
type=user_info.get('access_type', None),
email='*****@*****.**')
user.save()
contact = factory_contact_model()
contact.save()
contact_link = ContactLinkModel()
contact_link.contact = contact
contact_link.user = user
contact_link.save()
return user
def get(self):
api_key = request.headers.get('api_key')
u = UserModel.find_by_apikey(api_key)
if u:
return u.json()
return {'message': 'Invalid api_key'}