"""change_campaign_statuses

Revision ID: 35304b84ca0d
Revises: e01444cb82f7
Create Date: 2022-05-25 10:52:53.526995

"""
from alembic import op
import sqlalchemy as sa
from migrations.fixtures.delphi_tables.v4.data_refresher import DelphiTablesDataRefresher


# revision identifiers, used by Alembic.
revision = '35304b84ca0d'
down_revision = 'e01444cb82f7'
branch_labels = None
depends_on = None


def upgrade():
    bind = op.get_bind()
    meta = sa.MetaData(bind=bind)
    meta.reflect()
    campaigns_table = meta.tables["Campaign"]

    delphi_refresher = DelphiTablesDataRefresher()
    delphi_refresher.drop_triggers(bind)

    op.alter_column('Campaign', 'campaign_status_id', server_default='1')
    op.execute(
        campaigns_table.update().values(
            campaign_status_id=1
        ).where(campaigns_table.c.campaign_status_id == 0)
    )

    bind.execute("""DELETE FROM "CampaignStatus" WHERE id = 0""")

    delphi_refresher.refresh(bind)


def downgrade():
    bind = op.get_bind()
    meta = sa.MetaData(bind=bind)
    meta.reflect()
    campaigns_table = meta.tables["Campaign"]

    op.execute(
        campaigns_table.insert().values(
            id=0,
            name="Planned",
        )
    )
    op.alter_column('Campaign', 'campaign_status_id', server_default='0')