from .base import ObjModelMixin, Macronutrient, Calories
from pydantic import BaseModel


class UserSettingsModel(ObjModelMixin, BaseModel):
    protein_goal: Macronutrient
    carb_goal: Macronutrient
    fat_goal: Macronutrient
    fiber_goal: Macronutrient
    calories_goal: Calories


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