"""Article Pydantic schemas."""
from pydantic import BaseModel, ConfigDict, Field


class ArticleListParams(BaseModel):
    """Article list query parameters."""

    feed_id: str | None = None
    category: str | None = None
    tag: str | None = None
    search: str | None = None
    is_read: bool | None = None
    skip: int = 0
    limit: int = Field(default=50, le=200)


class ArticleOut(BaseModel):
    """Cleaned article output schema."""

    model_config = ConfigDict(from_attributes=True)

    id: str
    raw_article_id: str | None = None
    feed_id: str
    title: str | None = None
    link: str
    author: str | None = None
    feed_title: str | None = None
    feed_category: str | None = None
    published_at: str | None = None
    fetched_at: str
    content: str | None = None
    original_summary: str | None = None
    ai_summary: str | None = None
    category: str | None = None
    tags: list[str] = []
    heat_score: float = 0.0
    importance_score: float = 0.0
    duplication_score: float = 0.0
    composite_score: float = 0.0
    is_representative: bool = True
    reference_links: list[dict] = []
    processing_status: str = "pending"
    created_at: str
    updated_at: str

    @classmethod
    def model_validate(cls, obj):
        """Format datetime fields."""
        data = {}
        for key in obj.__dict__:
            value = getattr(obj, key)
            if key in ("created_at", "updated_at", "published_at", "fetched_at") and value is not None:
                data[key] = value.isoformat()
            else:
                data[key] = value
        return cls.model_construct(**data)