"""create_new_project_history_actions

Revision ID: c7c68325743c
Revises: e2cc39f66d0c
Create Date: 2022-12-28 12:15:26.114789

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

# revision identifiers, used by Alembic.
revision = 'c7c68325743c'
down_revision = '298f2a90b66b'
branch_labels = None
depends_on = None


def upgrade():
    op.add_column('ProjectHistoryItem', sa.Column('media_plan_id', sa.BigInteger(), nullable=True))
    bind = op.get_bind()
    bind.execute(
        """ALTER TABLE "ProjectHistoryItem" ADD CONSTRAINT "fk_ProjectHistoryItem_media_plan_id_MediaPlan"
            FOREIGN KEY ("media_plan_id") REFERENCES "public"."MediaPlan"("id") ON DELETE CASCADE;
        """)
    bind.execute("""
                INSERT INTO "ProjectHistoryAction"(id, type)
                VALUES
                        (35, 'add_media_plan'),
                        (36, 'remove_media_plan'),
                        (37, 'update_media_plan_name'),
                        (38, 'update_media_plan_order'),
                        (39, 'update_phase_name'),
                        (40, 'created_request_approval'),
                        (41, 'submit_approval'),
                        (42, 'update_campaign')
            """)


def downgrade():
    op.drop_constraint(op.f('fk_ProjectHistoryItem_media_plan_id_MediaPlan'), 'ProjectHistoryItem', type_='foreignkey')
    op.drop_column('ProjectHistoryItem', 'media_plan_id')
    bind = op.get_bind()
    bind.execute("""DELETE FROM "ProjectHistoryItem" WHERE action_id BETWEEN 35 AND 42""")
    bind.execute("""DELETE FROM "ProjectHistoryAction" WHERE id BETWEEN 35 AND 42""")