"""add_triggers_to_delphi_tables

Revision ID: 4a7d8076c573
Revises: f5b33a0b627d
Create Date: 2021-04-19 12:58:19.041637

"""
from alembic import op

from migrations.fixtures.delphi_tables.v1.tables import DelphiTables
from migrations.fixtures.delphi_tables.v1.data_migrator import DataMigrator

from migrations.fixtures.delphi_tables.v1.trigger_migrator import TriggerMigrator
from migrations.fixtures.delphi_tables.v1.triggers.label_trigger import LabelTrigger
from migrations.fixtures.delphi_tables.v1.triggers.campaign_type_group_trigger import CampaignTypeGroupTrigger
from migrations.fixtures.delphi_tables.v1.triggers.campaign_type_value_trigger import CampaignTypeValueTrigger
from migrations.fixtures.delphi_tables.v1.triggers.campaign_objective_trigger import CampaignObjectiveTrigger
from migrations.fixtures.delphi_tables.v1.triggers.campaign_provider_trigger import CampaignProviderTrigger
from migrations.fixtures.delphi_tables.v1.triggers.user_label_trigger import UserLabelTrigger
from migrations.fixtures.delphi_tables.v1.triggers.project_trigger import ProjectTrigger
from migrations.fixtures.delphi_tables.v1.triggers.project_phase_trigger import ProjectPhaseTrigger
from migrations.fixtures.delphi_tables.v1.triggers.project_target_item_trigger import ProjectTargetItemTrigger
from migrations.fixtures.delphi_tables.v1.triggers.user_project_trigger import UserProjectTrigger
from migrations.fixtures.delphi_tables.v1.triggers.campaign_trigger import CampaignTrigger
from migrations.fixtures.delphi_tables.v1.triggers.project_campaign_trigger import ProjectCampaignTrigger
from migrations.fixtures.delphi_tables.v1.triggers.campaign_type_trigger import CampaignTypeTrigger


# revision identifiers, used by Alembic.
revision = '4a7d8076c573'
down_revision = '43825f7bb6cc'
branch_labels = None
depends_on = None

triggers = [
    LabelTrigger,
    CampaignTypeGroupTrigger,
    CampaignTypeValueTrigger,
    CampaignObjectiveTrigger,
    CampaignProviderTrigger,
    CampaignTypeTrigger,
    UserLabelTrigger,
    ProjectTrigger,
    ProjectPhaseTrigger,
    ProjectTargetItemTrigger,
    UserProjectTrigger,
    CampaignTrigger,
    ProjectCampaignTrigger,
]

def upgrade():
    bind = op.get_bind()
    delphi_tables = DelphiTables(bind)
    data_migrator = DataMigrator(bind)
    delphi_tables.drop()
    delphi_tables.create()
    data_migrator.execute()
    delphi_tables.reindex()

    for klass in triggers:
        TriggerMigrator(bind, klass()).upgrade()


def downgrade():
    bind = op.get_bind()
    for klass in triggers:
        TriggerMigrator(bind, klass()).rollback()