Flask SQLAlchemy：如何扩展基础模型类？

Question

在 Flask/SQLAlchemy 中，我们使用来自 flask_sqlalchemy 的 Model 类而不是来自 sqlalchemy 的类，如下所示：

from flask import Flask
from flask_sqlalchemy import SQLAlchemy

app = Flask(__name__)
db = SQLAlchemy(app)

class Foo(db.Model):
    bar = db.Column(db.String)

我有一个添加到我定义的每个模型的函数。我想扩展一些基类并在那里添加函数，所以我不必重复自己。使用普通的 SQLAlchemy，我们可以扩展声明性基类来做到这一点。我们如何在flask_sqlalchemy 中实现这一点？

如果不提升，从 db.Model 类扩展是行不通的：

InvalidRequestError：类没有指定表或表名，也没有从现有的表映射类继承。

Answer 1

您可以通过设置__abstract__ = True 来修复错误。我修改了你的例子：

from flask import Flask
from flask_sqlalchemy import SQLAlchemy

app = Flask(__name__)
db = SQLAlchemy(app)

class Foo(db.Model):
    __abstract__ = True
    bar = db.Column(db.String)

__abstract__ 在the documentation 中描述。或者，您可以使用mixin class。

需要更多代码，您还可以通过继承SQLAlchemy 并覆盖其make_declarative_base() 方法来覆盖声明性基类。示例：

from flask import Flask
from flask_sqlalchemy import SQLAlchemy, Model, _BoundDeclarativeMeta, _QueryProperty
from sqlalchemy.ext.declarative import declarative_base

app = Flask(__name__)

class MyModel(Model):
    ...

class MySQLAlchemy(SQLAlchemy):
    def make_declarative_base(self, metadata=None):
        base = declarative_base(cls=MyModel, name='MyModel',
                                metadata=metadata,
                                metaclass=_BoundDeclarativeMeta)
        base.query = _QueryProperty(self)
        return base

db = MySQLAlchemy(app)

然后继承db.Model“像往常一样”。