mirror of
https://github.com/GSA/notifications-api.git
synced 2025-12-15 17:52:26 -05:00
50 lines
1.2 KiB
Python
50 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
|
|
|
|
"""
|
|
from alembic import op
|
|
import sqlalchemy as sa
|
|
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
|
|
""")
|