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

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

2、Flask-Migrate的使用

可以使用pip install flask-migrate进行安装。在程序中,我们实例化 Flask_Migrate 提供的 Migrate 类,进行初始化操作。

mange.py

class Config:
    DEBUG = True
    # 数据库连接配置
    SQLALCHEMY_DATABASE_URI = 'mysql://root:123456@localhost:3306/demo1'
    SQLALCHEMY_TRACK_MODIFICATIONS = False

demo4.py

from flask import Flask
from flask_sqlalchemy import SQLAlchemy
from mange import Config
from flask_script import Manager
from flask_migrate import Migrate, MigrateCommand



app = Flask(__name__)
app.config.from_object(Config)

manager = Manager(app)

db = SQLAlchemy(app)

Migrate(app, db)

manager.add_command('mysql', MigrateCommand)


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

实例化 Migrate 类,需要传入 Flask 实例 app 和 SQLAlchemy创建的实例 mysql

2.1 数据库的迁移过程

先定义一个模型类(User)

class User(db.Model):

    __tablename__ = 'user'

    id = db.Column(db.Integer, primary_key=True)

    user_name = db.Column(db.String)

    password = db.Column(db.String)

在 Pycharm自带的控制台输入命令

flask使用orm操作mysql数据库不迁移数据库 flask 数据库迁移_python

依次输入

1、python demo4.py mysql init

demo4.py是文件名

迁移环境只需创建一次,创建后会在项目根目录下生成一个 migrations 目录,其中包含了自动生成的配置文件和迁移版本目录。

flask使用orm操作mysql数据库不迁移数据库 flask 数据库迁移_flask_02

2、python demo4.py mysql migrate -m “init”

-m 选项添加备注信息,执行后迁移版本目录生成了迁移脚本。 此备注信息必须添加

3、python demo4.py mysql upgrade

迁移脚本内有两个函数:
upgrade():把迁移中的改动应用到数据库中
downgrade():将改动撤销

2.1.3 更新数据库

如果之后我们需要改动 user 表中的字段,比如添加一个mobile字段,我们只需在 User 模型类中添加该属性,之后分别执行上方的 ② 号命令和 ③ 号命令即可

注意:

  1. 如果在迁徙数据时出现报错,请一定要把数据库清空,然后重新开始执行命令
  2. 此文件使用的是flask_script 包中的 Manager 来启动文件的。所以,想运行代码的话输入
python demo4.py runserver -h 0.0.0.0 -p 8000 -d

demo4.py是文件名

-h ip地址
-p 端口号
-d debug

  1. 当然如果你觉得上方的②号方法较麻烦,可以如下图配置,配置完后就可以正常的右键运行了:

flask使用orm操作mysql数据库不迁移数据库 flask 数据库迁移_python_03

flask使用orm操作mysql数据库不迁移数据库 flask 数据库迁移_数据库_04