lunch & diner fields

apps/establishment/models.py

@@ -1266,6 +1266,10 @@ class Plate(TranslatedFieldsMixin, models.Model):
 
 
 class MenuQuerySet(models.QuerySet):
+
+    def with_base_related(self):
+        return self.prefetch_related('establishment')
+
     def with_schedule_plates_establishment(self):
         return self.select_related(
             'establishment',
@@ -1309,6 +1313,12 @@ class Menu(GalleryMixin, TranslatedFieldsMixin, BaseAttributes):
         verbose_name=_('Establishment schedule'),
         related_name='menus',
     )
+    lunch = ArrayField(models.BooleanField(
+        default=False,
+    ), size=7)
+    diner = ArrayField(models.BooleanField(
+        default=False,
+    ), size=7)
     old_id = models.PositiveIntegerField(_('old id'), blank=True, null=True, default=None)
 
     uploads = models.ManyToManyField(

apps/establishment/views/back.py

@@ -158,7 +158,7 @@ class EstablishmentScheduleCreateView(generics.CreateAPIView):
 class MenuListCreateView(generics.ListCreateAPIView):
     """Menu list create view."""
     serializer_class = serializers.MenuSerializers
-    queryset = models.Menu.objects.all()
+    queryset = models.Menu.objects.with_base_related()
     permission_classes = get_permission_classes(
         IsEstablishmentManager,
         IsEstablishmentAdministrator,