notifications-api/migrations/versions/0391_update_sms_numbers.py

"""

Revision ID: 0391_update_sms_numbers
Revises: 0390_drop_dvla_provider
Create Date: 2023-03-01 12:36:38.226954

"""
import sqlalchemy as sa
from alembic import op
from flask import current_app
from sqlalchemy import text

revision = "0391_update_sms_numbers"
down_revision = "0390_drop_dvla_provider"
OLD_SMS_NUMBER = "18446120782"
NEW_SMS_NUMBER = current_app.config["NOTIFY_INTERNATIONAL_SMS_SENDER"].strip("+")


def upgrade():
    conn = op.get_bind()
    op.alter_column(
        "service_sms_senders", "sms_sender", type_=sa.types.String(length=255)
    )
    op.alter_column("inbound_numbers", "number", type_=sa.types.String(length=255))
    input_params = {
        "new_sms_plus": f"+{NEW_SMS_NUMBER}",
        "old_sms_number": OLD_SMS_NUMBER,
        "new_sms_number": NEW_SMS_NUMBER,
    }
    conn.execute(
        text(
            "UPDATE service_sms_senders SET sms_sender = :new_sms_plus "
            "WHERE sms_sender IN (:old_sms_number, :new_sms_number)"
        ),
        input_params,
    )
    conn.execute(
        text(
            "UPDATE inbound_numbers SET number = :new_sms_plus "
            "WHERE number IN (:old_sms_number, :new_sms_number)"
        ),
        input_params,
    )


def downgrade():
    conn = op.get_bind()
    input_params = {
        "old_sms_number": OLD_SMS_NUMBER,
        "new_sms_plus": f"+{NEW_SMS_NUMBER}",
    }
    conn.execute(
        text(
            "UPDATE service_sms_senders SET sms_sender = :old_sms_number "
            "WHERE sms_sender = :new_sms_plus"
        ),
        input_params,
    )
    conn.execute(
        text(
            "UPDATE inbound_numbers SET number = :old_sms_number "
            "WHERE number = :new_sms_plus"
        ),
        input_params,
    )
    op.alter_column(
        "service_sms_senders", "sms_sender", type_=sa.types.String(length=11)
    )
    op.alter_column("inbound_numbers", "number", type_=sa.types.String(length=11))