def get(self):
user_info = UserSchema(only=("username", "id", "email", "role",
"avatarURL"))
_id = get_jwt_identity()
user = UserModule(id=_id)
current_user = UserModule(**user.find_by_id())
reponse_user = user_info.dump(current_user)
return make_response(reponse_user)
def add_claims_to_jwt(identity):
user = UserModule(id=identity)
current_user = user.find_by_id()
if current_user["role"] == 'admin':
return {'role': 'admin'}
elif current_user["role"] == 'artist' or current_user[
"role"] == 'collector':
return {'role': 'vendor'}
else:
return {'role': 'user'}