时间戳

首页 > 解决方案 > 对多个文件中的模型使用 Flask-migrate

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

问题描述

当我有多个模型时，我对如何使用 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)

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

标签： pythonpython-3.xflaskflask-sqlalchemyflask-migrate

解决方案

您误读了引用的文本。那是在谈论完全不同的事情。这谈到了将您的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

推荐阅读