简单分析Flask 数据库迁移详情

共计 1737 个字符，预计需要花费 5 分钟才能阅读完成。

db.create_all() 不会重新创建表或是更新表，需要先使用 db.drop_all() 删除数据库中所有的表之后再调用 db.create_all() 才能重新创建表，但是这样的话，原来表中的数据就都被删除了，这肯定是不行的，这时就出现了数据库迁移的概念。

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

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

from flask_migrate import Migrate
 
from flask_sqlalchemy import SQLAlchemy
 
app = Flask(__name__)
 
···
 
db = SQLAlchemy(app)
 
migrate = Migrate(app, db)

实例化 Migrate 类，需要传入 Flask 实例 app 和 SQLAlchemy 创建的实例 db。

先定义 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)

在开始迁移数据之前，需要先使用下面的命令创建一个迁移环境：

flask db init

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

使用如下命令自动生成迁移脚本：

flask db migrate -m "create_table"

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

迁移脚本内有两个函数：

upgrade()：把迁移中的改动应用到数据库中

downgrade()：将改动撤销

自动生成的迁移脚本会根据模型定义和数据库当前状态的差异，生成 upgrade() 和 downgrade() 函数的内容，不一定完全正确，有必要再进行检查一下。

生成了迁移脚本后，使用 flask db upgrade 命令可完成对数据库的更新。执行后即可生成数据库及表。

如果之后我们需要改动 user 表中的字段，比如添加一个 mobile 字段，我们只需在 User 模型类中添加该属性，之后执行 flask db migrate -m ‘ 注释 ’ 和 flask db upgrade 命令即可。

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)
 
    mobile = db.Column(db.String)

如果想要回滚迁移的话，可以执行 flask db downgrade 命令。

这里只是介绍如何在 Flask 中进行数据库迁移，关于在生产环境下，是否需要使用迁移工具或者使用何种工具进行迁移，这里不做讨论，至于我的话，在生产环境中，我没有使用过 Flask-Migrate，而是选择编写 SQL 脚本来处理数据库及表的更新或改动，我觉得这样更不容易出错，其实各有各的好处，看自己选择。

阿里云 2 核 2G 服务器 3M 带宽 61 元 1 年，有高配

腾讯云新客低至 82 元 / 年，老客户 99 元 / 年

代金券：在阿里云专用满减优惠券