flask 更新数据库

时间:2021-04-19 19:47:54

在做项目的过程中,我们都遇到过,经常需要修改我们数据库的字段,在flask中,是通过ORM(对象关系映射)来创建数据库的,表--->model class,字段---->属性

在flask中,我们是通过第三方插件SQLAlchemy来创建数据库表,采用的是db.create_all()方法,这样,在我们修改数据库的Model的字段之后,想要同步数据库,就只能通过删除表,然后重新调用db.create_all()来完成。

所以,flask引入了flask_script 和 flask_migrate(pip install)来解决这个问题:

1、新建manager.py文件,编写如下:

  from flask_script import Manager

  from flask_migrate import Migrate, MigrateCommond

  from app import app

  from apps.models.base import db

  manager = Manager(app)

  # 要使用flask-migrate,必须先绑定db和app

  migrate = Migrate(app, db)

  # 将MigrateCommand添加到manager中,"db"是自定义命令

  manager.add_command("db", MigrateCommond)

  

  if __name__ == "__main__":

    manager.run()

2、打开终端,进入项目根目录,执行下面命令

  python manager.py db init

  初始化成功后,你的项目目录多了一个migrattions文件夹

  flask 更新数据库

3、然后每次在修改了model的字段后,执行下面的命令

  python manager.py db migrate

  从执行过程中我们可以看到我们修改的字段

  flask 更新数据库

  执行完成后,再去查看生成的migrations文件夹,下面的versions里面会多出来一个.py的迁移脚本文件,检查表格及字段

  flask 更新数据库

3、更新数据库

  python manager.py db upgrade

  upgrade会自动读取刚才生成的文件,将数据更新到数据库中。

  flask 更新数据库

  执行完成后,检查你的数据库,发现数据库字段按照我们的需要更新了。

【注1: python manager.py db migrate 将模型生成迁移文件,只要模型修改了,就执行一次这个命令】

【注2:python manger.py db upgrade 将迁移文件真正映射到数据库中,每次运行了migrate命令,就要记得运行这个命令,不然数据库中的信息是不会被修改的。】

【注3:记得要把对应的模型引入到manager.py文件中,不然执行命令是不会起作用的哦^_^】

4、回滚事物

  1)运行 python manager.py db history

    查看历史版本,得到history-id

  2)运行 python manager.py db downgrade history-id

    回退到指定历史版本