"""add_uncategorised_category

Revision ID: 7e5d1f68c125
Revises: dcfb5b5b0a09
Create Date: 2021-02-18 10:02:40.931398

"""
from alembic import op
import sqlalchemy as sa
from sqlalchemy import orm

# revision identifiers, used by Alembic.
revision = '7e5d1f68c125'
down_revision = '8fb30f22406d'
branch_labels = None
depends_on = None


def upgrade():

    op.add_column('CampaignTypeValue', sa.Column('is_hidden', sa.Boolean(), nullable=False, server_default="false"))
    op.add_column('CampaignTypeGroup', sa.Column('is_hidden', sa.Boolean(), nullable=False, server_default="false"))

    bind = op.get_bind()
    meta = sa.MetaData(bind=bind)
    meta.reflect()

    campaign_type_group_table = meta.tables["CampaignTypeGroup"]
    campaign_type_value_table = meta.tables["CampaignTypeValue"]

    op.execute(
        campaign_type_group_table.insert().values(
            id=4,
            name="Uncategorized",
            is_hidden=True
        )
    )

    op.execute(
        campaign_type_value_table.insert().values(
            id=236,
            name='Other Digital',
            category_id=1,
            group_id=1,
            is_hidden=True
        )
    )
    op.execute(
        campaign_type_value_table.insert().values(
            id=237,
            name='Other Traditional',
            category_id=1,
            group_id=2,
            is_hidden=True
        )
    )
    op.execute(
        campaign_type_value_table.insert().values(
            id=238,
            name='Other Radio',
            category_id=1,
            group_id=3,
            is_hidden=True
        )
    )
    op.execute(
        campaign_type_value_table.insert().values(
            id=239,
            name='Other Uncategorized',
            category_id=1,
            group_id=4,
            is_hidden=True
        )
    )


def downgrade():
    bind = op.get_bind()

    op.drop_column('CampaignTypeValue', 'is_hidden')
    op.drop_column('CampaignTypeGroup', 'is_hidden')

    bind.execute(''' DELETE FROM "CampaignTypeValue" WHERE id in (236, 237, 238, 239)''')
    bind.execute(''' DELETE FROM "CampaignTypeGroup" WHERE id = 4''')