Flask-Migrate
这个扩展对Alembic做了轻量级包装,并集成到Flask-Script中,所有操作都通过Flask-Script命令完成。
-
安装
pip install flask-migrate
-
修改Flask App部分的代码,以增加Migrate相关的Command
from flask.ext.script import Manager
from flask.ext.migrate import Migrate, MigrateCommand
# 设置配置项
app = Flask(__name__)
# 配置你自己的数据库连接字符串
app.config['SQLALCHEMY_DATABASE_URI'] = 'sqlite://./db/app.db'
db = SQLAlchemy(app)
migrate = Migrate(app, db)
# 导入所有模型
from app.models import *
migrate = Migrate()
# 可指定migrations目录
migrate.init_app(app, db, directory='./migrations')
manager = Manager(app)
manager.add_command('db', MigrateCommand)
if __name__ == '__main__':
manager.run()
3.初始化(venv) python app.py db init
4.数据迁移,自动创建迁移代码(venv) python app.py db migrate
在migrations目录下的versions中查看和修改。
5.更新数据库(venv) python app.py db upgrade
注:
- 如果upgrade时与现有数据库发生冲突,可以用
python app.py db edit
来调用编辑器修改生成的脚本来解决冲突。 - 如果有Model再改动,开发环境下重复执行第4、5步即可。服务器端执行第5步即可实现数据库的迁移工作。