使用alembic进行数据库版本管理

Alembic简介：

SQLAlchemy是一款非常优秀的ORM框架，但是本身没有带数据库版本控制功能，这很不方便，进行开发过程中难免修改数据模型，添加一个表，修改一个字段，都需要手动修改的话就比较费事了，还不如不用SQLAlchemy呢。

在这里介绍一款SQLAlchemy作者写的数据库版本控制工具---Alembic。另外还有一个工具叫做SQLAlchemy-Migrate，在使用过程中感觉Alembic更为灵活。

安装alembic

$ pip3 install alembic

初始化

使用之前，先在项目根目录进行初始化。

$ alembic init alembic

初始化完成后，会生成一个alembic.ini配置文件及一个alembic目录。

 项目目录树

创建模型类

#!/usr/bin/env python                       # -*- coding:utf-8 -*-                      # File Name    : sb.py                      # Author       : hexm                       # Mail         : [e-mail]    # Created Time : 2017-03-29 20:03           from sqlalchemy import Column, Integer, String, ForeignKey, Date, Table, Textfrom sqlalchemy.ext.declarative import declarative_basefrom sqlalchemy.orm import relationshipBase = declarative_base()# 用户id对应角色id 多对多useridToRoleid = Table('useridToRoleid', Base.metadata,        Column('userid', Integer, ForeignKey('users.id')),        Column('roleid', Integer, ForeignKey('roles.id')),    )# 角色id对应权限id 多对多roleidToIdentityid = Table('roleidToIdentityid', Base.metadata,        Column('roleid', Integer, ForeignKey('roles.id')),        Column('identityid', Integer, ForeignKey('identities.id')),    )# 文章id和标签id 多对多articleidToTagid = Table('articleidToTagid', Base.metadata,        Column('articleid', Integer, ForeignKey('articles.id')),        Column('tagid', Integer, ForeignKey('tags.id')),    )class User(Base):    """    用户表    """    __tablename__ = 'users'    id = Column(Integer, primary_key=True)    username = Column(String(32), nullable=False, unique=True)    password = Column(String(32), nullable=False)    email = Column(String(32), nullable=False, unique=True)    roles = relationship('Role', secondary=useridToRoleid, backref='users')    def __repr__(self):        return "<%s users.username: %s>" % (self.id, self.username)class Role(Base):    """    角色表    """    __tablename__ = 'roles'    id = Column(Integer, primary_key=True)    name = Column(String(16), nullable=False)    identity = relationship('Identity', secondary=roleidToIdentityid, backref='roles')    def __repr__(self):        return "<%s roles.name: %s>" % (self.id, self.name)class Identity(Base):    """    权限表    """    __tablename__ = 'identities'    id = Column(Integer, primary_key=True)    name = Column(String(16))    def __repr__(self):        return "<%s identities.name: %s>" % (self.id, self.name)class Category(Base):    __tablename__ = "categories"    id = Column(Integer, primary_key=True)    name = Column(String(16), nullable=False)    title = relationship('Article', backref='category')    def __repr__(self):        return "<%s categories.name: %s" % (self.id, self.name)class Tag(Base):    __tablename__ = "tags"    id = Column(Integer, primary_key=True)    name = Column(String(16))    def __repr__(self):        return "<%s categories.name: %s" % (self.id, self.name)class Article(Base):    __tablename__ = "articles"    id = Column(Integer, primary_key=True)    title = Column(String(50))    description = Column(String(300))    content = Column(Text)    click_count = Column(Integer, default=0)    category_id = Column(Integer, ForeignKey('categories.id'))    tags = relationship('Tag', secondary=articleidToTagid, backref='articles')    def __repr__(self):        return "<%s categories.name: %s" % (self.id, self.title)

修改配置文件

修改alembic.ini配置文件，只修改数据库连接部分即可，

将

sqlalchemy.url = driver://user:pass@localhost:port/dbname

修改为

sqlalchemy.url = mysql+pymysql://root:@localhost/linux_study

修改alembic/env.py

将

target_metadata = None

修改为

import sys                                             from os.path import abspath, dirname                   sys.path.append(dirname(dirname(abspath(__file__))))   from modules.models import Base                        target_metadata = Base.metadata                        

 自动创建版本

使用alembic revision -m "注释" 创建数据库版本，上面我们修改了配置文件alembic/env.py，指定了target_metadata，这里可以使用--autogenerate参数自动生成迁移脚本。

$ alembic revision --autogenerate -m "initdb"

其他常用参数

更新数据库

$ alembic upgrade 版本号

更新到最新版

alembic upgrade head

降级数据库

$ alembic downgrade 版本号

更新到最初版

alembic downgrade head

离线更新（生成sql）

alembic upgrade 版本号 --sql > migration.sql

从特定起始版本生成sql

alembic upgrade 版本一:版本二 --sql > migration.sql

查询当前数据库版本号

查看alembic_version表。

清除所有版本

将versions删掉，并删除alembic_version表

原文链接:http://www.cnblogs.com/xiaoming279/p/6641601.html