"""add_accounts_to_labels

Revision ID: ab8aae6065b1
Revises: e8ee3046e854
Create Date: 2020-10-09 11:36:32.080923

"""
from alembic import op
import sqlalchemy as sa


# revision identifiers, used by Alembic.
revision = 'ab8aae6065b1'
down_revision = 'e8ee3046e854'
branch_labels = None
depends_on = None

labels_marketing_accounts = [
    {"id": 1, "external_id": 8653387137}, # Columbia
    {"id": 2, "external_id": 2333825793}, # Epic
    {"id": 4, "external_id": 6449900252}, # Legacy
    {"id": 6, "external_id": 3029121817}, # SME US Latin
    {"id": 6, "external_id": 4732633891},  # SME US Latin
    {"id": 6, "external_id": 3098880474},  # SME US Latin
    {"id": 7, "external_id": 6483554878}, # Arista Records
    {"id": 13, "external_id": 2413196617},  # SME Mexico
    {"id": 23, "external_id": 2802154940},  # SME Portugal
    {"id": 19, "external_id": 9963444288},  # SME Argentina
    {"id": 9, "external_id": 1073652073},  # SME Brasil
    {"id": 42, "external_id": 5176914881},  # SME España

]

def upgrade():
    op.create_table('MarketingAccount',
        sa.Column('id', sa.Integer(), nullable=False),
        sa.Column('label_id', sa.Integer(), nullable=False),
        sa.Column('type', sa.Integer(), server_default='0', nullable=False),
        sa.Column('external_id', sa.String(), nullable=False),
        sa.Column('provider_id', sa.Integer(), nullable=False, server_default='1'),
        sa.ForeignKeyConstraint(['label_id'], ['Label.id'], name=op.f('fk_MarketingAccount_label_id_Label')),
        sa.ForeignKeyConstraint(['provider_id'], ['CampaignProvider.id'], name=op.f('fk_MarketingAccount_provider_id_CampaignProvider')),
        sa.PrimaryKeyConstraint('id', name=op.f('pk_MarketingAccount')),
        sa.UniqueConstraint('external_id', name=op.f('uq_MarketingAccount_external_id'))
    )
    bind = op.get_bind()
    meta = sa.MetaData(bind=bind)
    meta.reflect()
    marketing_account_table = meta.tables["MarketingAccount"]
    for account in labels_marketing_accounts:
        op.execute(
            marketing_account_table.insert().values(
                label_id=account["id"],
                external_id=account["external_id"],
            )
        )


def downgrade():
    op.drop_table('MarketingAccount')