mirror of
https://github.com/GSA/notifications-api.git
synced 2025-12-12 08:12:27 -05:00
35ffb048e5
Concourse had passed the tests before someone elses migration had beaten me to getting merged. So I merged and got an error. This should fix it
40 lines
1.2 KiB
Python
40 lines
1.2 KiB
Python
"""
|
|
|
|
Revision ID: 0345_move_broadcast_provider
|
|
Revises: 0344_stubbed_not_nullable
|
|
Create Date: 2021-02-09 09:19:07.957980
|
|
|
|
"""
|
|
from alembic import op
|
|
import sqlalchemy as sa
|
|
from sqlalchemy.dialects import postgresql
|
|
|
|
revision = '0345_move_broadcast_provider'
|
|
down_revision = '0344_stubbed_not_nullable'
|
|
|
|
|
|
def upgrade():
|
|
op.add_column('service_broadcast_settings', sa.Column('provider', sa.String(), nullable=True))
|
|
|
|
sql = """
|
|
select service_id, provider
|
|
from service_broadcast_provider_restriction
|
|
where service_id NOT IN (select service_id from service_broadcast_settings)
|
|
"""
|
|
insert_sql = """
|
|
insert into service_broadcast_settings(service_id, channel, provider, created_at, updated_at)
|
|
values('{}', 'test', '{}', now(), null)
|
|
"""
|
|
conn = op.get_bind()
|
|
results = conn.execute(sql)
|
|
restrictions = results.fetchall()
|
|
for x in restrictions:
|
|
f = insert_sql.format(x.service_id, x.provider)
|
|
conn.execute(f)
|
|
|
|
|
|
def downgrade():
|
|
# Downgrade does not try and fully undo the upgrade, in particular it does not
|
|
# delete the rows added to the service_broadcast_settings table
|
|
op.drop_column('service_broadcast_settings', 'provider')
|