"""add_performance_breakdown_option_table

Revision ID: deca345a03c2
Revises: d97d6efcb37e
Create Date: 2020-11-04 08:47:19.362119

"""
from alembic import op
import sqlalchemy as sa

# revision identifiers, used by Alembic.
revision = 'deca345a03c2'
down_revision = 'd97d6efcb37e'
branch_labels = None
depends_on = None

breakdown_options = [
    {"name": "Country", "order": 1, "group": "Audience"},
    {"name": "Age", "order": 2, "group": "Audience"},
    {"name": "Gender", "order": 3, "group": "Audience"},
    {"name": "Age & Gender", "order": 4, "group": "Audience"},

    {"name": "Platform", "order": 5, "group": "Campaign Details"},
    {"name": "Provider", "order": 6, "group": "Campaign Details"},

    {"name": "Category", "order": 7, "group": "Category"},
    {"name": "Objective", "order": 8, "group": "Category"},
    {"name": "Goal", "order": 9, "group": "Category"},
    {"name": "Campaign Type", "order": 10, "group": "Category"},

    {"name": "Artist", "order": 11, "group": "Entity"},
    {"name": "Project", "order": 12, "group": "Entity"},
    {"name": "Campaign", "order": 13, "group": "Entity"},
    {"name": "Ad Set", "order": 14, "group": "Entity"},

    {"name": "Day", "order": 15, "group": "Time"},
    {"name": "Week", "order": 16, "group": "Time"},
    {"name": "Chartweek", "order": 17, "group": "Time"},
    {"name": "Month", "order": 18, "group": "Time"},
]


def upgrade():
    table = op.create_table(
        'PerformanceBreakdownOption',
        sa.Column('id', sa.Integer(), nullable=False),
        sa.Column('name', sa.String(), nullable=False),
        sa.Column('group', sa.String()),
        sa.Column('order', sa.Integer(), nullable=False),
        sa.PrimaryKeyConstraint('id', name=op.f('pk_PerformanceBreakdownOption'))
    )

    op.bulk_insert(table, breakdown_options)


def downgrade():
    op.drop_table('PerformanceBreakdownOption')