"""add_CMG_accounts

Revision ID: 5893e1fbcf45
Revises: 5508ac229aca
Create Date: 2021-03-04 11:09:35.809511

"""
from alembic import op
import sqlalchemy as sa


# revision identifiers, used by Alembic.
revision = '5893e1fbcf45'
down_revision = '5508ac229aca'
branch_labels = None
depends_on = None

marketing_accounts = [
    {"label_id": 44, "provider_id": 1, "type": 1, "external_id": 768164510232785},
    {"label_id": 44, "provider_id": 1, "type": 1, "external_id": 562408950939930},
    {"label_id": 44, "provider_id": 1, "type": 1, "external_id": 403395130429589},
]


def upgrade():
    bind = op.get_bind()
    meta = sa.MetaData(bind=bind)
    meta.reflect()
    marketing_account_table = meta.tables["MarketingAccount"]

    op.execute(f"""ALTER SEQUENCE "MarketingAccount_id_seq" RESTART WITH 56""")

    for account in marketing_accounts:
        op.execute(
            marketing_account_table.insert().values(**account)
        )


def downgrade():
    bind = op.get_bind()
    meta = sa.MetaData(bind=bind)
    meta.reflect()
    marketing_account_table = meta.tables["MarketingAccount"]

    for account in marketing_accounts:
        op.execute(
            marketing_account_table.delete().where(marketing_account_table.c.external_id == account["external_id"])
        )