flask-alembic数据迁移工具

alembic是用来做ORM模型与数据库的迁移与映射。alembic使用方式跟git有点类似,表现在两个方面,
第一,alemibi的所有命令都是以alembic开头;
第二,alembic的迁移文件也是通过版本进行控制的。安装方式:pip install alembic

1.1.alembic的使用

 (1)定义模型

models.py

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

DB_URI = "mysql+pymysql://root:123456@127.0.0.1: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:123456@127.0.0.1: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步骤即可

 

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

常用命令和参数解释:

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

 经典错误

1.FAILED:Target databases is not up to date.

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

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

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

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

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

 

1.3.current命令使用介绍

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

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

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

alembic current

 

1.4.Flask-SQLAlchemy下使用alembic

(1)config.py

DB_URI = "mysql+pymysql://root:123456@127.0.0.1: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:123456@127.0.0.1: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

 

posted @ 2018-12-11 13:55  Hear7  阅读(271)  评论(0编辑  收藏  举报