Flask flask-migrate 数据库迁移
简介
flask-migrate是flask的一个扩展模块,主要是扩展数据库表结构的.
官方文档:http://flask-migrate.readthedocs.io/en/latest/
使用:
使用一共分为三步:
创建迁移环境->生成迁移脚本->更新数据库
本地数据库连接与相关配置文件 config.py
import os DEBUG = True # DEBUG = False SECRET_KEY = os.urandom(24) # HOSTNAME = 'mysql' HOSTNAME = '127.0.0.1' PORT = '3306' DATABASE = 'autumnwater' USERNAME = 'root' PASSWORD = 'root' SQLALCHEMY_DATABASE_URI = 'mysql+pymysql://{}:{}@{}:{}/{}?charset=utf8'.format(USERNAME, PASSWORD, HOSTNAME, PORT, DATABASE) SQLALCHEMY_TRACK_MODIFICATIONS = False
数据库迁移文件 manage.py
from flask_script import Manager from flask_migrate import MigrateCommand, Migrate from index import app from exts import db # 存放命令脚本 manager = Manager(app) # 使用Migrate绑定app和db migrate = Migrate(app, db) # 添加迁移脚本的命令到manager中 manager.add_command('db', MigrateCommand) if __name__ == '__main__': manager.run()
如果没有下载flask-migrate会报错,使用pip下载,下载后可查看已安装库文件:
执行命令
python manage.py db init #初始化数据库,会创建一个migations文件夹,并且会在数据库中生成一个alembic_version表 python manage.py db migrate #创建迁移历史(迁移脚本) python manage.py db upgrade #更新数据库
注:只有在第一次迁移的时候需要初始化数据库
具体步骤如下:
-
-
使用命令
python manage.py db init
初始化数据库,会创建一个migations文件夹,该文件夹是Alembic模块自动创建的,默认名字叫migrations,可以在创建migrate=Migrate(app,db)对象时传入directory="filename"参数来自定义目录名
migrations 里面有一个versions文件夹,这个文件夹用来存放迁移脚本,执行迁移命令后会自动生成迁移脚本保存在里面
同时会在数据库中生成一个alembic_version表
我这里是python3的环境,所以上述命令 python -> python3
同时可见migations文件夹已生成:
-
使用命令
python manage.py db migrate
创建迁移历史(迁移脚本),可以通过 -m 参数添加迁移信息,类似于git提交代码时添加提交信息,例如:
python manage.py db migrate -m "create table"
执行该步骤时显示
解决方案寻找:
https://blog.csdn.net/qq_41389354/article/details/104254677
直接使用pip install pymysql即可
又出现新的错误
截取部分报错为:
sqlalchemy.exc.ProgrammingError: (pymysql.err.ProgrammingError) (1146, "Table 'performance_schema.session_variables' doesn't exist") [SQL: SHOW VARIABLES LIKE 'sql_mode'] (Background on this error at: http://sqlalche.me/e/13/f405)
在 stackoverflow 上找到解决方案:
https://stackoverflow.com/questions/31967527/table-performance-schema-session-variables-doesnt-exist
进入mysql的bin目录下,打开cmd运行:
mysql_upgrade -u root -p --force
出现该问题的原因应该是之前我将phpstudy里面的mysql强制升级为了5.7版本
重启phpstudy,然后重新运行
python manage.py db migrate
如图:
-
最后更新数据库,完成迁移
python manage.py db upgrade
查看数据库的表
如图,执行 upgrade 命令后,会在数据库中创建一张 alembic_version 表,这张表不是代码中定义的,是 Alembic 自动创建的(看名字就知道了),里面保存的是当前数据库的版本 id
alembic_version 表不能删除,删除后就不能继续执行数据库迁移操作了,除非重新初始化。
同时,执行 upgrade 命令后,会根据代码中定义的模型类创建对应的表,表的字段与模型类中定义的一致。
如上面的数据库中创建了user表,因为在我的models里面定义了该表:
from exts import db from datetime import datetime from werkzeug.security import generate_password_hash, check_password_hash class User(db.Model): __tablename__ = 'user' id = db.Column(db.Integer, primary_key=True, autoincrement=True) email = db.Column(db.String(20), nullable=False) username = db.Column(db.String(50), nullable=False) pw_hash = db.Column(db.String(128), nullable=False) def __init__(self, email, username, password): self.email = email self.username = username self.set_password(password) def set_password(self, password): self.pw_hash = generate_password_hash(password) def check_password(self, password): return check_password_hash(self.pw_hash, password)
而如果数据库中有其他表(没有对应模型类的表),会被删除。这点需要特别注意,数据库迁移时最好使用一个新的数据库(不要与其他项目用同一个数据库),避免造成数据丢失。
优势:
个人直观感受使用数据库迁移的优势有:
-
使用数据库迁移,可以直接建表,而不用我们自己写sql语句用来建表。就是将关系型数据库的一张张表转化成了Python的一个个类。
这里即我们之前在 models.py 文件中创建的类,可以通过数据库迁移直接生成表
-
在开发中经常会遇到需要修改原来的数据库模型,修改之后更新数据库,最简单粗暴的方式就是删除旧表,然后在增加新表,这样做的缺点是会造成数据丢失。在 Flask 中,可以使用数据库迁移来解决这个问题,数据库迁移可以追踪数据模型类的变化,然后把变动应用到数据库中,不会删表造成数据丢失。
大部分参考COPY自:
https://www.cnblogs.com/jiangchunsheng/p/9218338.html
https://blog.csdn.net/weixin_43790276/article/details/103554632