将Flask-migrate用于多个文件中的模型

时间:2019-06-16 21:35:57

标签: python python-3.x flask flask-sqlalchemy flask-migrate

当我有多个模型时,我对如何使用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对象用于迁移类实例化,那么如何添加Flasks的迁移功能:

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

在这种情况下,最好在__init__.py文件中定义单个sql炼金术对象并将其导入我的所有模型中吗?

1 个答案:

答案 0 :(得分:0)

您误读了所引用的文本。那是在谈论完全不同的东西。那是关于将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