python - SQLAlchemy automap - 向自动映射模型添加方法

Question

我有一个与 SQLAlchemy 一起使用的预先存在的数据库，因此我使用 automap 从数据库中获取模型。向这些类添加方法的最佳方法是什么？例如，对于 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 类 Flask-login 兼容，请实现 listed attributes and methods ，或添加提供的 UserMixin到您的 User 类。 mixin 似乎只期望 User 类中存在 id 属性/列。