mirror of
https://github.com/GSA/notifications-api.git
synced 2026-09-11 10:28:55 -04:00
reformat
This commit is contained in:
@@ -3,6 +3,7 @@ import os
|
||||
from sqlalchemy import text
|
||||
|
||||
from app import config
|
||||
|
||||
"""
|
||||
|
||||
Revision ID: 0133_set_services_sms_prefix
|
||||
@@ -14,36 +15,46 @@ 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'
|
||||
revision = "0133_set_services_sms_prefix"
|
||||
down_revision = "0132_add_sms_prefix_setting"
|
||||
|
||||
|
||||
config = config.configs[os.environ['NOTIFY_ENVIRONMENT']]
|
||||
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("""
|
||||
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("""
|
||||
"""
|
||||
),
|
||||
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)
|
||||
"""
|
||||
),
|
||||
input_params,
|
||||
)
|
||||
|
||||
|
||||
def downgrade():
|
||||
op.execute("""
|
||||
op.execute(
|
||||
"""
|
||||
UPDATE services set prefix_sms = null
|
||||
""")
|
||||
"""
|
||||
)
|
||||
|
||||
Reference in New Issue
Block a user