SQLAlchemy automap - 为自动化模型添加方法

Question

我有一个与SQLAlchemy一起使用的预先存在的数据库，所以我使用的是自动从数据库中获取模型。向这些类添加方法的最佳方法是什么？例如，对于User类，我想添加诸如验证密码之类的方法。另外，我想添加flask-login（UserMixin）方法的方法。

Answer 1

事先

Specify your classes explicitly，并按照正常情况定义您的方法：

Base = automap_base()

class User(Base):
    __tablename__ = 'user'

    def verify_password(self, password):
        ...

Base.prepare(engine, reflect=True)

现在Base.classes.User和User与您的其他方法相同。要使您的User类瓶子登录兼容，请实施listed attributes and methods，或将提供的UserMixin添加到User班级。 mixin似乎只期望id类中存在User属性/列。