From 498e8ac126e5646441b3c617cbd53440e7e58be1 Mon Sep 17 00:00:00 2001
From: Anatoly <a.feteleu@spider.ru>
Date: Wed, 20 Nov 2019 11:56:25 +0300
Subject: [PATCH] fix product country filter in ES

---
 apps/search_indexes/documents/product.py | 14 +++++++++++++-
 apps/search_indexes/views.py             |  2 +-
 2 files changed, 14 insertions(+), 2 deletions(-)

diff --git a/apps/search_indexes/documents/product.py b/apps/search_indexes/documents/product.py
index 35b1956a..13943191 100644
--- a/apps/search_indexes/documents/product.py
+++ b/apps/search_indexes/documents/product.py
@@ -34,7 +34,19 @@ class ProductDocument(Document):
             'id': fields.IntegerField(),
             'name': fields.KeywordField(),
             'slug': fields.KeywordField(),
-            # 'city' TODO: city indexing
+            'address': fields.ObjectField(
+                properties={
+                    'city': fields.ObjectField(
+                        properties={
+                            'country': fields.ObjectField(
+                                properties={
+                                    'code': fields.KeywordField()
+                                }
+                            )
+                        }
+                    )
+                }
+            )
         }
     )
     wine_colors = fields.ObjectField(
diff --git a/apps/search_indexes/views.py b/apps/search_indexes/views.py
index 46f25545..783754c7 100644
--- a/apps/search_indexes/views.py
+++ b/apps/search_indexes/views.py
@@ -243,7 +243,7 @@ class ProductDocumentViewSet(BaseDocumentViewSet):
             'lookups': [constants.LOOKUP_QUERY_IN],
         },
         'country': {
-            'field': 'wine_region.country.code',
+            'field': 'establishment.address.city.country.code',
         },
         'wine_colors_id': {
             'field': 'wine_colors.id',