"""add_external_field_to_platforms

Revision ID: 10603462e93f
Revises: 78663da4a8cb
Create Date: 2021-10-06 10:17:25.727877

"""
from alembic import op
import sqlalchemy as sa


# revision identifiers, used by Alembic.
revision = '10603462e93f'
down_revision = '78663da4a8cb'
branch_labels = None
depends_on = None


def upgrade():
    op.add_column(
        'CampaignTypeValue',
        sa.Column('is_external_platform', sa.Boolean(), server_default='FALSE', nullable=False)
    )

    bind = op.get_bind()
    meta = sa.MetaData(bind=bind)
    meta.reflect()
    campaign_type_value = meta.tables["CampaignTypeValue"]

    op.execute(
        campaign_type_value.update().values(
            is_external_platform=True
        ).where(campaign_type_value.c.name.in_([
            'Google',
            'Facebook',
            'YouTube',
            'Instagram',
            'Audience Network',
            'Messenger',
            'Google Search',
            'Google Display'
        ]))
    )


def downgrade():
    op.drop_column('CampaignTypeValue', 'is_external_platform')