import datetime

from pydantic import BaseModel

from fooder.model.base import ObjModelMixin, Calories
from fooder.model.meal import MealModel


class DiaryModel(ObjModelMixin, BaseModel):
    date: datetime.date
    protein_goal: Calories
    carb_goal: Calories
    fat_goal: Calories
    fiber_goal: Calories
    calories_goal: Calories
    protein: float
    carb: float
    fat: float
    fiber: float
    calories: float
    meals: list[MealModel]


class DiaryCreateModel(BaseModel):
    date: datetime.date


class DiaryUpdateModel(BaseModel):
    protein_goal: Calories | None = None
    carb_goal: Calories | None = None
    fat_goal: Calories | None = None
    fiber_goal: Calories | None = None
    calories_goal: Calories | None = None