mealie/mealie/schema/cookbook/cookbook.py

from fastapi_camelcase import CamelModel
from pydantic import UUID4, validator
from slugify import slugify

from ..recipe.recipe_category import CategoryBase, RecipeCategoryResponse


class CreateCookBook(CamelModel):
    name: str
    description: str = ""
    slug: str = None
    position: int = 1
    categories: list[CategoryBase] = []

    @validator("slug", always=True, pre=True)
    def validate_slug(slug: str, values):
        name: str = values["name"]
        calc_slug: str = slugify(name)

        if slug != calc_slug:
            slug = calc_slug

        return slug


class SaveCookBook(CreateCookBook):
    group_id: UUID4


class UpdateCookBook(SaveCookBook):
    id: UUID4


class ReadCookBook(UpdateCookBook):
    group_id: UUID4
    categories: list[CategoryBase] = []

    class Config:
        orm_mode = True


class RecipeCookBook(ReadCookBook):
    group_id: UUID4
    categories: list[RecipeCategoryResponse]

    class Config:
        orm_mode = True