如果没有,是否有任何项目已将此功能添加到Flask-Login?否则,从Flask-Login迁移到Flask-User似乎有点令人生畏。否则,是否有任何方向可以从Flask-Login迁移到Flask-User?
再一次,在这里回答我自己的问题让其他人好奇你如何在仍然使用Flask-Login时添加处理多个角色的功能。我创建了下面的装饰器,它只检查current_user.role以查看它是否为“Admin”。在让用户登录时,您还应该检查相同的内容,具体取决于他们是否登录到管理员或用户面板。
from functools import wraps
def admin_required(f):
@wraps(f)
def wrap(*args, **kwargs):
if current_user.role == "Admin":
return f(*args, **kwargs)
else:
flash("You need to be an admin to view this page.")
return redirect(url_for('index'))
return wrap