From 5ee7c314c0c961777e1dcbfe959e30cd79239ad1 Mon Sep 17 00:00:00 2001
From: Kuroshini <o.hayatov@id-east.ru>
Date: Mon, 2 Dec 2019 14:31:24 +0300
Subject: [PATCH] Fix product document serialization

---
 apps/search_indexes/serializers.py | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/apps/search_indexes/serializers.py b/apps/search_indexes/serializers.py
index 7f1999a2..b45bd2e3 100644
--- a/apps/search_indexes/serializers.py
+++ b/apps/search_indexes/serializers.py
@@ -135,6 +135,9 @@ class ProductEstablishmentDocumentSerializer(serializers.Serializer):
     index_name = serializers.CharField()
     city = AnotherCityDocumentShortSerializer()
 
+    def get_attribute(self, instance):
+        return instance.establishment if instance and instance.establishment else None
+
 
 class AddressDocumentSerializer(serializers.Serializer):
     """Address serializer for ES Document."""