"""update-campaign-trigger

Revision ID: d7e93c06f7d9
Revises: 413480848b31
Create Date: 2021-06-16 10:36:00.103525

"""
from alembic import op
import sqlalchemy as sa
from sqlalchemy.dialects import postgresql
from alembic import op

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 = 'd7e93c06f7d9'
down_revision = '413480848b31'
branch_labels = None
depends_on = None


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

    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()

    TriggerMigrator(bind, CampaignTrigger()).rollback()
    TriggerMigrator(bind, CampaignTrigger()).upgrade()
    TriggerMigrator(bind, LinkfireCampaignTrigger()).rollback()
    TriggerMigrator(bind, LinkfireCampaignTrigger()).upgrade()

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