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")