flask中数据库迁移

flask中数据库迁移

1.在flask开发过程中，需要修改数据库模型，而且还要在修改之后更新数据库，最直接的方式就是删除旧表，但这样会丢失数据。
2.更好的解决办法是使用数据迁移框架，他可以追踪数据库模式的变化，然后把变动应用到数据库中。
3.在Flask中可以使用flask-migrate扩展，来实现数据迁移。并且集成到flask-scrpit中，所有操作通过命令就能完成。
4.为导出数据库迁移命令，flask-migrate提供一个migrateCommand类，可以附加到flask-script的manager对象上。

database_migrate.py

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文件，就说明数据库迁移成功了。