notifications-api/migrations/versions/0140_sms_prefix_non_nullable.py

"""

Revision ID: 0140_sms_prefix_non_nullable
Revises: 0139_migrate_sms_allowance_data
Create Date: 2017-11-07 13:04:04.077142

"""
import sqlalchemy as sa
from alembic import op
from flask import current_app
from sqlalchemy import text
from sqlalchemy.dialects import postgresql

revision = "0140_sms_prefix_non_nullable"
down_revision = "0139_migrate_sms_allowance_data"


def upgrade():
    conn = op.get_bind()
    input_params = {"id": current_app.config["NOTIFY_SERVICE_ID"]}
    conn.execute(
        text(
            """
        update services
        set prefix_sms = false
        where id = :id
    """
        ),
        input_params,
    )

    op.alter_column(
        "services",
        "prefix_sms",
        existing_type=sa.BOOLEAN(),
        nullable=False,
    )


def downgrade():
    op.alter_column(
        "services",
        "prefix_sms",
        existing_type=sa.BOOLEAN(),
        nullable=True,
    )

    conn = op.get_bind()
    input_params = {"id": current_app.config["NOTIFY_SERVICE_ID"]}
    conn.execute(
        text(
            """
        update services
        set prefix_sms = null
        where id = :id
    """
        ),
        input_params,
    )