Sith/sas/schemas.py

from datetime import datetime

from ninja import FilterSchema, ModelSchema
from pydantic import Field

from core.schemas import SimpleUserSchema
from sas.models import Picture


class PictureFilterSchema(FilterSchema):
    before_date: datetime | None = Field(None, q="date__lte")
    after_date: datetime | None = Field(None, q="date__gte")
    users_identified: set[int] | None = Field(None, q="people__user_id__in")
    album_id: int | None = Field(None, q="parent_id")


class PictureSchema(ModelSchema):
    class Meta:
        model = Picture
        fields = ["id", "name", "date", "size"]

    author: SimpleUserSchema = Field(validation_alias="owner")
    full_size_url: str
    compressed_url: str
    thumb_url: str