Slide 84
Slide 84 text
def login_required(func):
def wrapper(*args, **kwargs):
user_id = session.get('user_id')
if user_id is None:
return redirect('/login', code=302)
return func(*args, **kwargs)
return wrapper
def role_required(role):
def decorator(func):
def wrapper(*args, **kwargs):
user = users.get(session.get('user_id'))
if not user.has_role(role):
abort(401)
return func(*args, **kwargs)
return wrapper
return decorator