首页 > 代码库 > 使用Flask-Migrate进行管理数据库升级

使用Flask-Migrate进行管理数据库升级

我们在升级系统的时候,经常碰到需要更新服务器端数据结构等操作,之前的方式是通过手工编写alter sql脚本处理,经常会发现遗漏,导致程序发布到服务器上后无法正常使用。

现在我们可以使用Flask-Migrate插件来解决之,Flask-Migrate插件是基于Alembic,Alembic是由大名鼎鼎的SQLAlchemy作者开发数据迁移工具。

具体操作如下:

1. 安装Flask-Migrate插件

$ pip install Flask-Migrate

2. 修改Flask App部分的代码,以增加Migrate相关的Command

db = SQLAlchemy(app)
migrate = Migrate(app, db)

manager = Manager(app)
manager.add_command(db‘, MigrateCommand)

3. 初始化

$ python app.py db init

4. 数据迁移,自动创建迁移代码

$ python app.py db migrate

5. 更新数据库

$ python app.py db upgrade

后面如果有Model变更的话,在开发环境下只需要重复执行第4、5步即可。

而在服务器端只需要执行第5步即可实现数据库的迁移工作。

使用Flask-Migrate进行管理数据库升级