在开发过程中,有时需要 修改数据库模型,修改之后还需要更新数据库。使用SQLAlchmey更新数据库表,只能删除旧表,新建新表,数据也会丢失。
因此,SQLAlchemy编写了一个迁移框架:Alembic,除了直接使用Alembic外。还可以使用Flask-Migrate集成到Flask-Scriot中,所有操作都通过Flask-Script命令完成。
一、将MigrateCommand类添加到manager对象上
# manager.py
import os
from app import create_app, db
from app.models import User,Role
from flask_script import Manager,Shell
from flask_migrate import Migrate, MigrateCommand
app = create_app(os.getenv('FLASK_CONFIG') or 'default')
manager = Manager(app)
migrate = Migrate(app,db) #
def make_shell_context():
return dict(app=app, db=db, User=User, Role=Role)
manager.add_command('shell',Shell(make_context=make_shell_context)) #为python manager.py 添加启动选项
manager.add_command('db',MigrateCommand) #添加命