notifications-api/migrations/versions/0133_set_services_sms_prefix.py

import os

from sqlalchemy import text

from app import config

"""

Revision ID: 0133_set_services_sms_prefix
Revises: 0132_add_sms_prefix_setting
Create Date: 2017-11-03 15:55:35.657488

"""
import sqlalchemy as sa
from alembic import op
from sqlalchemy.dialects import postgresql

revision = "0133_set_services_sms_prefix"
down_revision = "0132_add_sms_prefix_setting"


config = config.configs[os.environ["NOTIFY_ENVIRONMENT"]]
default_sms_sender = config.FROM_NUMBER


def upgrade():
    conn = op.get_bind()
    input_params = {"default_sms_sender": default_sms_sender}
    conn.execute(
        text(
            """
        update services set prefix_sms = True
        where id in (
            select service_id from service_sms_senders
            where is_default = True and sms_sender = :default_sms_sender
        )
    """
        ),
        input_params,
    )
    conn.execute(
        text(
            """
        update services set prefix_sms = False
        where id in (
            select service_id from service_sms_senders
            where is_default = True and sms_sender != :default_sms_sender
        )
    """
        ),
        input_params,
    )


def downgrade():
    op.execute(
        """
        UPDATE services set prefix_sms = null
    """
    )