"""add_ceremony_of_roses_label

Revision ID: 711fbe1fddaf
Revises: 080bd9f14a21
Create Date: 2023-03-30 08:20:47.918895

"""
from alembic import op
import sqlalchemy as sa
from sqlalchemy.dialects import postgresql

from migrations.utils import get_table

# revision identifiers, used by Alembic.
revision = '711fbe1fddaf'
down_revision = '080bd9f14a21'
branch_labels = None
depends_on = None

SME_UK_LIMITED_ID = 11

ceremony_of_roses = {
    "id": 73,
    "name": "Ceremony of Roses UK",
    "external_name": "Ceremony of Roses UK",
    "country": "UK",
    "abbreviation": "COR",
}


def upgrade():
    bind = op.get_bind()
    labels_table = get_table(bind, "Label")
    profit_centers_table = get_table(bind, "ProfitCenter")
    projects_table = get_table(bind, "Project")

    op.execute(labels_table.insert().values(**ceremony_of_roses))

    op.execute(
        profit_centers_table.update().where(profit_centers_table.c.code == "UK1675")
        .values(label_id=ceremony_of_roses["id"])
    )

    op.execute(
        projects_table.update().where(projects_table.c.gras_project_code == "1511737")
        .values(label_id=ceremony_of_roses["id"])
    )


def downgrade():
    bind = op.get_bind()
    labels_table = get_table(bind, "Label")
    profit_centers_table = get_table(bind, "ProfitCenter")
    projects_table = get_table(bind, "Project")

    op.execute(
        projects_table.update().where(projects_table.c.gras_project_code == "1511737")
        .values(label_id=SME_UK_LIMITED_ID)
    )

    op.execute(
        profit_centers_table.update().where(profit_centers_table.c.code == "UK1675")
        .values(label_id=SME_UK_LIMITED_ID)
    )

    op.execute(labels_table.delete().where(labels_table.c.id == ceremony_of_roses["id"]))