add address to establishment similar and collection establishment

apps/collection/views/common.py

@@ -1,11 +1,11 @@
+from django.shortcuts import get_object_or_404
 from rest_framework import generics
 from rest_framework import permissions
 
 from collection import models
-from utils.pagination import ProjectPageNumberPagination
-from django.shortcuts import get_object_or_404
-from establishment.serializers import EstablishmentBaseSerializer
 from collection.serializers import common as serializers
+from establishment.serializers import EstablishmentSimilarSerializer
+from utils.pagination import ProjectPageNumberPagination
 
 
 # Mixins
@@ -53,7 +53,7 @@ class CollectionEstablishmentListView(CollectionListView):
     """Retrieve list of establishment for collection."""
     lookup_field = 'slug'
     pagination_class = ProjectPageNumberPagination
-    serializer_class = EstablishmentBaseSerializer
+    serializer_class = EstablishmentSimilarSerializer
 
     def get_queryset(self):
         """

apps/establishment/serializers/common.py

@@ -17,6 +17,7 @@ from review.serializers import ReviewShortSerializer
 
 class ContactPhonesSerializer(serializers.ModelSerializer):
     """Contact phone serializer"""
+
     class Meta:
         model = models.ContactPhone
         fields = [
@@ -26,6 +27,7 @@ class ContactPhonesSerializer(serializers.ModelSerializer):
 
 class ContactEmailsSerializer(serializers.ModelSerializer):
     """Contact email serializer"""
+
     class Meta:
         model = models.ContactEmail
         fields = [
@@ -35,6 +37,7 @@ class ContactEmailsSerializer(serializers.ModelSerializer):
 
 class SocialNetworkRelatedSerializers(serializers.ModelSerializer):
     """Social network serializers."""
+
     class Meta:
         model = models.SocialNetwork
         fields = [
@@ -45,7 +48,6 @@ class SocialNetworkRelatedSerializers(serializers.ModelSerializer):
 
 
 class PlateSerializer(ProjectModelSerializer):
-
     name_translated = TranslatedField()
     currency = CurrencySerializer(read_only=True)
 
@@ -316,6 +318,12 @@ class EstablishmentDetailSerializer(EstablishmentBaseSerializer):
         ]
 
 
+class EstablishmentSimilarSerializer(EstablishmentBaseSerializer):
+    """Serializer for Establishment model."""
+
+    address = AddressDetailSerializer(read_only=True)
+
+
 class EstablishmentCommentCreateSerializer(comment_serializers.CommentSerializer):
     """Create comment serializer"""
     mark = serializers.IntegerField()
@@ -379,4 +387,3 @@ class EstablishmentFavoritesCreateSerializer(FavoritesCreateSerializer):
             'content_object': validated_data.pop('establishment')
         })
         return super().create(validated_data)
-

apps/establishment/views/web.py

@@ -72,7 +72,7 @@ class EstablishmentRecentReviewListView(EstablishmentListView):
 class EstablishmentSimilarListView(EstablishmentListView):
     """Resource for getting a list of establishments."""
 
-    serializer_class = serializers.EstablishmentBaseSerializer
+    serializer_class = serializers.EstablishmentSimilarSerializer
     pagination_class = EstablishmentPortionPagination
 
     def get_queryset(self):