"""Task schemas."""

from pydantic import BaseModel, ConfigDict, Field, HttpUrl


class CreateTemplateTaskRequest(BaseModel):
    model_config = ConfigDict(title="Request model for creating a template task.")

    artist_urls: list[HttpUrl] = Field(
        default_factory=list,
        description="List of Spotify artist URLs.",
    )
    album_urls: list[HttpUrl] = Field(
        default_factory=list,
        description="List of Spotify album URLs.",
    )


class TemplateTaskStatusResponse(BaseModel):
    model_config = ConfigDict(title="Response model for template task status.")

    status: str = Field(..., description="The current status of the task.")
    download_url: HttpUrl | None = Field(
        default=None,
        description="URL to download the task result, if available.",
    )