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'}