"""fix_projects_related_triggers

Revision ID: ce5be15884af
Revises: 8804ac239c9d
Create Date: 2021-05-18 08:35:57.330555

"""
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.project_trigger import ProjectTrigger
from migrations.fixtures.delphi_tables.v1.triggers.project_target_item_trigger import ProjectTargetItemTrigger
from migrations.fixtures.delphi_tables.v1.triggers.project_campaign_trigger import ProjectCampaignTrigger


# revision identifiers, used by Alembic.
revision = 'ce5be15884af'
down_revision = '8804ac239c9d'
branch_labels = None
depends_on = None


def upgrade():
    # ### commands auto generated by Alembic - please adjust! ###
    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 [ProjectTrigger, ProjectTargetItemTrigger, ProjectCampaignTrigger]:
        TriggerMigrator(bind, klass()).rollback()
        TriggerMigrator(bind, klass()).upgrade()

    # ### end Alembic commands ###