mirror of
https://github.com/GSA/notifications-api.git
synced 2025-12-09 14:42:24 -05:00
99 lines
4.2 KiB
Python
99 lines
4.2 KiB
Python
"""empty message
|
|
|
|
Revision ID: 0011_ad_provider_details
|
|
Revises: 0010_events_table
|
|
Create Date: 2016-05-05 09:14:29.328841
|
|
|
|
"""
|
|
|
|
# revision identifiers, used by Alembic.
|
|
from sqlalchemy import text
|
|
|
|
revision = '0011_ad_provider_details'
|
|
down_revision = '0010_events_table'
|
|
|
|
import uuid
|
|
|
|
from alembic import op
|
|
import sqlalchemy as sa
|
|
from sqlalchemy.dialects import postgresql
|
|
|
|
def upgrade():
|
|
op.create_table('provider_details',
|
|
sa.Column('id', postgresql.UUID(as_uuid=True), nullable=False),
|
|
sa.Column('display_name', sa.String(), nullable=False),
|
|
sa.Column('identifier', sa.String(), nullable=False),
|
|
sa.Column('priority', sa.Integer(), nullable=False),
|
|
sa.Column('notification_type', sa.Enum('email', 'sms', 'letter', name='notification_type'), nullable=False),
|
|
sa.Column('active', sa.Boolean(), nullable=True),
|
|
sa.PrimaryKeyConstraint('id')
|
|
)
|
|
|
|
op.add_column('provider_rates', sa.Column('provider_id', postgresql.UUID(as_uuid=True), nullable=True))
|
|
op.create_index(op.f('ix_provider_rates_provider_id'), 'provider_rates', ['provider_id'], unique=False)
|
|
op.create_foreign_key("provider_rate_to_provider_fk", 'provider_rates', 'provider_details', ['provider_id'], ['id'])
|
|
op.add_column('provider_statistics', sa.Column('provider_id', postgresql.UUID(as_uuid=True), nullable=True))
|
|
op.create_index(op.f('ix_provider_statistics_provider_id'), 'provider_statistics', ['provider_id'], unique=False)
|
|
op.create_foreign_key('provider_stats_to_provider_fk', 'provider_statistics', 'provider_details', ['provider_id'], ['id'])
|
|
|
|
conn = op.get_bind()
|
|
input_params = {
|
|
"id": uuid.uuid4()
|
|
}
|
|
conn.execute(
|
|
text("INSERT INTO provider_details (id, display_name, identifier, priority, notification_type, active) values (:id, 'MMG', 'mmg', 10, 'sms', true)"), input_params
|
|
)
|
|
input_params = {
|
|
"id": uuid.uuid4()
|
|
}
|
|
conn.execute(
|
|
text("INSERT INTO provider_details (id, display_name, identifier, priority, notification_type, active) values (:id, 'Firetext', 'firetext', 20, 'sms', true)"), input_params
|
|
)
|
|
input_params = {
|
|
"id": uuid.uuid4()
|
|
}
|
|
conn.execute(
|
|
text("INSERT INTO provider_details (id, display_name, identifier, priority, notification_type, active) values (:id, 'AWS SES', 'ses', 10, 'email', true)"), input_params
|
|
)
|
|
input_params = {
|
|
"id": uuid.uuid4()
|
|
}
|
|
conn.execute(
|
|
text("INSERT INTO provider_details (id, display_name, identifier, priority, notification_type, active) values (:id, 'AWS SNS', 'sns', 10, 'sms', true)"), input_params
|
|
)
|
|
op.execute(
|
|
"UPDATE provider_rates set provider_id = (select id from provider_details where identifier = 'mmg') where provider = 'mmg'"
|
|
)
|
|
op.execute(
|
|
"UPDATE provider_rates set provider_id = (select id from provider_details where identifier = 'firetext') where provider = 'firetext'"
|
|
)
|
|
op.execute(
|
|
"UPDATE provider_rates set provider_id = (select id from provider_details where identifier = 'ses') where provider = 'ses'"
|
|
)
|
|
op.execute(
|
|
"UPDATE provider_rates set provider_id = (select id from provider_details where identifier = 'sns') where provider = 'sns'"
|
|
)
|
|
op.execute(
|
|
"UPDATE provider_statistics set provider_id = (select id from provider_details where identifier = 'mmg') where provider = 'mmg'"
|
|
)
|
|
op.execute(
|
|
"UPDATE provider_statistics set provider_id = (select id from provider_details where identifier = 'firetext') where provider = 'firetext'"
|
|
)
|
|
op.execute(
|
|
"UPDATE provider_statistics set provider_id = (select id from provider_details where identifier = 'ses') where provider = 'ses'"
|
|
)
|
|
op.execute(
|
|
"UPDATE provider_statistics set provider_id = (select id from provider_details where identifier = 'sns') where provider = 'sns'"
|
|
)
|
|
|
|
|
|
def downgrade():
|
|
|
|
op.drop_index(op.f('ix_provider_statistics_provider_id'), table_name='provider_statistics')
|
|
op.drop_column('provider_statistics', 'provider_id')
|
|
op.drop_index(op.f('ix_provider_rates_provider_id'), table_name='provider_rates')
|
|
op.drop_column('provider_rates', 'provider_id')
|
|
|
|
op.drop_table('provider_details')
|
|
op.execute('drop type notification_type')
|