Flask数据库迁移简单介绍

前言

熟悉Django的开发者都知道，Django内置了ORM框架，功能丰富且集成了数据库迁移等特性。所谓ORM（对象关系映射），其核心在于简化SQL语句的编写，将关系型数据库中的表映射为Python类。这种方式显著降低了学习门槛，即使不熟悉SQL的程序员也能顺畅操作数据库，同时天然具备防止SQL注入的安全优势。在Flask中，我们可以使用flask-sqlalchemy包来实现这一功能。具体的使用方法官方文档已有清晰说明，读者可直接参考文档上手，本文重点在于讲解如何进行数据库迁移。

flask的小工具

从Django转向Flask时，最直观的差异在于服务启动方式：Flask可以直接运行代码启动服务，而Django通常需要使用runserver命令。至于哪种方式更优，取决于具体场景。这里推荐一个插件：flask-script，通过pip安装即可使用。

from flask_script import Manager
......
manager = Manager(app)
......
#app.run() #注释这句
manager.run()

使用该插件后，启动服务的命令变为：python main.py runserver

这看似多余，实则与下文要介绍的数据库迁移紧密相关。

数据库迁移

我们需要使用的插件是flask-migrate，安装该插件即可。

main.py

from flask import Flask
from flask_script import Manager
from flask_sqlalchemy import SQLAlchemy
from flask_migrate import Migrate,MigrateCommand

app = Flask(__name__)
app.config['SQLALCHEMY_DATABASE_URI'] = 'sqlite:////opt/db.sqlite' #使用sqlite3数据库
db = SQLAlchemy(app) #ORM

migrate = Migrate(app,db)
manager = Manager(app)
manager.add_comman('db',MigrateCommand) #添加db 命令（runserver的用法）

#一个用户表
class User(db.Model):
...
...
省略

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

至此，准备工作已完成。运行命令python main.py db init以创建数据表，这会在项目目录下生成migrations/目录，用于保存数据库每次变更的历史记录。

修改User类定义。

执行python main.py db migrate提交修改，随后运行python main.py db upgrade执行这些修改。

此时查看User表，结构已发生变化。

若需回退修改，可执行python main.py db downgrade。

需要注意一点：SQLite3无法删除包含数据的列，因此建议避免执行删除列的操作。

以上就是本文的全部内容，希望对大家的学习有所帮助，也希望大家多多支持本站。