Fix product type serializer

apps/search_indexes/serializers.py

@@ -79,6 +79,18 @@ class WineColorDocumentSerializer(serializers.Serializer):
         return get_translated_value(obj.label)
 
 
+class ProductTypeDocumentSerializer(serializers.Serializer):
+    """Product type ES document serializer."""
+
+    id = serializers.IntegerField()
+    index_name = serializers.CharField()
+    name_translated = serializers.SerializerMethodField()
+
+    @staticmethod
+    def get_name_translated(obj):
+        return get_translated_value(obj.name)
+
+
 class ProductEstablishmentDocumentSerializer(serializers.Serializer):
     """Related to Product Establishment ES document serializer."""
 
@@ -202,13 +214,9 @@ class ProductDocumentSerializer(DocumentSerializer):
     subtypes = ProductSubtypeDocumentSerializer(many=True)
     wine_region = WineRegionDocumentSerializer(allow_null=True)
     wine_colors = WineColorDocumentSerializer(many=True)
-    product_type = serializers.SerializerMethodField()
+    product_type = ProductTypeDocumentSerializer(allow_null=True)
     establishment_detail = ProductEstablishmentDocumentSerializer(source='establishment', allow_null=True)
 
-    @staticmethod
-    def get_product_type(obj):
-        return get_translated_value(obj.product_type.name if obj.product_type else {})
-
     class Meta:
         """Meta class."""
 

apps/search_indexes/views.py

@@ -251,6 +251,9 @@ class ProductDocumentViewSet(BaseDocumentViewSet):
         'type': {
             'field': 'product_type.index_name',
         },
+        'product_type': {
+            'field': 'product_type.index_name',
+        },
         'subtype': {
             'field': 'subtypes.index_name',
             'lookups': [