from __future__ import annotations

from typing import Literal

from pydantic import BaseModel


class RotationFilter(BaseModel):
    instance_id: str | None = None
    group_id: str | None = None
    status: Literal[
        "queued",
        "running",
        "needs_attention",
        "cleanup_pending",
        "succeeded",
        "failed",
        "cancelled",
    ] | None = None