zoukankan      html  css  js  c++  java
  • flask-admin章节三:数据库迁移工具 alembic初步使用

    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的使用还在继续,刚接触,记录下。

  • 相关阅读:
    testDecoration
    python装饰器详解
    开闭原则, 对扩展开放、对修改关闭
    使用元类 编写ORM
    Python 实现累加计数的几种方法
    python 查找目录下 文件名中含有某字符串的文件
    android应用程序的混淆打包规范
    自定义Tabs
    android-Service
    Loader异步装载器
  • 原文地址:https://www.cnblogs.com/bitCoin/p/5485032.html
Copyright © 2011-2022 走看看