"""

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)