Python学习笔记：Flask-Migrate基于model做upgrade的基本原理

 

1）flask-migrate的官网：https://flask-migrate.readthedocs.io/en/latest/ 

2）获取帮助，在pycharm的控制台中输入 flask db  --help，或者flask db migrate  --help，这样就可以了解各个命令的信息了

3）一般用法，

    >flask db init

    >flask db migrate

    >flask db upgrade

    后续对脚本有任何的修改，只要重复执行后两句就可以了。

 

4）测试1：添加一行数据，删除email列，然后再加上

    针对下面这个表：

 

class User(db.Model):
    __tablename__ = 'user'
    id = db.Column(db.Integer,primary_key=True)
    username = db.Column(db.String(64),index=True,unique=True)
    email = db.Column(db.String(120),index=True,unique=True)
    password_hash = db.Column(db.String(128))
 
    def __repr__(self):
        return '<用户名:{}>'.format(self.username)

 

 

    a）先通过以上3个命令创建表；

    b）然后再数据库中插入一条数据；

    c）然后从models中把email列删除，执行migrate ,updrade，此时发现，数据还在，只是列删除了；

    d）然后恢复email列，执行migrate，upgrade，此时发现，数据还在，增加了一个email的空列；

    仔细观察日志，发现这个upgrade执行的是升级，而不是重新创建

 

INFO  [alembic.runtime.migration] Running upgrade 04a26df9c6d3 -> 04f13e3c2f7e, empty message

 

 

其中04a26df9c6d3_.py的内容为：

 

 

 

"""empty message
Revision ID: 04a26df9c6d3
Revises: 4ed84444672c
Create Date: 2018-08-29 07:14:44.779039
 """
from alembic import op
import sqlalchemy as sa
 
 
# revision identifiers, used by Alembic.
revision = '04a26df9c6d3'
down_revision = '4ed84444672c'
branch_labels = None
depends_on = None
 
 
def upgrade():
    # ### commands auto generated by Alembic - please adjust! ###
    op.drop_index('ix_user_email', table_name='user')
    op.drop_column('user', 'email')
    # ### end Alembic commands ###
 
 
def downgrade():
    # ### commands auto generated by Alembic - please adjust! ###
    op.add_column('user', sa.Column('email', sa.VARCHAR(length=120), autoincrement=False, nullable=True))
    op.create_index('ix_user_email', 'user', ['email'], unique=True)
    # ### end Alembic commands ###

 

 

 

04f13e3c2f7e_.py 的内容为

"""empty message
 
Revision ID: 04f13e3c2f7e
Revises: 04a26df9c6d3
Create Date: 2018-08-29 07:19:33.876815
 
"""
from alembic import op
import sqlalchemy as sa
 
 
# revision identifiers, used by Alembic.
revision = '04f13e3c2f7e'
down_revision = '04a26df9c6d3'
branch_labels = None
depends_on = None
 
 
def upgrade():
    # ### commands auto generated by Alembic - please adjust! ###
    op.add_column('user', sa.Column('email', sa.String(length=120), nullable=True))
    op.create_index(op.f('ix_user_email'), 'user', ['email'], unique=True)
    # ### end Alembic commands ###
 
 
def downgrade():
    # ### commands auto generated by Alembic - please adjust! ###
    op.drop_index(op.f('ix_user_email'), table_name='user')
    op.drop_column('user', 'email')
    # ### end Alembic commands ###

 

 

 

 

5）测试2：把user表删除

    a）如果只执行flask db upgrade，则什么都不执行

(venv) D:WORKgitbaseHelloFlask>flask db upgrade

INFO  [alembic.runtime.migration] Context impl PostgresqlImpl.

INFO  [alembic.runtime.migration] Will assume transactional DDL.

 

 

    b）执行flask db migrate，然后再执行flask db upgrade，则可以重新创建表

    通过migrate，程序会先比较本地的model和数据库中实际的表，然后创建如下脚本 69f5e753ee6c_.py

"""empty message
 
Revision ID: 69f5e753ee6c
Revises: 04f13e3c2f7e
Create Date: 2018-08-29 07:31:19.236320
 
"""
from alembic import op
import sqlalchemy as sa
 
 
# revision identifiers, used by Alembic.
revision = '69f5e753ee6c'
down_revision = '04f13e3c2f7e'
branch_labels = None
depends_on = None
 
 
def upgrade():
    # ### commands auto generated by Alembic - please adjust! ###
    op.create_table('user',
    sa.Column('id', sa.Integer(), nullable=False),
    sa.Column('username', sa.String(length=64), nullable=True),
    sa.Column('email', sa.String(length=120), nullable=True),
    sa.Column('password_hash', sa.String(length=128), nullable=True),
    sa.PrimaryKeyConstraint('id')
    )
    op.create_index(op.f('ix_user_email'), 'user', ['email'], unique=True)
    op.create_index(op.f('ix_user_username'), 'user', ['username'], unique=True)
    # ### end Alembic commands ###
 
 
def downgrade():
    # ### commands auto generated by Alembic - please adjust! ###
    op.drop_index(op.f('ix_user_username'), table_name='user')
    op.drop_index(op.f('ix_user_email'), table_name='user')
    op.drop_table('user')
    # ### end Alembic commands ###

 

6）结论

    对模型的任何修改，都要再执行flask db migrate, flask db upgrade

    这里做的任何的更新，实际上都是把模型和数据库当前的状态做比较的

    而且历次的版本都有，所以实际上可以回退到之前的任何一个版本。

    

    这个思路其实挺好，这就是我们之前写数据库补丁脚本的思路，一环套一环，只不过之前是靠人的，现在是靠程序自动生成的。

    而且他这里更好的地方是，不但支持upgrade，还支持downgrade