subtypes to establishments list

This commit is contained in:
Kuroshini 2020-02-04 16:05:25 +03:00
parent 5687eff646
commit e633b4f5f5
2 changed files with 4 additions and 1 deletions

View File

@ -128,7 +128,7 @@ class EstablishmentQuerySet(models.QuerySet):
def with_base_related(self): def with_base_related(self):
"""Return qs with related objects.""" """Return qs with related objects."""
return self.select_related('address', 'establishment_type'). \ 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): def with_schedule(self):
"""Return qs with related schedule.""" """Return qs with related schedule."""

View File

@ -83,6 +83,8 @@ class EstablishmentListCreateSerializer(model_serializers.EstablishmentBaseSeria
child=serializers.CharField(max_length=128), child=serializers.CharField(max_length=128),
required=False, required=False,
) )
subtypes = model_serializers.EstablishmentSubTypeBaseSerializer(source='establishment_subtypes',
read_only=True, many=True)
class Meta(model_serializers.EstablishmentBaseSerializer.Meta): class Meta(model_serializers.EstablishmentBaseSerializer.Meta):
fields = [ fields = [
@ -98,6 +100,7 @@ class EstablishmentListCreateSerializer(model_serializers.EstablishmentBaseSeria
'toque_number', 'toque_number',
'type_id', 'type_id',
'type', 'type',
'subtypes',
'socials', 'socials',
'image_url', 'image_url',
'slug', 'slug',