diff --git a/vrobbler/apps/foods/models.py b/vrobbler/apps/foods/models.py index 5cb93a5..581c55d 100644 --- a/vrobbler/apps/foods/models.py +++ b/vrobbler/apps/foods/models.py @@ -8,14 +8,15 @@ from django.urls import reverse from django_extensions.db.models import TimeStampedModel from imagekit.models import ImageSpecField from imagekit.processors import ResizeToFit -from scrobbles.dataclasses import BaseLogData +from scrobbles.dataclasses import BaseLogData, WithPeopleLogData from scrobbles.mixins import ScrobblableConstants, ScrobblableMixin BNULL = {"blank": True, "null": True} @dataclass -class FoodLogData(BaseLogData): +class FoodLogData(BaseLogData, WithPeopleLogData): + calories: Optional[int] = None meal: Optional[str] = None rating: Optional[str] = None