"""add_manual_campaigns_to_campaigns_tables

Revision ID: 2302afc25984
Revises: 51369b83c7fa
Create Date: 2021-06-08 09:24:06.571961

"""
from alembic import op
import sqlalchemy as sa
from migrations.fixtures.functions.campaign_dsp_id_function import CampaignDspIdFunction
from migrations.fixtures.delphi_tables.views import DelphiViews
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.linkfire_campaign_trigger import LinkfireCampaignTrigger
from migrations.fixtures.delphi_tables.v1.triggers.campaign_trigger import CampaignTrigger


# revision identifiers, used by Alembic.
revision = '2302afc25984'
down_revision = 'ad7fa901b370'
branch_labels = None
depends_on = None


def upgrade():
    bind = op.get_bind()
    bind.execute(CampaignDspIdFunction.create_or_replace())

    delphi_views = DelphiViews(bind)
    delphi_views.drop()
    delphi_views.create()
    bind.execute(delphi_views.linkfire_campaign_view())
    bind.execute(delphi_views.linkfire_project_view())
    delphi_tables = DelphiTables(bind)
    delphi_tables.drop()
    delphi_tables.create()

    for klass in [CampaignTrigger, LinkfireCampaignTrigger]:
        TriggerMigrator(bind, klass()).rollback()
        TriggerMigrator(bind, klass()).upgrade()

    data_migrator = DataMigrator(bind)
    data_migrator.execute()
    data_migrator.refresh_linkfire_tables()
    delphi_tables.reindex()