Flask扩展之 【Flask-Migrate】数据库的迁移

数据库迁移：

• 在开发过程中，需要修改数据库模型，而且还要在修改之后更新数据库。最直接的方式就是删除旧表，但这样会丢失数据。
• 更好的解决办法是使用数据库迁移框架，它可以追踪数据库模式的变化，然后把变动应用到数据库中。

数据库的迁移操作，是基于Flask-SQLAlchemy的，相关操作可参考：
Flask-SQLAlchemy ，在flask中通过ORM方式来操作数据库

---

1.安装： 扩展的万年不变法则 当然，第一步还是需要在安装了flask框架的虚拟环境中，安装扩展：

pip install flask-migrate

2.导入扩展中的包：

咱们这里就是Migrate 和 MigrateCommand 用的多，所以就先导入这两个，不够再导呗。

from flask_migrate import migrate,MigrateCommand

3.使用迁移类将应用（app）和数据库对象（db）保存起来：

Migrate(app, db)

4.创建终端命令的对象：

由于数据库的迁移，是需要在终端上进行命令操作的，所以，在这里，我们还需要利用flask-script扩展的相关使用：
Flask-script–通过命令行运行文件的插件

from flask.ext.script import Manager

manager = Manager(app)

5.将数据库的迁移命令添加到manager中
MigrateCommand是迁移的命令，manager是Flask-Script的实例，这条语句在flask-Script中添加一个db命令

manager.add_command('db', MigrateCommand)

注意：
这里的db是迁移命令，和之前的【db = SQLAlchemy(app)】中的数据库对象db不一样哦。

---

好的，现在开始数据库的迁移的相关操作了：

①创建迁移仓库：

这个命令会创建migrations文件夹，所有迁移文件都放在里面。

在pycharm的终端里面执行这段代码哦：

python database.py db init

在执行这段代码后，我们的 database.py 的相同等级目录下就会出现migrations这个文件夹，然后，咱们的所有迁移文件都会保存在versions里面：如下图

②创建迁移脚本：

就是创建存放在versions里面的文件：

python database.py db migrate -m 'initial migration'

-m 是注释，后面的内容写本次迁移修改的内容
下面versions里面是增加了一个迁移文件吧…

③更新到数据库：

迁移文件生成之后，我们就需要提交到数据库了吧：

python database.py db upgrade

④返回以前的版本：

当然，咱们在这个数据库迁移是很人性化的，如果你修改迁移之后后悔了，是有后悔药可以吃的：

首先，先看看你想迁移了哪些版本：

python datanase.py db history

输出格式： -> 版本号 (head), initial migration

回滚到指定版本：

python database.py db downgrade 版本号

代码：

from flask import Flask
from flask.ext.script import Manager
from flask_sqlalchemy import SQLAlchemy
from flask_migrate import Migrate, MigrateCommand

app = Flask(__name__)
app.config['SQLALCHEMY_DATABASE_URI'] = 'mysql://xiaomin:xiaomin123@127.0.0.1:8888/xiaominsql'
app.config['SQLALCHEMY_TRACK_MODIFICATIONS'] = True

db = SQLAlchemy(app)

Migrate(app, db)
manager = Manager(app)
manager.add_command('db', MigrateCommand)


@app.route('/')
def index():
    return 'hello world'


class Role(db.Model):
    id = db.Column(db.Integer, primary_key=True)
    title = db.Column(db.String(64))
    category = db.Column(db.String(64))


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

---

实际应用的操作步骤，参考如下：

1.python 文件 db init
2.python 文件 db migrate -m"版本名(注释)"
3.python 文件 db upgrade 然后观察表结构
4.根据需求修改模型
5.python 文件 db migrate -m"新版本名(注释)"
6.python 文件 db upgrade 然后观察表结构
7.若返回版本,则利用 python 文件 db history查看版本号
8.python 文件 db downgrade(upgrade) 版本号