Trakt

"""Sync watchlist request models for the Trakt MCP server.""" from typing import ClassVar from pydantic import BaseModel, ConfigDict, Field, model_validator from pydantic_core import PydanticCustomError from .base import TraktSyncWatchlistItem __all__ = ["TraktSyncWatchlistRequest"] class TraktSyncWatchlistRequest(BaseModel): """Request structure for POST/DELETE sync watchlist operations. Exactly one of movies, shows, seasons, or episodes must be provided and non-empty. """ # Names of the allowed collections FIELDS: ClassVar[tuple[str, ...]] = ("movies", "shows", "seasons", "episodes") # Reject unknown fields to avoid silent payload issues model_config: ClassVar[ConfigDict] = ConfigDict(extra="forbid") movies: list[TraktSyncWatchlistItem] | None = Field( default=None, min_length=1, description="Watchlist payload for movies" ) shows: list[TraktSyncWatchlistItem] | None = Field( default=None, min_length=1, description="Watchlist payload for shows" ) seasons: list[TraktSyncWatchlistItem] | None = Field( default=None, min_length=1, description="Watchlist payload for seasons" ) episodes: list[TraktSyncWatchlistItem] | None = Field( default=None, min_length=1, description="Watchlist payload for episodes" ) def _non_empty_lists(self) -> list[str]: return [name for name in self.FIELDS if getattr(self, name)] @model_validator(mode="after") def validate_exactly_one_list(self) -> "TraktSyncWatchlistRequest": """Validate that exactly one non-empty watchlist list is provided.""" present = self._non_empty_lists() if len(present) == 0: raise PydanticCustomError( "watchlist.collection_missing", "At least one watchlist list must be provided", ) if len(present) > 1: raise PydanticCustomError( "watchlist.multiple_collections", "Only one watchlist list allowed per request, got: {collections}", {"collections": ", ".join(present)}, ) return self