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

冷不防 2022-05-13 14:22 332阅读 0赞

数据库迁移：

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

数据库的迁移操作，是基于Flask-SQLAlchemy的，相关操作可参考：  
[Flask-SQLAlchemy ，在flask中通过ORM方式来操作数据库][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–通过命令行运行文件的插件][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里面：如下图  
![在这里插入图片描述][70]

②创建迁移脚本：

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

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

**-m** 是注释，后面的内容写本次迁移修改的内容  
下面versions里面是增加了一个迁移文件吧…  
![在这里插入图片描述][70 1]

③更新到数据库：

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

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) 版本号

[Flask-SQLAlchemy _flask_ORM]: https://blog.csdn.net/gymaisyl/article/details/83348296
[Flask-script]: https://blog.csdn.net/gymaisyl/article/details/83063324
[database.py]: http://database.py
[70]: /images/20220504/d4092d74edfa47d19782fb854c9f1151.png
[70 1]: /images/20220504/a9a621c84ec44473a4188ed105c08fa6.png