對多個文件中的模型使用 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 對象進行遷移類實例化，我該如何添加 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命令行工具。

接下來，您也不需要創建管理器。 Manager對象是 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