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


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


class UserSettingsUpdateModel(BaseModel):
    protein_goal: OptionalCalories = None
    carb_goal: OptionalCalories = None
    fat_goal: OptionalCalories = None
    fiber_goal: OptionalCalories = None
    calories_goal: OptionalCalories = None