对多个文件中的模型使用 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() 对象，然后将此数据库对象导入到应用程序的 __init__.py 中，如下所示：

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

但是，如果我对多个模型文件执行此操作，考虑到我只能使用 1 个 sql alchemy 对象进行迁移 class 实例化，我该如何添加 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 command-line 工具 无论如何.

接下来，您不需要创建一个管理器或者。 Managar 对象是 Flask 项目添加对脚本本身的支持之前的 hold-over。如果您使用的是 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

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

Using Flask-migrate for models in multiple files

python

python-3.x

flask

flask-sqlalchemy

flask-migrate