flask 数据迁移

本文介绍了如何在Flask应用中使用Flask-Migrate进行数据库迁移。首先通过安装Flask-Migrate和初始化迁移仓库来设置基础。接着,讲解了如何创建迁移脚本,包括手动和自动创建,并强调了检查和手动调整的重要性。最后,阐述了如何使用`db upgrade`命令更新数据库,确保改动正确无误地应用到数据库中。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

一.创建迁移仓库

首先,我们要在虚拟环境中安装Flask-Migrate:

pip install flask-migrate

这个扩展的初始化方法如下所示:

from flask.ext.migrate import Migrate, MigrateCommand
migrate = Migrate(app, db)
manager.add_command('db', MigrateCommand)

为了导出数据库迁移命令,Flask-Migrate 提供了一个MigrateCommand 类,可附加到Flask-Script 的manager 对象上。在这个例子中,MigrateCommand 类使用db 命令附加。

在维护数据库迁移之前,要使用init 子命令创建迁移仓库:

复制代码

python hello.py db init
Creating directory /home/flask/flask1/migrations ... done
  Creating directory /home/flask/flask1/migrations/versions ... done
  Generating /home/flask/flask1/migrations/env.pyc ... done
  Generating /home/flask/flask1/migrations/alembic.ini ... done
  Generating /home/flask/flask1/migrations/README ... done
  Generating /home/flask/flask1/migrations/script.py.mako ... done
  Generating /home/flask/flask1/migrations/env.py ... done
  Please edit configuration/connection/logging settings in
  '/home/flask/flask1/migrations/alembic.ini' before proceeding.

复制代码
  这个命令会创建migrations 文件夹,所有迁移脚本都存放其中。数据库迁移仓库中的文件要和程序的其他文件一起纳入版本控制。

二.创建迁移脚本

在Alembic 中,数据库迁移用迁移脚本表示。脚本中有两个函数,分别是upgrade() 和downgrade()。upgrade() 函数把迁移中的改动应用到数据库中,downgrade() 函数则将改动删除。Alembic 具有添加和删除改动的能力,因此数据库可重设到修改历史的任意一点。

我们可以使用revision 命令手动创建Alembic 迁移,也可使用migrate 命令自动创建。手动创建的迁移只是一个骨架,upgrade() 和downgrade() 函数都是空的,开发者要使用Alembic 提供的Operations 对象指令实现具体操作。自动创建的迁移会根据模型定义和数据库当前状态之间的差异生成upgrade() 和downgrade() 函数的内容。自动创建的迁移不一定总是正确的,有可能会漏掉一些细节。自动生成迁移脚本后一定要进行检查。

migrate 子命令用来自动创建迁移脚本:

复制代码

 python hello.py db migrate -m "initial migration"
INFO  [alembic.runtime.migration] Context impl MySQLImpl.
INFO  [alembic.runtime.migration] Will assume non-transactional DDL.
INFO  [alembic.autogenerate.compare] Detected removed table u'sys_user'
INFO  [alembic.autogenerate.compare] Detected removed table u'sys_role_privilege'
INFO  [alembic.autogenerate.compare] Detected removed table u'sys_role'
INFO  [alembic.autogenerate.compare] Detected removed table u'sys_privilege'
INFO  [alembic.autogenerate.compare] Detected removed table u'sys_dict'
INFO  [alembic.autogenerate.compare] Detected removed table u'user info'
INFO  [alembic.autogenerate.compare] Detected removed table u'country'
INFO  [alembic.autogenerate.compare] Detected removed table u'sys_user_role'
  Generating
  /home/flask/flask1/migrations/versions/f52784fdd592_initial_migration.py ... done

复制代码
三.更新数据库

检查并修正好迁移脚本之后,我们可以使用db upgrade 命令把迁移应用到数据库中:

python hello.py db upgrade
INFO  [alembic.runtime.migration] Context impl MySQLImpl.
INFO  [alembic.runtime.migration] Will assume non-transactional DDL.
INFO  [alembic.runtime.migration] Running upgrade  -> f52784fdd592, initial migration

对第一个迁移来说, 其作用和调用db.create_all() 方法一样。但在后续的迁移中,upgrade 命令能把改动应用到数据库中,且不影响其中保存的数据。

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值