数据库迁移

什么是数据库迁移？
当我们在看关于web开发的教程时，教程在设计数据表时，不是一下子就设计好的，而是先设计一个能用的数据表，然后再慢慢的往里面增加东西。当我们往表里面增加东西时，比如说增加一个字段，那么之前的数据怎么办？
这时候就要用到数据库迁移了，在数据表进行更新时，可以利用数据库迁移脚本在当前数据表进行更新，而不是重新创建一个数据表。
设计一个用户模型

from app import db

class User(db.Model):
    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 '<User {}>'.format(self.username)

创建一个迁移存储库

(venv) $ flask db init

数据库迁移

(venv) $ flask db migrate -m "users table"

这时候只是生成了一个脚本文件位于/migrations/versions中
打开，可以观察到两个函数

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 ###

执行upgrade会更新，执行downgrade会回退到上一版本

(venv) $ flask db upgrade

关于用户表，还想让它再存储点关于用户本人的信息，更新User模型

class User(UserMixin, db.Model):
    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))
    posts = db.relationship('Post', backref='author', lazy='dynamic')
    about_me = db.Column(db.String(140))
    last_seen = db.Column(db.DateTime, default=datetime.utcnow)

    def __repr__(self):
        return '<User {}>'.format(self.username)

数据库迁移

(venv) $ flask db migrate -m "users table"

打开/migrations/versions/可以观察到多了一个脚本

def upgrade():
    # ### commands auto generated by Alembic - please adjust! ###
    op.add_column('user', sa.Column('about_me', sa.String(length=140), nullable=True))
    op.add_column('user', sa.Column('last_seen', sa.DateTime(), nullable=True))
    # ### end Alembic commands ###


def downgrade():
    # ### commands auto generated by Alembic - please adjust! ###
    op.drop_column('user', 'last_seen')
    op.drop_column('user', 'about_me')
    # ### end Alembic commands ###

可以看到upgrade函数把刚才在User模型中添加的about_me, last_seen这两个字段添加到数据表中了
更新

(venv) $ flask db upgrade

参考文章
The Flask Mega-Tutorial Part IV: Database