Flask-migrate 升级前更改数据库

Question

我设置了一个多租户结构，其中每个客户端都为他们设置了一个架构。该结构反映了“父”架构，因此发生的任何迁移都需要以相同的方式针对每个架构进行。

我正在使用 Flask-Script 和 Flask-Migrate 来处理迁移。

到目前为止我尝试的是遍历我的架构名称，为它们构建一个 URI，使用从 URI 生成的引擎确定一个新的 db.session 范围，最后运行来自flask_migrate.

    @manager.command
    def upgrade_all_clients():
        clients = clients_model.query.all()
        for c in clients:
            application.extensions["migrate"].migrate.db.session.close_all()
            application.extensions["migrate"].migrate.db.session = db.create_scoped_session(
                options={
                    "bind": create_engine(generateURIForSchema(c.subdomain)),
                    "binds": {},
                }
            )
            upgrade()
        return

我不完全确定为什么这不起作用，但结果是它只运行应用程序启动时设置的数据库的迁移。

我的理论是我没有更改最初在管理器脚本运行时设置的会话。

有没有更好的方法可以在不设置多个绑定和使用 --multidb 参数的情况下迁移这些模式中的每一个？我不认为我可以在配置中使用 SQLALCHEMY_BINDS 因为这些模式需要能够动态 created/destroyed.

Answer 1

对于那些遇到同样问题的人，针对我的具体情况的答案非常简单。

@manager.command
def upgrade_all_clients():
    clients = clients_model.query.all()
    for c in clients:
        print("Upgrading client '{}'...".format(c.subdomain))
        db.engine.url.database = c.subdomain
        _upgrade()
    return

db.engine.url 的数据库属性是针对架构的。我不知道这是否是解决此问题的最佳方法，但它确实有效，而且我可以单独迁移每个架构。

Flask-migrate 升级前更改数据库

Flask-migrate change db before upgrade

flask-sqlalchemy

alembic

flask-migrate

flask-script