Flask——migrate实现数据库迁移

migrate实现数据库迁移的指令如下:

指令	代码	说明
初始化	python 文件名.py db init	这个命令会创建migrations文件夹，所有迁移文件都放在里面
创建迁移脚本	python 文件名.py db migrate -m ‘记录’	创建的迁移脚本会根据模型定义和数据库当前状态的差异，生成upgrade和downgrade函数的内容
更新数据库	python 文件名.py db upgrade	将修改后的内容上传到数据库
回退数据库	python 文件名.py db downgrade 版本号	将数据库回退到指定版本
查看修改历史	python 文件名.py db history	查看历史版本的具体版本号

首先创建新的数据库表，命名位为my_test，进入MySQL输入命令如下：

create database my_test charset=utf8;

接下来往里面填充数据：

from flask import Flask
from flask_sqlalchemy import SQLAlchemy

app = Flask(__name__)

"""
第一步：创建my_project数据库(mysql> create database my_project default charset=utf8;),
        Flask创建对象与mysql进行联系
"""
class Config(object):
    """配置参数"""
    # sqlalchemy的配置参数
    SQLALCHEMY_DATABASE_URI = "mysql://root:123456@127.0.0.1:3306/my_test"

    # 设置每次请求结束后会自动提交数据库中的改动，一般都设置手动保存
    SQLALCHEMY_COMMIT_ON_TEARDOWN = False

    # 设置sqlalchemy自动更新跟踪数据库
    SQLALCHEMY_TRACK_MODIFICATIONS = True

# 连接数据库
app.config.from_object(Config)

# 创建数据库aqlalchemy工具对象
db = SQLAlchemy(app)


"""
第二步：创建作者数据库表（Author）和所著书籍数据库表（Book）并使两者进行关联
"""
# 创建数据库模型类
class Author(db.Model):
    """作者"""
    __tablename__ = "tbl_authors"

    id = db.Column(db.Integer, primary_key=True)  # 整型的主键，会默认设置为自增主键
    name = db.Column(db.String(32), unique=True)
    books = db.relationship("Book", backref="author")

# 创建数据库模型类
class Book(db.Model):
    """所著书籍"""
    __tablename__ = "tbl_books"  # 指明数据库的表名

    id = db.Column(db.Integer, primary_key=True)  # 整型的主键，会默认设置为自增主键
    name = db.Column(db.String(64), unique=True)
    author_id = db.Column(db.Integer, db.ForeignKey("tbl_authors.id"))

@app.route("/")
def index():
    return "This is the beginning"

if __name__ == '__main__':
	# 以下两行代码是数据库表第一次创建使用,已有的数据表使用时请注释
	db.drop_all()
    db.create_all()
    app.run(debug=True, host="127.0.0.1", port=8000)

运行完上面的代码后, MySQL上的表格为:

使用migrate实现数据库迁移需要安装flask_migrate库，但是最新版的flask_migrate里的MigrateCommand模块已合入导致无法调用，因此要安装flask_migrate版本为2.7.0版本的库，打开cmd输入以下命令：

pip install flask_migrate==2.7.0

实现数据库迁移要先导入以下库文件：

from flask_script import Shell,Manager
from flask_migrate import Migrate, MigrateCommand

然后创建flask脚本管理工具对象——>创建数据库迁移工具对象——>向manger对象中天年数据库的操作命令,代码如下:

"""
数据库迁移
"""
# 创建flask脚本管理工具对象
manager = Manager(app)

#第一个参数是Flask的实例，第二个参数是Sqlalchemy数据库实例，创建数据库迁移工具对象
migrate = Migrate(app,db)

#manager是Flask-Script的实例，这条语句在flask-Script中添加一个db命令，向manger对象中天年数据库的操作命令
manager.add_command('db',MigrateCommand)

启动程序修改为:

if __name__ == '__main__':
    # 通过manager启动程序
    manager.run()

点击Termianl进行初始化,命令如下;

python database.py db init

上面的图片显示migrations文件夹已创建完成,接下来的迁移文件都放在里面,将数据库模型Author增加email属性再创建迁移脚本,命令如下:
数据库模型Author增加email属性:

创建迁移脚本:

python database.py db migrate -m "add email"

数据库界面显示:

创建迁移脚本后要更新数据库(即执行上传动作),命令如下:

python database.py db upgrade

将数据库模型Book增加summary属性,更新数据库后查看修改历史:

数据库显示:

查看修改历史:

python database.py db history

执行降版本操作,命令如下:

python database.py db dowmgrade b004803e0884

数据库界面显示:

最后附上完整的代码:

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

app = Flask(__name__)

"""
第一步：创建my_project数据库(mysql> create database my_project default charset=utf8;),
        Flask创建对象与mysql进行联系
"""
class Config(object):
    """配置参数"""
    # sqlalchemy的配置参数
    SQLALCHEMY_DATABASE_URI = "mysql://root:123456@127.0.0.1:3306/my_test"

    # 设置每次请求结束后会自动提交数据库中的改动，一般都设置手动保存
    SQLALCHEMY_COMMIT_ON_TEARDOWN = False

    # 设置sqlalchemy自动更新跟踪数据库
    SQLALCHEMY_TRACK_MODIFICATIONS = True

# 连接数据库
app.config.from_object(Config)

# 创建数据库aqlalchemy工具对象
db = SQLAlchemy(app)

"""
数据库迁移
"""
# 创建flask脚本管理工具对象
manager = Manager(app)

#第一个参数是Flask的实例，第二个参数是Sqlalchemy数据库实例，创建数据库迁移工具对象
migrate = Migrate(app,db)

#manager是Flask-Script的实例，这条语句在flask-Script中添加一个db命令，向manger对象中天年数据库的操作命令
manager.add_command('db',MigrateCommand)


"""
第二步：创建作者数据库表（Author）和所著书籍数据库表（Book）并使两者进行关联
"""
# 创建数据库模型类
class Author(db.Model):
    """作者"""
    __tablename__ = "tbl_authors"

    id = db.Column(db.Integer, primary_key=True)  # 整型的主键，会默认设置为自增主键
    name = db.Column(db.String(32), unique=True)
    books = db.relationship("Book", backref="author")
    # 使用migrate添加新内容
    email = db.Column(db.String(32))

# 创建数据库模型类
class Book(db.Model):
    """所著书籍"""
    __tablename__ = "tbl_books"  # 指明数据库的表名

    id = db.Column(db.Integer, primary_key=True)  # 整型的主键，会默认设置为自增主键
    name = db.Column(db.String(64), unique=True)
    author_id = db.Column(db.Integer, db.ForeignKey("tbl_authors.id"))
    # 使用migrate添加新内容
    summary = db.Column(db.String(32))


@app.route("/")
def index():
    return "This is the beginning"

if __name__ == '__main__':
    # 通过manager启动程序
    manager.run()

以上就是Flask使用migrate实现数据库迁移的基本方法，各位读者有疑问的可以私信博主或评论留下足迹。