Flask 数据库迁移详情

Flask 数据库迁移是将数据库中的表结构变化应用到现有数据库上的过程。在 Flask 应用中，通常使用 Flask-Migrate 扩展来实现数据迁移。下面是实现 Flask 数据库迁移的完整攻略。

准备工作

在使用 Flask-Migrate 扩展进行数据库迁移之前，需要先安装好 Flask 和 Flask-Migrate 扩展。

1. 安装 Flask

可以通过 pip 包管理器来安装 Flask，命令如下：

pip install Flask

1. 安装 Flask-Migrate

可以通过 pip 包管理器来安装 Flask-Migrate，命令如下：

pip install Flask-Migrate

数据库配置

在 Flask 应用中使用数据库需要进行数据库配置。下面是一个 SQLite 数据库的配置示例：

from flask import Flask
from flask_sqlalchemy import SQLAlchemy
from flask_migrate import Migrate

app = Flask(__name__)

app.config['SQLALCHEMY_DATABASE_URI'] = 'sqlite:///example.db'
app.config['SQLALCHEMY_TRACK_MODIFICATIONS'] = False
db = SQLAlchemy(app)
migrate = Migrate(app, db)

在上面的代码中：

• app.config['SQLALCHEMY_DATABASE_URI'] 配置了 SQLite 数据库的链接地址；
• app.config['SQLALCHEMY_TRACK_MODIFICATIONS'] 配置了数据库模型的修改情况是否追踪，该项配置可以提高程序性能；

创建数据库模型

创建好数据库配置之后，需要创建数据库模型，包括表和字段。下面是一个用户表模型的示例：

from app import db

class User(db.Model):
    id = db.Column(db.Integer, primary_key=True)
    name = db.Column(db.String(255))

在上面的代码中，定义了 User 模型类，使用 db.Column 来定义字段，可以指定字段类型、长度、是否为主键等信息。

数据库迁移

在进行数据库迁移之前，需要首先创建一个迁移仓库。通过命令 flask db init 可以创建一个迁移仓库。下面是详细的步骤：

1. 当前目录进入到项目根目录

cd /path/to/your/project

1. 初始化 Flask-Migrate 扩展

flask db init

1. 在初始化的过程中，会生成一个名为 migrations 的目录，用于存放数据迁移相关的文件。

在创建好迁移仓库之后，就可以进行数据迁移了。下面是详细的步骤：

1. 生成数据迁移脚本

flask db migrate -m "add user table"

通过 flask db migrate 命令生成一个数据迁移脚本，-m 选项用于指定迁移描述信息。

1. 应用数据迁移

flask db upgrade

通过 flask db upgrade 命令应用数据迁移脚本。

示例说明

下面简单说明两个数据迁移的示例。

示例一：添加字段

在已有的用户表中添加一个年龄字段 age。

1. 在 User 模型类中添加 age 字段

python
class User(db.Model):
# ...
age = db.Column(db.Integer)

1. 生成数据迁移脚本

flask db migrate -m "add age column"

在生成的数据迁移脚本中，会自动生成添加 age 字段的代码。

1. 应用数据迁移

flask db upgrade

应用数据迁移脚本。

示例二：删除字段

在已有的用户表中删除 name 字段。

1. 在 User 模型类中删除 name 字段

python
class User(db.Model):
# ...
id = db.Column(db.Integer, primary_key=True)

1. 生成数据迁移脚本

flask db migrate -m "remove name column"

在生成的数据迁移脚本中，会自动生成删除 name 字段的代码。

1. 应用数据迁移

flask db upgrade

应用数据迁移脚本。

通过上述示例可以看出，使用 Flask-Migrate 扩展进行数据迁移非常简单。只需要定义好数据库模型，然后执行几个简单的命令即可完成数据迁移。