DEV Community

Discussion on: Python Decorator and Flask

Collapse
 
yangshuxuan profile image
yangshuxuan

You'd chang the code
def admin_login_required():
def wrap(*args, **kwargs):
# user is available from @login_required
if not g.user.is_admin:
return "you need to be admin", 401
return f(*args, **kwargs)
to------------------------------------------
def admin_login_required(f):
def wrap(*args, **kwargs):
# user is available from @login_required
if not g.user.is_admin:
return "you need to be admin", 401
return f(*args, **kwargs)
return wrap