mirror of
https://github.com/GSA/notifications-api.git
synced 2026-05-02 07:10:14 -04:00
55 lines
1.2 KiB
Python
55 lines
1.2 KiB
Python
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
|
|
""")
|