flask-sqlalchemy 绑定多个数据库并实现多数据库迁移

配置

配置文件如下：

...
SQLALCHEMY_DATABASE_URI = "postgresql://postgres:123456@localhost:5432/db_name"

SQLALCHEMY_BINDS = {
    'other_db': "postgresql://postgres:123456@localhost:5432/other_db_name"
}
...
1
2
3
4
5
6
7

表模型定义

需要指定__bind_key__，数据迁移自动选择SQLALCHEMY_BINDS的相应数据库，不指定默认使用SQLALCHEMY_DATABASE_URI

class Test(Base):
    __bind_key__ = "other_db"
    __tablename__ = "test"
    __table_args__ = ({"comment": "测试表"})

    name = db.Column(db.String(128), comment="名称")
1
2
3
4
5
6

数据库迁移

...
app = create_app()

migrate = Migrate(app, db)
...
1
2
3
4
5

初始化迁移时需要--multidb参数来支持多数据库迁移

flask db init --multidb -d /data/data/migrations
flask db migrate -d /data/data/migrations
flask db upgrade -d /data/data/migrations
1
2
3

注意： 生产环境需要保存/data/data/migrations来存储迁移历史版本