flask 数据库迁移_数据库迁移方案

编程汇总 • 2025-02-07 10:17 • 阅读 19

在开发的过程中，需要修改数据库的模型，而且需要在修改之后更新数据库，最直接就是删除旧表，但是会丢失数据。所有最好的方式就是数据库迁移。它可以追踪数据库模型的变化，然后把变动应用到数据库中。

在flask中可以使用Flask-Migrate扩展，来实现数据迁移。并且集成到flask-Script中，所有的操作通过命令就能完成。

Flask-Migrate提供了一个MigrateCommand类，可以附加到flask-script的manage对象上。

from flask import Flask

from flask_sqlalchemy import SQLAlchemy

from flask_migrate import MigrateCommand, Migrate

from flask_script import Manager

app = Flask(__name__)

manage = Manager(app)

class Config(object):

    DEBUG = True

    SQLALCHEMY_DATABASE_URI = 'mysql://root:wjm@localhost/py9'

    SQLALCHEMY_TRACK_MODIFICATIONS = False

app.config.from_object(Config)

db = SQLAlchemy(app)

# 创建migrate实例，第一个参数是Flask的实例，第二个是sqlalchemy的数据库实例

migrate = Migrate(app, db)

# manage是Flask-Script的实例，这条语句在flask_script中添加了一个db命令

manage.add_command('db',MigrateCommand)

# 定义模型

class Role(db.Model):

    # 定义表名

    __tablename__ = 'roles2'

    # 定义字段

    id = db.Column(db.Integer,primary_key=True)

    name = db.Column(db.String(32),unique=True)

    # repr显示一个可读字符串

    def __repr__(self):

        return 'name:'.format(self.name)

class User(db.Model):

    # 定义表名

    __tablename__ = 'users2'

    # 定义字段

    id = db.Column(db.Integer,primary_key=True)

    name = db.Column(db.String(32), unique=True, index=True)

    # 设置外键

    role_id = db.Column(db.Integer, db.ForeignKey('roles2_id'))

    def __repr__(self):

        return 'User:'.format(self.name)

if __name__ == '__main__':

    manage.run()

在终端执行下面的命令。会创建migrations文件夹，所有的迁移文件都放在里面。

python manage.py db init

创建自动迁移脚本：

upgrade（）：函数把迁移中的改动应用到数据库中。

downgrate（）：函数则将改动删除。

自动创建的迁移脚本会
根据模型定义和数据库当前状态的差异，生成upgrade（）和downgrade（）函数的内容。

对比不一定完全正确，有可能会遗漏一些细节，需要进行检查。

python manage.py db migrade -m 'initial migration'

注意：这里有可能出现着个错误

只需要把-m给删掉就可以了，参数太多没办法

发现有不同，此时更新数据库

python manage.py db upgrade

更新完之后，在数据库会出现一个表

versions每迁移一次里面都会生成一个文件。

回退版本：先通过history命令来找到版本号，然后传给downgrade命令

python manage.py db history

python manage.py db downgrade 版本号

实际操作顺序：

1.先创建python文件 manage.py，内容为

2.python manage.py db init

3.python manage.py db migrate -m ‘版本号（注释）‘

4.python manage.py db upgrade

5.根据需求修改模型

6.python 文件 db migrate -m”新版本名(注释)”

7.python 文件 db upgrade 然后观察表结构

8.若返回版本,则利用 python 文件 db history查看版本号

9.python 文件 db downgrade(upgrade) 版本号