flask 如何创建数据库表

Flask 创建数据库表的方法包括使用 SQLAlchemy、Flask-Migrate 等工具。 其中，使用 SQLAlchemy 是一种常见且强大的方式，它能够方便地与 Flask 集成，并提供 ORM（对象关系映射）功能，使我们可以通过类和对象的形式来操作数据库。本文将详细介绍如何使用 SQLAlchemy 在 Flask 中创建数据库表，并结合 Flask-Migrate 实现数据库迁移。

一、安装和配置

1. 安装 Flask 和 SQLAlchemy

首先，我们需要安装 Flask 和 SQLAlchemy。你可以使用 pip 进行安装：

pip install flask pip install flask-sqlalchemy pip install flask-migrate

2. 配置应用

接下来，我们需要配置 Flask 应用，使其能够与 SQLAlchemy 一起工作。创建一个名为 app.py 的文件，并添加以下内容：

from flask import Flask
from flask_sqlalchemy import SQLAlchemy
app = Flask(__name__)
app.config['SQLALCHEMY_DATABASE_URI'] = 'sqlite:///site.db'
app.config['SQLALCHEMY_TRACK_MODIFICATIONS'] = False
db = SQLAlchemy(app)

在上面的代码中，我们创建了一个 Flask 应用，并配置了数据库 URI，这里使用的是 SQLite 数据库。SQLALCHEMY_TRACK_MODIFICATIONS 用于关闭对象修改跟踪，以提高性能。

二、定义数据库模型

1. 创建模型类

在 SQLAlchemy 中，我们通过定义模型类来创建数据库表。每个模型类代表数据库中的一个表。我们可以在 app.py 中添加以下代码：

class User(db.Model):
    id = db.Column(db.Integer, primary_key=True)
    username = db.Column(db.String(20), unique=True, nullable=False)
    email = db.Column(db.String(120), unique=True, nullable=False)
    image_file = db.Column(db.String(20), nullable=False, default='default.jpg')
    password = db.Column(db.String(60), nullable=False)
    def __repr__(self):
        return f"User('{self.username}', '{self.email}', '{self.image_file}')"

在上面的代码中，我们定义了一个 User 模型类，包含四个字段：id、username、email、image_file 和 password。每个字段都是一个 db.Column 对象，指定了数据类型和一些约束条件。

2. 创建数据库

为了在数据库中创建表，我们需要在 Python shell 中运行一些命令。首先，确保已经启动了 Flask 应用，然后进入 Python shell：

python

在 Python shell 中，导入 db 和我们的模型类，并创建数据库表：

from app import db
db.create_all()

运行以上命令后，SQLAlchemy 将会在数据库中创建 User 表。

三、使用 Flask-Migrate 进行数据库迁移

1. 初始化 Flask-Migrate

Flask-Migrate 是一个用于处理数据库迁移的工具。它基于 Alembic，可以方便地进行数据库模式的更改。首先，我们需要初始化 Flask-Migrate。在 app.py 中添加以下代码：

from flask_migrate import Migrate
migrate = Migrate(app, db)

然后，在项目根目录下初始化迁移环境：

flask db init

2. 创建迁移脚本

每当我们对模型类进行修改时，都需要创建一个新的迁移脚本。运行以下命令：

flask db migrate -m "Initial migration."

3. 应用迁移

最后，运行以下命令将迁移应用到数据库：

flask db upgrade

通过以上步骤，我们就成功地使用 Flask-Migrate 进行了数据库迁移。

四、在 Flask 中操作数据库

1. 插入数据

我们可以在 Flask 应用中插入数据。创建一个新的 Python 脚本 insert_data.py，并添加以下代码：

from app import db
from app import User
user = User(username='JohnDoe', email='john@example.com', password='password')
db.session.add(user)
db.session.commit()

运行这个脚本后，一个新的用户记录将被插入到 User 表中。

2. 查询数据

我们可以在 Flask 应用中查询数据。在 app.py 中添加一个路由，用于显示所有用户：

@app.route('/users')
def users():
    users = User.query.all()
    return '<br>'.join([f"{user.username} - {user.email}" for user in users])

启动 Flask 应用，并访问 /users 路径，将会显示所有用户的用户名和电子邮件。

五、总结

在本文中，我们详细介绍了如何在 Flask 中使用 SQLAlchemy 创建数据库表，并结合 Flask-Migrate 实现数据库迁移。我们从安装和配置开始，定义了数据库模型类，并通过命令行工具创建和迁移数据库。最后，我们展示了如何在 Flask 应用中插入和查询数据。

通过掌握这些知识，你可以在 Flask 项目中高效地管理数据库，确保数据的一致性和完整性。如果你正在寻找一款项目管理系统，可以考虑使用 研发项目管理系统PingCode 或 通用项目协作软件Worktile，它们能够帮助你更好地管理项目和团队。