"""add deleted environment_role status

Revision ID: 1497926ddec1
Revises: e3d93f9caba7
Create Date: 2019-10-04 10:44:54.198368

"""
from alembic import op
import sqlalchemy as sa


# revision identifiers, used by Alembic.
revision = "1497926ddec1"  # pragma: allowlist secret
down_revision = "e3d93f9caba7"  # pragma: allowlist secret
branch_labels = None
depends_on = None


def upgrade():
    # op.alter_column("environment_roles", "status", new_column_name="status_old")
    op.alter_column(
        "environment_roles",
        "status",
        type_=sa.Enum(
            "PENDING",
            "COMPLETED",
            "PENDING_DELETE",
            "DELETED",
            name="status",
            native_enum=False,
        ),
        existing_type=sa.Enum(
            "PENDING",
            "COMPLETED",
            "PENDING_DELETE",
            name="status",
            native_enum=False,
        ),
    )

    # conn = op.get_bind()
    # conn.execute("UPDATE environment_roles SET status = status_old")
    # op.drop_column("environment_roles", "status_old")


def downgrade():
    conn = op.get_bind()
    conn.execute(
        """
        UPDATE environment_roles
        SET status = (CASE WHEN status = 'DELETED' THEN 'PENDING_DELETE' ELSE status END)
        """
    )

    op.alter_column(
        "environment_roles",
        "status",
        type_=sa.Enum(
            "PENDING",
            "COMPLETED",
            "PENDING_DELETE",
            name="status",
            native_enum=False,
        ),
        existing_type=sa.Enum(
            "PENDING",
            "COMPLETED",
            "PENDING_DELETE",
            "DELETED",
            name="status",
            native_enum=False,
        ),
    )