Flask-SQLAlchemy 数据库管理与迁移实践-CSDN博客

背景简介

在软件开发过程中，数据库管理是不可或缺的一环。Flask框架通过Flask-SQLAlchemy扩展提供了与SQLAlchemy ORM的集成，极大地方便了开发者对数据库的操作。本篇文章将基于提供的章节内容，深入探讨如何在Flask中配置和管理数据库，以及如何进行数据库迁移。

Flask-SQLAlchemy 配置

在开发小型应用或原型时，SQLite提供了便捷的解决方案，因为它不依赖外部数据库服务器，且数据库文件易于存储和分发。在Flask应用中配置SQLite数据库，需要在配置文件中指定数据库的URI。

class Config(object):
    # ...
    SQLALCHEMY_DATABASE_URI = os.environ.get('DATABASE_URL') or \
        'sqlite:///' + os.path.join(basedir, 'app.db')
    SQLALCHEMY_TRACK_MODIFICATIONS = False

SQLALCHEMY_TRACK_MODIFICATIONS 设置为False可以关闭SQLAlchemy的跟踪修改功能，减少不必要的性能开销。

初始化Flask-SQLAlchemy和Flask-Migrate

在 app/__init__.py 文件中，需要初始化Flask-SQLAlchemy和Flask-Migrate扩展，并创建数据库和迁移引擎实例。

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

app = Flask(__name__)
app.config.from_object(Config)
db = SQLAlchemy(app)
migrate = Migrate(app, db)

from app import routes, models

数据库模型

数据库模型是定义数据库结构的核心。在模型中，每个字段都映射到数据库表中的列。例如，一个用户模型可能包含 id 、 username 、 email 和 password_hash 字段。

class User(db.Model):
    # ...
    password_hash = db.Column(db.String(128))

存储密码哈希而非明文密码是安全最佳实践。密码哈希通过哈希算法（如bcrypt）生成，即使数据库被泄露，攻击者也无法直接获取用户的原始密码。

创建迁移仓库

随着应用的增长，数据库结构可能需要变更，此时使用Flask-Migrate和Alembic框架来管理数据库迁移变得至关重要。迁移仓库保存了所有的迁移脚本，这些脚本记录了数据库模式的变更。

(venv) $ flask db init

该命令将创建一个包含迁移脚本的目录结构，并准备好进行后续的迁移操作。

第一次数据库迁移

迁移仓库创建后，就可以执行首次迁移来创建数据库模式了。这可以通过 flask db migrate 和 flask db upgrade 命令完成。首次迁移会将模型定义的数据库结构添加到迁移脚本中，并应用到实际的数据库。

(venv) $ flask db migrate -m "users table"
(venv) $ flask db upgrade

通过这些步骤，你将成功地将应用的数据库结构变更自动化，并且可以方便地回滚到之前的版本。

总结与启发

通过本章内容的学习，我们了解了如何在Flask应用中配置SQLite数据库，创建数据库模型，以及执行数据库迁移。这些实践提高了开发效率，同时也强化了应用的安全性。尤其在处理用户密码时，我们学会了如何通过哈希算法来保护用户数据。数据库迁移的自动化管理，也使得数据库结构的变更变得可追溯且易于维护。希望这些知识能为你的项目开发带来更多的便利和安全保障。