我们在升级系统的时候,经常碰到需要更新服务器端数据结构等操作,之前的方式是通过手工编写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步即可实现数据库的迁移工作。