如何使用flask_login库上的current_user函数使用两种用户？

Question

我尝试构建一个使用两种用户的烧瓶应用程序，我想在两个类之间使用 current_user 分开。 我需要让 load_user 返回普通用户或交付用户。

这是加载用户功能：

@login.user_loader
def load_user(id):
    return User.query.get(int(id))

这是用户类：

class User(UserMixin, db.Model):
    id = db.Column(db.Integer, primary_key=True)
    username = db.Column(db.String(64), index=True, unique=True)
    email = db.Column(db.String(120), index=True, unique=True)
    password_hash = db.Column(db.String(128))
    about_me = db.Column(db.String(140))
    last_seen = db.Column(db.DateTime, default=datetime.utcnow)

class Delivery(UserMixin, db.Model):
    id = db.Column(db.Integer, primary_key=True)
    username = db.Column(db.String(64), index=True, unique=True)
    email = db.Column(db.String(120), index=True, unique=True)
    password_hash = db.Column(db.String(128))
    about_me = db.Column(db.String(140))
    last_seen = db.Column(db.DateTime, default=datetime.utcnow)

Answer 1

只需使用一个用户模型并键入字段或组模型。 例子：

class User(UserMixin, db.Model):
    id = db.Column(db.Integer, primary_key=True)
    username = db.Column(db.String(64), index=True, unique=True)
    email = db.Column(db.String(120), index=True, unique=True)
    password_hash = db.Column(db.String(128))
    about_me = db.Column(db.String(140))
    last_seen = db.Column(db.DateTime, default=datetime.utcnow)
    group = db.Column(db.ForeignKey('group.id'), nullable=False)

class Group(db.Model):
    id = db.Column(db.Integer, primary_key=True)
    users = db.relationship('User', backref='user', lazy=True)
    description = db.Column(db.String(50))