使用多个用户类别实现Flask-Login

Question

我正在编写一个具有多个充当用户的类的应用程序（例如，学校帐户和员工帐户）。我正在尝试使用Flask-
Login简化操作，但是我不太确定如何做到这一点，因此当用户登录时，我可以让我的应用检查用户名是否属于School帐户或职员帐户，然后正确登录。

我知道如何找出它属于哪种 类型 的帐户（因为所有用户名都必须是唯一的）。但是之后，我不确定如何告诉应用程序我希望它登录该特定用户。

现在，我只有一个通用登录页面。如果我为职员帐户和学校帐户分别创建登录页面会更容易吗？我正在通过Flask-SQLAlchemy使用MySQL数据库。

Answer 1

您可以使用特定角色定义每个用户。例如，用户“ x”可以是SCHOOL，而用户“ y”可以是“ STAFF”。

class User(db.Model):

    __tablename__ = 'User'
    id = db.Column(db.Integer,primary_key=True)
    username = db.Column(db.String(80),unique=True)
    pwd_hash = db.Column(db.String(200))
    email = db.Column(db.String(256),unique=True)
    is_active = db.Column(db.Boolean,default=False)
    urole = db.Column(db.String(80))


    def __init__(self,username,pwd_hash,email,is_active,urole):
            self.username = username
            self.pwd_hash = pwd_hash
            self.email = email
            self.is_active = is_active
            self.urole = urole

    def get_id(self):
            return self.id
    def is_active(self):
            return self.is_active
    def activate_user(self):
            self.is_active = True         
    def get_username(self):
            return self.username
    def get_urole(self):
            return self.urole

但是Flask-login还没有用户角色的概念，我写了自己的login_required装饰器版本来覆盖它。因此，您可能想要使用类似以下的内容：

def login_required(role="ANY"):
    def wrapper(fn):
        @wraps(fn)
        def decorated_view(*args, **kwargs):

            if not current_user.is_authenticated():
               return current_app.login_manager.unauthorized()
            urole = current_app.login_manager.reload_user().get_urole()
            if ( (urole != role) and (role != "ANY")):
                return current_app.login_manager.unauthorized()      
            return fn(*args, **kwargs)
        return decorated_view
    return wrapper

然后，可以在如下视图函数上使用此装饰器：

@app.route('/school/')
@login_required(role="SCHOOL")
def restricted_view_for_school():
    pass