alembic数据迁移工具(四)

    alembic是用来做ORM模型与数据库的迁移与映射。alembic使用方式跟git有点类似,表现在两个方面,第一个,alemibi的所有命令都是以alembic开头;

第二,alembic的迁移文件也是通过版本进行控制的。安装方式:pip install alembic

alembic的使用

(1)定义模型

models.py

from sqlalchemy import Column,Integer,String,create_engine
from  sqlalchemy.ext.declarative import declarative_base

DB_URI = "mysql+pymysql://root:[email protected]:3306/alembic_demo?charset=utf8"

engine = create_engine(DB_URI)

Base = declarative_base(engine)

class User(Base):
    __tablename__ = 'user'
    id = Column(Integer,primary_key=True,autoincrement=True)
    username = Column(String(50),nullable=False)

(2)在cmd终端初始化,创建一个仓库

alembic init learn_alembic

(3)修改配置文件,指定连接的数据库

alembic.ini

sqlalchemy.url = mysql+pymysql://root:[email protected]:3306/alembic_demo?charset=utf8

(4)将models所在的目录路径添加到env.py,并指定target_metadata

import sys,os
# 1.__file__:当前文件(env.py)
#2.os.path.dirname(__file__):获取当前文件的目录
#3.os.path.dirname(os.path.dirname(__file__)):获取当前文件目录的上一级目录
#4.sys.path: python寻找导入的包的所有路径
sys.path.append(os.path.dirname(os.path.dirname(__file__)))
import models


target_metadata = models.Base.metadata

(5)生成迁移脚本

alembic revision --autogenerate -m "第一次提交"

(6)将生成的迁移脚本映射到数据库中

alembic upgrade head

以后如果想要添加或修改模型,重复5,6步骤即可

alembic常用命令和经典错误解决办法

常用命令和参数解释:

init:创建一个alembic仓库
rebision:创建一个新的版本文件
--autogenerate:自动将当前模型的修改,生成迁移脚本
-m:本次迁移做了哪些修改
upgrade:将指定版本的迁移文件映射到数据库中,会执行版本文件中的upgrade函数
head:代表当前的迁移脚本的版本号
downgrade:会执行指定版本的迁移文件中的downgrade函数
heads:展示当前可用的heads脚本文件
history:列出所有的迁移版本及其信息
current:展示当前数据库中的版本号

经典错误

FAILED:Target databases is not up to date.

原因:主要是heads和current不相同。current落后于heads的版本

解决办法:将current移动到head上。alembic upgrade head

FAILED:Can't locate revision identified by 'xxxxxxx'

原因:数据库中存的版本号不在迁移脚本文件中

解决办法:删除数据的alembic_version表中的数据,重新执行alembic upgrade head

current命令使用介绍

用alembic工具:数据库中会自动生成一张表alembic_version

12206509-72cd547c8f4ae028.png
image

在数据库中可以查看当前的版本号

12206509-ac72f4600356b44e.png
image

在cmd终端也可以通过current命令查看

alembic current

Flask-SQLAlchemy下使用alembic

(1)config.py

DB_URI = "mysql+pymysql://root:[email protected]:3306/flask_alembic_demo?charset=utf8"

SQLALCHEMY_DATABASE_URI = DB_URI

(2)flask_alembic_demo.py

from flask import Flask
from flask_sqlalchemy import SQLAlchemy
import config

app = Flask(__name__)
app.config.from_object(config)

db = SQLAlchemy(app)

class User(db.Model):
    __tablename__ = 'user'
    id = db.Column(db.Integer,primary_key=True,autoincrement=True)
    username = db.Column(db.String(50),nullable=False)
  

@app.route('/')
def hello_world():
    return 'Hello World!'

if __name__ == '__main__':
    app.run()

(3)初始化

alembic init alembic

(4)alembic.ini

sqlalchemy.url = mysql+pymysql://root:[email protected]:3306/flask_alembic_demo?charset=utf8

(5)env.py

import sys,os
sys.path.append(os.path.dirname(os.path.dirname(__file__)))
import flask_alembic_demo

#用的是db.Model
target_metadata = flask_alembic_demo.db.Model.metadata

(6)生成迁移脚本

alembic revision --autogenerate -m "first commit"

(7)upgrade到数据库

alembic upgrade head

(8)添加字段

假入想添加一个字段age

class User(db.Model):
    __tablename__ = 'user'
    id = db.Column(db.Integer,primary_key=True,autoincrement=True)
    username = db.Column(db.String(50),nullable=False)
    age = db.Column(db.Integer)

重复步骤即可

alembic revision --autogenerate -m "add column age"

alembic upgrade head

猜你喜欢

转载自blog.csdn.net/weixin_33709364/article/details/87636279