Flask 数据迁移

在开发过程中，有时候需要修改数据库模型，比如新功能需要增加一个字段，在 Flask 代码中修改模型类后，要将新增的字段同步到数据库中。这时候是不能删表重建的。

在 Flask 中，可以使用数据库迁移来解决这个问题，数据库迁移可以追踪数据模型类的变化，然后把变动应用到数据库中，不会删表造成数据丢失。

使用 Flask-Migrate 实现数据库迁移

在开发过程中，随着需求的变化，有可能需要添加或修改表的一些字段，但是原表中的数据不能删除，此时就需要创建新表，并将旧表中的数据迁移至新表中，Flask-Migrate这个扩展就可以在不破坏数据的情况下更新数据库表的结构，并完成数据从旧表到新表的迁移。

import pymysql
from flask import Flask
from flask_migrate import Migrate,MigrateCommand
from flask_script import Manager
from flask_sqlalchemy import SQLAlchemy
app = Flask(__name__)
manager = Manager(app)

app.config['SQLALCHEMY_DATABASE_URI'] = 'mysql://root:mysql@127.0.0.1:3306/Flask_test'
app.config['SQLALCHEMY_COMMIT_ON_TEARDOWN'] = True
app.config['SQLALCHEMY_TRACK_MODIFICATIONS'] = True

db = SQLAlchemy(app)
pymysql.install_as_MySQLdb()    # TODO  解决：Mysqldb 不兼容 python3.5 以后的版本
# 第一个参数是flask的实现，第二个参数是sqlalchemy数据库实例
migrate = Migrate(app,db)

# manager是flask-scriptde 实例，这条语句在flask-script中添加一个db命令
manager.add_command('db',MigrateCommand)

# 定义模型Role
class Role(db.Model):
    # 定义表名
    __tablename__ = 'roles'
    # 定义列对象
    id = db.Column(db.Integer,primary_key=True)
    name = db.Column(db.String(64),unique=True)
    user = db.relationship('User',backref='role')

    # repr() 方法显示一个可读字符串
    def __repr__(self):
        return 'Role:'.format(self.name)

# 定义用户
class User(db.Model):
    __tablename__ = 'users'
    id = db.Column(db.Integer,primary_key=True)
    username = db.Column(db.String(64),unique=True,index=True)

    # 设置外键
    role_id = db.Column(db.Integer,db.ForeignKey('roles.id'))

    def __repr__(self):
        return 'User:'.format(self.username)


if __name__ == '__main__':
    manager.run()

1.在终端 切换到当前目录下，
2.执行 python database_migrate.py db init 命令
3.在当前目录下生成migrations文件，就说明数据库迁移成功了。