Flask 学习-15.flask-migrate数据迁移

2022-08-31 14:49:40 浏览数 (1)

前言

Flask-SQLAlchemy ORM 可以直接操作数据库,可以用db.create_all()代码同步表到数据库。 当我们需要修改表的字段,比如对表新增字段,修改字段的时候需用到flask-migrate 插件来同步迁移数据。

flask-migrate 迁移数据

django 框架里面可以用migrate 命令进行数据库的同步,Flask 框架有个flask-migrate 插件可以实现类似功能。 通过命令行操作实现数据库的同步。

代码语言:javascript复制
from flask import Flask
from flask_sqlalchemy import SQLAlchemy
from flask_migrate import Migrate

app = Flask(__name__)
# 设置数据库连接地址
DB_URI = 'mysql pymysql://root:123456@127.0.0.1:3306/web'
app.config['SQLALCHEMY_DATABASE_URI'] = DB_URI
# 是否追踪数据库修改,一般不开启, 会影响性能
app.config['SQLALCHEMY_TRACK_MODIFICATIONS'] = False
# 是否显示底层执行的SQL语句
app.config['SQLALCHEMY_ECHO'] = True

# 初始化db,关联flask 项目
db = SQLAlchemy(app)
# 迁移组件初始化
Migrate(app, db)

# 创建模型
class Students(db.Model):
    __tablename__ = 'students'  # 数据库表名

    id = db.Column(db.Integer, primary_key=True)
    name = db.Column(db.String(20))
    fullname = db.Column(db.String(30))
    nickname = db.Column(db.String(30))

    def __repr__(self):
        return f"<Students(name='{self.name}', fullname='{self.fullname}', nickname='{self.nickname}')>"

class Users(db.Model):
    __tablename__ = 'user'  # 数据库表名

    id = db.Column(db.Integer, primary_key=True)
    name = db.Column(db.String(30))

    def __repr__(self):
        return f"<Users(id='{self.id}', name='{self.name}')>"

执行迁移命令,先设置环境变量, 如果启动文件是app.py 这步可以省略。

代码语言:javascript复制
set FLASK_APP=app.py  # 设置环境变量指定启动文件

windows用set设置环境变量,linux 用export 设置环境变量

代码语言:javascript复制
export FLASK_APP=app.py  # 设置环境变量指定启动文件

执行相关命令

代码语言:javascript复制
flask db init  # 生成迁移文件夹  只执行一次
flask db migrate  # ⽣成迁移版本, 保存到迁移文件夹中
flask db upgrade  # 执行迁移

迁移命令

init 命令执行后,会生成migrations目录

代码语言:javascript复制
flask db init

migrate 命令生成迁移脚本,保存到迁移文件夹中versions目录

代码语言:javascript复制
flask db migrate

upgrade 命令执行迁移

代码语言:javascript复制
flask db upgrade

执行完upgrade 命令才会在数据库中生成表

2022年第 12期《python接口web自动化 测试开发》课程,9月17号开学!

本期上课时间:2022年9月17号 - 2022年12月17号,周六周日上午9:00-11:00

报名费:报名费3000一人(周期3个月)

联系微信/QQ:283340479

0 人点赞