"""add_linkfire_integration_tables_views

Revision ID: e54ad31b2dab
Revises: ff131364a862
Create Date: 2021-05-31 10:08:55.843604

"""
from alembic import op
import sqlalchemy as sa
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.linkfire_project_trigger import LinkfireProjectTrigger


# revision identifiers, used by Alembic.
revision = 'e54ad31b2dab'
down_revision = 'ff131364a862'
branch_labels = None
depends_on = None


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

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

    TriggerMigrator(bind, LinkfireCampaignTrigger()).upgrade()
    TriggerMigrator(bind, LinkfireProjectTrigger()).upgrade()


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

    bind.execute('''DROP VIEW IF EXISTS linkfire_campaign_view;''')
    bind.execute('''DROP VIEW IF EXISTS linkfire_project_view;''')

    TriggerMigrator(bind, LinkfireCampaignTrigger()).rollback()
    TriggerMigrator(bind, LinkfireProjectTrigger()).rollback()

    bind.execute('''DROP TABLE IF EXISTS linkfire_campaign;''')
    bind.execute('''DROP TABLE IF EXISTS linkfire_project;''')