Flask db 初始化|极客笔记

Flask db 初始化

在使用 Flask 构建 Web 应用程序时，经常会涉及到对数据库的操作，比如创建表、修改表结构、插入数据等。Flask 提供了 Flask-SQLAlchemy 扩展来方便地操作数据库，而 Flask-Migrate 扩展则可以帮助我们管理数据库迁移。

什么是数据库迁移

数据库迁移是指在应用程序已经上线运行的情况下，对数据库的结构进行修改的过程。这样可以保证已有数据不会丢失，并且可以方便地在开发、测试和生产环境之间进行数据库结构同步。

Flask-Migrate 的作用

Flask-Migrate 是 Flask-SQLAlchemy 扩展的一个组件，它提供了一个命令行工具，用来帮助我们管理数据库迁移。通过 Flask-Migrate，我们可以将数据库操作封装成迁移脚本，并简单地应用到数据库中。

初始化数据库迁移

在使用 Flask-Migrate 之前，我们需要先初始化数据库迁移。初始化的过程会在项目目录下创建一个 migrations 文件夹，并生成一些必要的配置文件。

首先，确保已经安装了 Flask-Migrate 和 Flask-SQLAlchemy 扩展。可以使用 pip 安装：

pip install Flask-Migrate Flask-SQLAlchemy

然后，在 Flask 应用程序的工程目录下执行以下命令来初始化数据库迁移：

flask db init

这样就完成了数据库迁移的初始化。接下来我们就可以通过 Flask-Migrate 来管理数据库迁移了。

示例代码

假设我们有一个简单的 Flask 应用程序，其中定义了一个名为 User 的数据模型，表示用户信息。我们现在来演示如何使用 Flask-Migrate 来初始化数据库迁移。

首先，在 Flask 应用程序的入口文件中（比如 app.py）添加以下代码：

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

app = Flask(__name__)
app.config['SQLALCHEMY_DATABASE_URI'] = 'sqlite:///app.db'
app.config['SQLALCHEMY_TRACK_MODIFICATIONS'] = False

db = SQLAlchemy(app)
migrate = Migrate(app, db)

class User(db.Model):
    id = db.Column(db.Integer, primary_key=True)
    username = db.Column(db.String(50), nullable=False)

@app.route('/')
def index():
    return 'Hello, Flask!'

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

然后在项目目录下执行以下命令初始化数据库迁移：

flask db init

运行上面的命令后，会在项目目录下生成一个名为 migrations 的文件夹，里面包含一些用于管理数据库迁移的文件。

运行结果

执行完 flask db init 命令后，项目目录下会多出一个名为 migrations 的文件夹。该文件夹包含一些用于管理数据库迁移的文件，具体结构如下：

migrations/
    ├── versions/
    │     ├── __init__.py
    └── __init__.py

在 migrations 目录下的 init.py 文件中，会包含一些用于配置数据库迁移的代码。而 versions 目录下的 init.py 文件为空，其中会存放每个数据库迁移的版本文件。

这样就完成了数据库迁移的初始化，接下来可以使用 Flask-Migrate 来管理数据库结构的变更和迁移了。

总结

通过本文的介绍，我们了解了什么是数据库迁移以及 Flask-Migrate 的作用。并演示了如何使用 Flask-Migrate 初始化数据库迁移。

Flask db 初始化