"""add_tiktok_marketing_accounts

Revision ID: dd2ee98d45e4
Revises: cecc9dfc497c
Create Date: 2022-02-21 10:43:42.751604

"""
from alembic import op
import sqlalchemy as sa


# revision identifiers, used by Alembic.
revision = 'dd2ee98d45e4'
down_revision = 'cecc9dfc497c'
branch_labels = None
depends_on = None

tiktok_marketing_accounts = [

    # Label - SME UK; Provider - 7Stars
    {
        "label_id": 11,
        "provider_id": 3,
        "accounts_info": [
            {"id": 7018159350493528066, "name": "Relentless Records UK", "currency": "GBP"},
            {"id": 7018158931528712193, "name": "MagicStar UK", "currency": "GBP"},
            {"id": 7018159196784852994, "name": "Music For Nations UK", "currency": "GBP"},
            {"id": 7021106214905888769, "name": "WEAREBLK UK", "currency": "GBP"},
            {"id": 7018159104283656193, "name": "Ministry of Sound UK", "currency": "GBP"},
            {"id": 7018158620722380802, "name": "Commercial Group UK", "currency": "GBP"},
            {"id": 7018158947211165697, "name": "Masterworks UK", "currency": "GBP"},
            {"id": 7018158390929047554, "name": "Black Butter Records UK", "currency": "GBP"},
            {"id": 7018158777199247362, "name": "Dreamlife Records UK", "currency": "GBP"},
            {"id": 7018159444684980225, "name": "Since '93 UK", "currency": "GBP"},
            {"id": 7018158817217150978, "name": "Insanity UK", "currency": "GBP"},
            {"id": 7018159277405208578, "name": "RCA UK", "currency": "GBP"},
            {"id": 7021106262922231809, "name": "5K Records UK", "currency": "GBP"},
            {"id": 7018158514702974978, "name": "Columbia Records UK", "currency": "GBP"}
        ]
    },

    # Label - SME US; Provider - Sony Music
    {
        "label_id": 31,
        "provider_id": 1,
        "accounts_info": [
            {"id": 7004722611523469313, "name": "SME Germany - H2 2021", "currency": "EUR"},
            {"id": 6861887201337147397, "name": "Sony Music Entertainment Germany - Auction Account", "currency": "EUR"},
            {"id": 7025666915741794306, "name": "SME Germany - Classical International", "currency": "EUR"},
            {"id": 7025976193434288130, "name": "SME Germany - Pixel", "currency": "EUR"},
        ]
    },
    {
        "label_id": 5,
        "provider_id": 1,
        "accounts_info": [
            {"id": 6803037823101304838, "name": "Sony Music_Nashville Auction", "currency": "USD"},
        ]
    },
    {
        "label_id": 1,
        "provider_id": 1,
        "accounts_info": [
            {"id": 6932184346828767234, "name": "Columbia Records (USD)", "currency": "USD"},
        ]
    },
    {
        "label_id": 2,
        "provider_id": 1,
        "accounts_info": [
            {"id": 6932193654383575042, "name": "Epic Records (USD)", "currency": "USD"},
        ]
    },
    {
        "label_id": 2,
        "provider_id": 1,
        "accounts_info": [
            {"id": 7002697221502156802, "name": "Sony Music | Peru | Ad Credit", "currency": "USD"},
        ]
    }
]


def upgrade():
    # ### commands auto generated by Alembic - please adjust! ###
    bind = op.get_bind()
    meta = sa.MetaData(bind=bind)
    meta.reflect()
    marketing_account_table = meta.tables["MarketingAccount"]
    for account in tiktok_marketing_accounts:
        for account_info in account["accounts_info"]:
            op.execute(
                marketing_account_table.insert().values(
                    label_id=account["label_id"],
                    provider_id=account["provider_id"],
                    external_id=account_info["id"],
                    name=account_info["name"],
                    currency=account_info["currency"],
                    source='tiktok'
                )
            )
    # ### end Alembic commands ###


def downgrade():
    # ### commands auto generated by Alembic - please adjust! ###
    op.execute(
        """
        DELETE FROM "MarketingAccount" WHERE source= 'tiktok';
        """
    )
    # ### end Alembic commands ###