"""Bénévole : retrait de tshirt_size et diet

Revision ID: 0005_drop_volunteer_tshirt_diet
Revises: 0004_shift_day_part
Create Date: 2026-09-10

"""

from collections.abc import Sequence

import sqlalchemy as sa
from alembic import op

revision: str = "0005_drop_volunteer_tshirt_diet"
down_revision: str | None = "0004_shift_day_part"
branch_labels: str | Sequence[str] | None = None
depends_on: str | Sequence[str] | None = None


_TSHIRT_SIZE = sa.Enum(
    "XS", "S", "M", "L", "XL", "XXL", "XXXL",
    name="tshirtsize", native_enum=False,
)


def upgrade() -> None:
    op.drop_column("volunteer", "tshirt_size")
    op.drop_column("volunteer", "diet")


def downgrade() -> None:
    op.add_column(
        "volunteer", sa.Column("diet", sa.String(), nullable=True)
    )
    op.add_column(
        "volunteer",
        sa.Column("tshirt_size", _TSHIRT_SIZE, nullable=True),
    )