notifications-api/migrations/versions/0076_add_intl_flag_to_provider.py

"""empty message

Revision ID: 0076_add_intl_flag_to_provider
Revises: 0075_create_rates_table
Create Date: 2017-04-25 09:44:13.194164

"""

# revision identifiers, used by Alembic.
revision = '0076_add_intl_flag_to_provider'
down_revision = '0075_create_rates_table'

from alembic import op
import sqlalchemy as sa


def upgrade():
    op.add_column('provider_details', sa.Column('supports_international', sa.Boolean(), nullable=False, server_default=sa.false()))
    op.add_column('provider_details_history', sa.Column('supports_international', sa.Boolean(), nullable=False, server_default=sa.false()))

    op.execute("UPDATE provider_details SET supports_international=True WHERE identifier='sns'")
    op.execute("UPDATE provider_details_history SET supports_international=True WHERE identifier='sns'")


def downgrade():
    op.drop_column('provider_details_history', 'supports_international')
    op.drop_column('provider_details', 'supports_international')