Question

我试图在我的烧瓶应用程序中声明一个User模型，以便使用Flask-Login扩展实现登录。从关于sql炼金术的烧瓶文档中有一个我用于另一个名为employees的模型的例子。这是代码：

class Employee(db.Model):
    __tablename__ = "employees"
    id = db.Column(db.Integer, primary_key=True)
    name = db.Column(db.String(200))
    title = db.Column(db.String(200))
    email = db.Column(db.String(200))
    department = db.Column(db.String(200))

    def __init__(self, name, title, email, department):
        self.name = name
        self.title = title
        self.email = email
        self.department = department

    def __repr__(self):
        return '<Employee %r>' % self.name

这是作为例外Flask SQL-Alchemy Docs

从此页面中获取的

我很困惑，因为我也使用alembic来运行迁移，所以使用alembic revision -m "create user table我已经创建了一个＆＃39;用户＆＃39;表。我使用Flask Sql-Alchemy指南的建议创建了我的第一个模型表（员工）：

from yourapplication.database import init_db
init_db()

这是混乱。我现在需要建立用户身份验证模型。我该怎么做呢？到目前为止，这是我的代码：

class User(flask_login.UserMixin):
    def __init__(self, username, password):
        self.id = username
        self.password = password

请注意，我的User类继承自flask_login.Usermixin。我需要这个才能让Flask Login工作，所以现在它没有像db.Model中的Employee模型那样实例化。但问题是我创建了一个简化迁移来建立我的用户模型。为什么我无法查询数据库。我错过了哪一块拼图？

Answer 1

所有UserMixin都提供了一个通用接口，任何用户模型都需要实现这个接口才能使用Flask-Login。它没有设置SQLAlchemy模型，它将混合到模型中。

class User(db.Model, UserMixin):
    id = db.Column(db.Integer, primary_key=True)
    name = db.Column(db.String, nullable=False, unique=True)
    active = db.Column(db.Boolean, nullable=False, default=True)

    @property
    def is_active(self):
        # override UserMixin property which always returns true
        # return the value of the active column instead
        return self.active

SQLAlchemy将Python类映射到数据库表。要创建这些表格（如果它们不存在），请致电db.create_all()。如果表已经存在，如果您更改了模型，则不会更改它，您需要删除并重新创建表。

由于删除表意味着您丢失了所有数据，因此在更改模型时迁移架构和数据更好。 Alembic将生成表示这些更改的脚本。不使用create_all，而是使用Alembic生成创建表的迁移，然后运行迁移。

alembic revision --autogenerate -m 'create user'
alembic upgrade head

现在您已拥有用户模型，并使用初始迁移脚本创建了表，您可以对表执行查询并对实例执行与登录相关的操作。

# create a user
u = User(name='davidism')
db.session.add(u)
db.session.commit()

# query all users
users = User.query.all()

# update a user
u = User.query.filter_by(name='davidism').one()
u.active = False
db.session.commit()

# check if a user is active like Flask-Login
print(u.is_active)

如何定义和创建用于Flask-Login的用户模型？

1 个答案: