"""add test_category.modality (pathology/imaging work-queue routing)

Revision ID: 20260929_modality
Revises: 20260928_portalpin
Create Date: 2026-09-29 00:00:00.000000

"""

from collections.abc import Sequence

import sqlalchemy as sa
from alembic import op

revision: str = "20260929_modality"
down_revision: str | None = "20260928_portalpin"
branch_labels: str | Sequence[str] | None = None
depends_on: str | Sequence[str] | None = None


def upgrade() -> None:
    modality = sa.Enum("pathology", "imaging", name="category_modality")
    modality.create(op.get_bind(), checkfirst=True)
    op.add_column(
        "test_category",
        sa.Column("modality", modality, nullable=False, server_default="pathology"),
    )


def downgrade() -> None:
    op.drop_column("test_category", "modality")
    sa.Enum(name="category_modality").drop(op.get_bind(), checkfirst=True)