From e633b4f5f523400e0903f4ae43dbbc92495f4ba9 Mon Sep 17 00:00:00 2001 From: Kuroshini Date: Tue, 4 Feb 2020 16:05:25 +0300 Subject: [PATCH] subtypes to establishments list --- apps/establishment/models.py | 2 +- apps/establishment/serializers/back.py | 3 +++ 2 files changed, 4 insertions(+), 1 deletion(-) diff --git a/apps/establishment/models.py b/apps/establishment/models.py index 519af701..9d9c88ea 100644 --- a/apps/establishment/models.py +++ b/apps/establishment/models.py @@ -128,7 +128,7 @@ class EstablishmentQuerySet(models.QuerySet): def with_base_related(self): """Return qs with related objects.""" return self.select_related('address', 'establishment_type'). \ - prefetch_related('tags', 'tags__translation').with_main_image() + prefetch_related('tags', 'tags__translation', 'establishment_subtypes').with_main_image() def with_schedule(self): """Return qs with related schedule.""" diff --git a/apps/establishment/serializers/back.py b/apps/establishment/serializers/back.py index 1448fafe..841f0b80 100644 --- a/apps/establishment/serializers/back.py +++ b/apps/establishment/serializers/back.py @@ -83,6 +83,8 @@ class EstablishmentListCreateSerializer(model_serializers.EstablishmentBaseSeria child=serializers.CharField(max_length=128), required=False, ) + subtypes = model_serializers.EstablishmentSubTypeBaseSerializer(source='establishment_subtypes', + read_only=True, many=True) class Meta(model_serializers.EstablishmentBaseSerializer.Meta): fields = [ @@ -98,6 +100,7 @@ class EstablishmentListCreateSerializer(model_serializers.EstablishmentBaseSeria 'toque_number', 'type_id', 'type', + 'subtypes', 'socials', 'image_url', 'slug',