"""
Pydantic схемы для API
"""
from pydantic import BaseModel, HttpUrl
from typing import List


class ParserOneRequest(BaseModel):
    """Запрос для парсинга первого источника"""
    time: str


class Parserall(BaseModel):
    """Запрос для парсинга любого источника"""
    url: HttpUrl
    promt: str


class Parserall_url(BaseModel):
    """Запрос URL для источника"""
    url: HttpUrl


class Source(BaseModel):
    """Модель источника для настроек"""
    name: str
    promt: str


class DownloadRange(BaseModel):
    """Диапазон дат для скачивания файлов"""
    data_start: str
    data_finish: str
    field_name: str = "status"


class DownloadCountsResponse(BaseModel):
    """Ответ с количеством статей для выгрузки по каждому полю"""
    tematik: int
    svodka: int
    donesenie: int
    bilutene: int