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

Question

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

app
├── __init__.py
├── config.py
├── manage.py
├── migrations
├── models
│   ├── model1.py
│   ├── model2.py
├── resources
├── run.py
└── tests

我有 read ，对于每个模型，最好在文件中创建 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 项目添加对脚本本身的支持之前的一个保留。如果您使用 Flask 0.12 或更新版本，您 不希望 使用 Flask-Script 及其管理器。

因此，在 __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