diary-news/backend/app/schemas/source.py

"""Source schemas."""
from __future__ import annotations

from datetime import datetime

from pydantic import BaseModel, ConfigDict, Field, HttpUrl

from app.models.source import SourceKind


class SourceOut(BaseModel):
    model_config = ConfigDict(from_attributes=True)

    id: int
    name: str
    slug: str
    kind: SourceKind
    url: str
    enabled: bool
    region: str | None = None
    language_src: str | None = None
    priority: int
    fetch_interval_min: int
    translate_to: str
    last_fetched_at: datetime | None = None
    last_status: str | None = None
    consecutive_failures: int = 0
    # 源级屏蔽分类标签;与 llm_settings.blocklist_tags 合并后注入 classify prompt
    blocklist_tags: list[str] = []


class SourceIn(BaseModel):
    name: str = Field(min_length=1, max_length=128)
    slug: str = Field(min_length=1, max_length=128, pattern=r"^[a-z0-9-]+$")
    kind: SourceKind = SourceKind.RSS
    url: HttpUrl
    region: str | None = None
    language_src: str | None = None
    priority: int = Field(default=50, ge=1, le=100)
    fetch_interval_min: int = Field(default=60, ge=5, le=1440)
    translate_to: str = "zh"
    enabled: bool = True
    detail_selector: dict | None = None
    headers_json: dict | None = None
    blocklist_tags: list[str] = []


class SourceUpdate(BaseModel):
    name: str | None = None
    enabled: bool | None = None
    priority: int | None = Field(default=None, ge=1, le=100)
    fetch_interval_min: int | None = Field(default=None, ge=5, le=1440)
    region: str | None = None
    translate_to: str | None = None
    blocklist_tags: list[str] | None = None