"""remove_campaign_link

Revision ID: qfep3td233b2
Revises: ffef3bd223a1
Create Date: 2021-02-03 14:15:12.946442

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

# revision identifiers, used by Alembic.
revision = 'qfep3td233b2'
down_revision = 'ffef3bd223a1'
branch_labels = None
depends_on = None


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

    op.drop_column('Campaign', 'campaign_link')
    op.drop_column('CampaignAd', 'campaign_link')

    bind.execute("""
        UPDATE "Campaign"
        SET metainfo = jsonb_set(metainfo::jsonb, '{fields,spend}', 'false', FALSE)
        WHERE external_id IS NOT NULL
    """)


def downgrade():
    op.add_column('Campaign', sa.Column('campaign_link', postgresql.VARCHAR, nullable=True))