mirror of
https://github.com/GSA/notifications-api.git
synced 2025-12-10 15:22:24 -05:00
33 lines
861 B
Python
33 lines
861 B
Python
"""
|
|
|
|
Revision ID: 0261_service_volumes
|
|
Revises: 0260_remove_dvla_organisation
|
|
Create Date: 2019-02-13 13:45:00.782500
|
|
|
|
"""
|
|
from alembic import op
|
|
from itertools import product
|
|
import sqlalchemy as sa
|
|
|
|
|
|
revision = '0261_service_volumes'
|
|
down_revision = '0260_remove_dvla_organisation'
|
|
|
|
|
|
TABLES = ['services', 'services_history']
|
|
CHANNELS = ['volume_{}'.format(channel) for channel in ('email', 'letter', 'sms')]
|
|
|
|
|
|
def upgrade():
|
|
for table in TABLES:
|
|
op.add_column(table, sa.Column('consent_to_research', sa.Boolean(), nullable=False, server_default=sa.false()))
|
|
for channel in CHANNELS:
|
|
op.add_column(table, sa.Column(channel, sa.Integer(), nullable=True))
|
|
|
|
|
|
def downgrade():
|
|
for table in TABLES:
|
|
op.drop_column(table, 'consent_to_research')
|
|
for channel in CHANNELS:
|
|
op.drop_column(table, channel)
|