mirror of
https://github.com/GSA/notifications-api.git
synced 2025-12-10 07:12:20 -05:00
70 lines
1.9 KiB
Python
70 lines
1.9 KiB
Python
"""
|
|
|
|
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))
|