mirror of
https://github.com/GSA/notifications-api.git
synced 2025-12-09 06:32:11 -05:00
41 lines
951 B
Python
41 lines
951 B
Python
"""
|
|
|
|
Revision ID: 0261_service_volumes
|
|
Revises: 0260_remove_dvla_organisation
|
|
Create Date: 2019-02-13 13:45:00.782500
|
|
|
|
"""
|
|
from itertools import product
|
|
|
|
import sqlalchemy as sa
|
|
from alembic import op
|
|
|
|
revision = "0261_service_volumes"
|
|
down_revision = "0260_remove_dvla_organisation"
|
|
|
|
|
|
TABLES = ["services", "services_history"]
|
|
CHANNELS = ["volume_email", "volume_letter", "volume_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)
|