mirror of
https://github.com/GSA/notifications-api.git
synced 2026-08-01 12:18:50 -04:00
49 lines
1.7 KiB
Python
49 lines
1.7 KiB
Python
"""
|
|
|
|
Revision ID: 0348_migrate_broadcast_settings
|
|
Revises: 0347_add_dvla_volumes_template
|
|
Create Date: 2021-02-18 15:25:30.667098
|
|
|
|
"""
|
|
from alembic import op
|
|
import sqlalchemy as sa
|
|
from sqlalchemy.dialects import postgresql
|
|
|
|
revision = '0348_migrate_broadcast_settings'
|
|
down_revision = '0347_add_dvla_volumes_template'
|
|
|
|
|
|
def upgrade():
|
|
# For every service that has the broadcast permission we want it to have
|
|
# a row in the broadcast_service_settings table
|
|
#
|
|
# If it doesnt have a row already, then:
|
|
# - if the service is in trial mode, add a row and set the channel as 'severe'
|
|
# - if the service is in live mode, add a row and set the channel as 'test'
|
|
#
|
|
# If it does have a row already no action needed
|
|
conn = op.get_bind()
|
|
|
|
find_services_sql = """
|
|
SELECT services.id, services.restricted
|
|
FROM services
|
|
LEFT JOIN service_permissions
|
|
ON services.id = service_permissions.service_id
|
|
WHERE service_permissions.permission = 'broadcast'
|
|
"""
|
|
|
|
services = conn.execute(find_services_sql)
|
|
for service in services:
|
|
setting = conn.execute(f"SELECT service_id, channel, provider FROM service_broadcast_settings WHERE service_id = '{service.id}';").first()
|
|
if setting:
|
|
print(f"Service {service.id} already has service_broadcast_settings. No action required")
|
|
else:
|
|
channel = "severe" if service.restricted else "test"
|
|
print(f"Service {service.id} does not have service_broadcast_settings. Will insert one with channel {channel}")
|
|
conn.execute(f"INSERT INTO service_broadcast_settings (service_id, channel, created_at) VALUES ('{service.id}', '{channel}', now());")
|
|
|
|
|
|
def downgrade():
|
|
# No downgrade as we do not know what the state of the table was before that it should return to
|
|
pass
|