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類的燒瓶登錄兼容，請實施listed attributes and methods，或將所提供的UserMixin添加到User類。 mixin似乎只預計User類的id屬性/列的存在。