"""removing_old_campaign_statuses

Revision ID: 03e32671c45b
Revises: a2016e2b7112
Create Date: 2022-08-22 09:44:19.532108

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

# revision identifiers, used by Alembic.
revision = '03e32671c45b'
down_revision = '3fd263a27e38'
branch_labels = None
depends_on = None


def upgrade():
    bind = op.get_bind()
    bind.execute('ALTER TABLE "Campaign" DROP CONSTRAINT IF EXISTS fk_Campaign_campaign_status_id_CampaignStatus')
    op.drop_column('Campaign', 'campaign_status_id')
    op.drop_table('CampaignStatus')
    bind.execute('ALTER TABLE "Campaign" RENAME COLUMN approval_status TO workflow_status')


def downgrade():
    bind = op.get_bind()
    op.create_table('CampaignStatus',
                    sa.Column('id', sa.INTEGER(),
                              server_default=sa.text('nextval(\'"CampaignStatus_id_seq"\'::regclass)'),
                              autoincrement=True, nullable=False),
                    sa.Column('name', sa.VARCHAR(), autoincrement=False, nullable=False),
                    sa.PrimaryKeyConstraint('id', name='pk_CampaignStatus')
                    )
    bind.execute('''
                    INSERT INTO "CampaignStatus" (id, name)
                    VALUES  (1, 'Booked'),
                            (2, 'Live'),
                            (3, 'Completed'),
                            (4, 'Draft'),
                            (5, 'Pending Approval'),
                            (6, 'Approved'),
                            (7, 'Disapproved'),
                            (8, 'Paused')
                ''')
    op.add_column('Campaign', sa.Column('campaign_status_id', sa.INTEGER(), server_default=sa.text('4'), autoincrement=False, nullable=False))
    op.create_foreign_key('fk_Campaign_campaign_status_id_CampaignStatus', 'Campaign', 'CampaignStatus', ['campaign_status_id'], ['id'])

    bind = op.get_bind()
    bind.execute('ALTER TABLE "Campaign" RENAME COLUMN workflow_status TO approval_status')