123456789101112131415161718192021222324252627282930313233343536373839404142 |
- """Migrate current flag
- Revision ID: 6119fbbcc8e9
- Revises: 2b626a16542e
- Create Date: 2017-11-16 15:11:28.199971
- """
- # revision identifiers, used by Alembic.
- revision = '6119fbbcc8e9'
- down_revision = '2b626a16542e'
- from alembic import op
- import sqlalchemy as sa
- def upgrade():
- """ Add the status column to pull_request_flags and migrate the data.
- """
- op.add_column(
- 'pull_request_flags',
- sa.Column('status', sa.String(32), nullable=True)
- )
- op.execute(
- 'UPDATE pull_request_flags SET status=\'success\' '
- 'WHERE percent in (100, \'100\')')
- op.execute(
- 'UPDATE pull_request_flags SET status=\'failure\' '
- 'WHERE percent not in (100, \'100\')')
- op.alter_column(
- 'pull_request_flags', 'status',
- nullable=False, existing_nullable=True)
- def downgrade():
- """ Drop the status column in pull_request_flags.
- We can't undo the change to the status column since it may now
- contain empty rows.
- """
- op.drop_column('pull_request_flags', 'status')
|