对多个文件中的模型使用 Flask-migrate

Question

当我有多个模型时，我对如何使用 Flask-Migrate 感到困惑。基本上我的 Flask 应用程序如下所示：

app
??? __init__.py
??? config.py
??? manage.py
??? migrations
??? models
?   ??? model1.py
?   ??? model2.py
??? resources
??? run.py
??? tests

我已经读过，对于每个模型，最好db = SQLAlchemy()在文件中创建对象，然后将此 db 对象导入应用程序，__init__.py如下所示：

from models.model1 import db
db.init_app(app)
from models.model2 import db
db.init_app(app)

但是，如果我对多个模型文件执行此操作，考虑到我只能使用 1 个 sql alchemy 对象进行 migrate 类实例化，我该如何添加 Flasks 的迁移功能：

migrate = Migrate(app, db)
manager = Manager(app)
manager.add_command('db', MigrateCommand)

在这种情况下，最好在__init__.py文件中定义一个 sql alchemy 对象并将其导入我的所有模型中吗？

Answer 1

您误读了引用的文本。那谈论的是完全不同的事情。这谈到将您的db对象与对象分开app（并将两者在create_app工厂函数中联系在一起）。创建多个db对象只会使您的问题复杂化。

所有需要的是一个单一的 db = SQLAlchemy()对象，需要进口的所有定义模型的文件。通常是通过您直接或间接进行的create_app工厂函数，你需要调用create_app()反正功能，能够运行flask db命令行工具呢。

接下来，你就不会需要创建一个管理者要么。该Managar对象是 Flask 项目添加对脚本本身的支持之前的一个保留。如果您使用的瓶0.12或更高版本，你不要想用瓶脚本和它的管理者是。

因此，在您的 中__init_.py，您只需要：

from flask import Flask
from flask_sqlalchemy import SQLAlchemy
from flask_migrate import Migrate


db = SQLAlchemy()


def create_app(test_config=None):
    app = Flask(__name__)

    app.config.from_object(f"{__name__}.config")
    app.config.from_envvar("PROJECTNAME_SETTINGS", silent=True)
    if test_config:
        app.config.from_mapping(test_config)

    db.init_app(app)
    Migrate(app, db)

    # importing the models to make sure they are known to Flask-Migrate
    from models import models1, models2

    # any other registrations; blueprints, template utilities, commands

    return app