"""new_columbia_uk_profit_center

Revision ID: e0835a6aa5b9
Revises: 5356915b3df4
Create Date: 2023-02-22 10:21:22.199938

"""
from alembic import op

from migrations.utils import get_table

# revision identifiers, used by Alembic.
revision = 'e0835a6aa5b9'
down_revision = '5356915b3df4'
branch_labels = None
depends_on = None

profit_center = {"code": "UK1160", "name": "Room Two Recordings", "label_id": 47}


def upgrade():
    bind = op.get_bind()
    profit_center_table = get_table(bind, "ProfitCenter")

    op.execute(
        profit_center_table.insert().values(
            code=profit_center.get("code"),
            name=profit_center.get("name"),
            label_id=profit_center.get("label_id")
        )
    )


def downgrade():
    bind = op.get_bind()
    profit_center_table = get_table(bind, "ProfitCenter")

    op.execute(
        profit_center_table.delete().where(
            profit_center_table.c.code == profit_center.get("code"),
            profit_center_table.c.name == profit_center.get("name"),
            profit_center_table.c.label_id == profit_center.get("label_id")
        )
    )