52 lines
1.1 KiB
Python
52 lines
1.1 KiB
Python
from __future__ import annotations
|
|
|
|
from pydantic import BaseModel, ConfigDict, field_validator
|
|
|
|
from services.path import encode_path
|
|
|
|
|
|
class PhotoItem(BaseModel):
|
|
model_config = ConfigDict(from_attributes=True)
|
|
|
|
id: int
|
|
path: str
|
|
filename: str
|
|
category_id: int
|
|
|
|
@field_validator("path", mode="before")
|
|
@classmethod
|
|
def encode_urls(cls, v):
|
|
return encode_path(v)
|
|
|
|
@field_validator("path", mode="after")
|
|
@classmethod
|
|
def add_media(cls, v: str) -> str:
|
|
if not v: return v
|
|
clean_v = v.lstrip('/')
|
|
if clean_v.startswith('media/'): return f"{clean_v}"
|
|
return f"media/{clean_v}"
|
|
|
|
|
|
class Category(BaseModel):
|
|
model_config = ConfigDict(from_attributes=True)
|
|
|
|
id: int
|
|
title: str
|
|
photos: list[PhotoItem] = []
|
|
|
|
|
|
class PhotosOut(BaseModel):
|
|
photos: list[PhotoItem]
|
|
|
|
class CategoryItem(BaseModel):
|
|
id: int
|
|
title: str
|
|
photos_count: int
|
|
|
|
class CategoriesOut(BaseModel):
|
|
categories: list[CategoryItem]
|
|
|
|
class PhotosPaginatedOut(BaseModel):
|
|
photos: list[PhotoItem]
|
|
page_inc: str
|
|
page_dec: str |