diff --git a/apps/establishment/models.py b/apps/establishment/models.py index e02b3291..fddd82d5 100644 --- a/apps/establishment/models.py +++ b/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( diff --git a/apps/establishment/views/back.py b/apps/establishment/views/back.py index 39a7a54a..49b2349d 100644 --- a/apps/establishment/views/back.py +++ b/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,