flask-admin章节三:数据库迁移工具 alembic初步使用

时间:2020-12-09 21:07:20

1. 概述

基于flask框架构建web,一般会使用sqlchemy(在flask中使用sqlchemy可以参考这里)作为数据库引擎。

这样业务的逻辑就可以做到不跟具体的数据库类型相耦合,具体后端业务是使用那种数据库就全凭业务需要了。

但是数据表结构并不一定是一成不变的,可能随着业务的需要表的结构经常会变换。面对这种情况,单纯删除之前的表,

重新再建新的表会导致数据的丢失,如果自己迁移又比较麻烦。以为,自己迁移的话,可能只能切到比较新的版本,但是

想要回滚到之前的版本几乎是不可能的,这样涉及到使用数据库迁移工具了。

针对flask,有两个比较重要的数据库迁移工具:flask migratealembic。flask migrate是基于alembic,但是个人

感觉flask migrate不够灵活,而且使用alembic还是比较方便的。所以,我这边是直接使用alembic来做为数据库迁移工具的。

首先,稍微介绍下alembic,alembic是sqlchemy作者一手缔造的,对sqlchemy的兼容性肯定是其他组件根本无法媲美的。

2. 安装

安装alembic还是蛮方便的,直接使用python安装工具pip就可以了,具体命令如下:

sudo pip install alembic

 

3. 初始化

回到flask application所处目录,执行如下命令:

alembic init alembic

执行成功之后,就能在当前目录看到两个比较重要的文件:alembic.ini文件和alembic目录。

alembic.ini保存的是alembic的配置信息,而数据库版本信息的记录是在alembic目录下。

alembic目录的机构如下:

[root@localhost alembic]# ls -trlh
total 16K
-rw-r--r-- 1 root root 38 May 11 19:36 README
-rw-r--r-- 1 root root 494 May 11 19:36 script.py.mako
-rw-r--r-- 1 root root 2.1K May 12 02:43 env.py
drwxr-xr-x 2 root root 4.0K May 12 18:53 versions

这块起作用的有两个,一个是env.py文件以及versions目录。

env.py记录着跟环境有关的一些信息,比较起作用的是target_metadata这个字段,后面在介绍使用的时候会提到。

versions目录下面是记录当前数据库所有版本,以及对应操作的python文件。

每个文件内容大致如下:

"""empty message

Revision ID: 27c6a30d7c24
Revises: None
Create Date: 2011-11-08 11:40:27.089406 """ # revision identifiers, used by Alembic.
revision = '27c6a30d7c24'
down_revision = None from alembic import op
import sqlalchemy as sa def upgrade():
pass def downgrade():
pass

revision字段记录的是当前文件代表的版本,初始时down_revision为None。具体在使用过程中一般指向的是上一个版本,具体

在回滚过程中非常有用。默认情况下upgradedowngrade都会空。

在数据库升级或者回滚过程中必须要重写这这个函数,如果要升级就需要重写upgrade函数,回滚就需要重写downgrade函数。

4. 一个简答的例子

首先必须更改alembic.ini文件中的sqlalchemy.url字段,改为你目前在使用的数据路径就OK了。

此处,我是改为:

sqlalchemy.url = sqlite:////home/dashuju/admin_db.sqlite

这个文件其他字段可以暂时不用更改,目前的设置就可以使用了。

更改alembic目录下的env.py文件,主要是设置target_metadata这个字段:

import sys
sys.path.append('.')
from admin_app import db target_metadata = db.metadata

admin_app是flask application的文件名,db是在flask_app定义的全局变量,

具体初始化为:db = SQLAlchemy(app)

更改下versions目录下的当前版本py文件,并根据对应的upgrade和downgrade函数,具体代码如下:

def upgrade():
op.create_table(
'users',
sa.Column('id', sa.Integer, primary_key=True),
sa.Column('name', sa.String(50), nullable=False),
sa.Column('description', sa.Unicode(200)),
) def downgrade():
op.drop_table('users')

执行:

alembic upgrade head

指令来升级数据库,查看对应的数据库,可以看到数据库中多了一个users表。

查看versions目录下的文件,会发现多了一个py文件。可以在flask application所在目录执行:

alembic current来查看当前数据库所处版本,也可以使用alembic history来查看数据库变更的历史记录。

如果想要回滚,也可以使用alemibic downgrade -1指令来回滚数据库到之前的版本。

更改表结构

对于想要更改表结构,比如插入column或者删除column,删除column可能特殊些,所以要讲下:

插入行,一般如下:

op.add_column('users', sa.Column('id', sa.INTEGER(), nullable=True))

删除行,可能特殊些,具体如下:

with op.batch_alter_table('users') as batch_op:
batch_op.drop_column('id')

想要删除多列,可以在with语句下增加多行就可以了,否则会报错:

sqlalchemy.exc.OperationalError: (sqlite3.OperationalError) near "DROP": syntax error [SQL: u'ALTER TABLE posts DROP COLUMN XXX']

5. 后续

alembic的使用还在继续,刚接触,记录下。