From 1e0044cbd154729b29d52c7c56103a1d2cab1d48 Mon Sep 17 00:00:00 2001
From: Kuroshini <o.hayatov@id-east.ru>
Date: Thu, 5 Dec 2019 12:11:44 +0300
Subject: [PATCH] artisans tags for favs

(cherry picked from commit 7081e52)
---
 apps/establishment/serializers/common.py | 2 ++
 apps/favorites/views.py                  | 3 ++-
 2 files changed, 4 insertions(+), 1 deletion(-)

diff --git a/apps/establishment/serializers/common.py b/apps/establishment/serializers/common.py
index 4a419fee..23fc24df 100644
--- a/apps/establishment/serializers/common.py
+++ b/apps/establishment/serializers/common.py
@@ -427,11 +427,13 @@ class EstablishmentSimilarSerializer(EstablishmentBaseSerializer):
     address = AddressDetailSerializer(read_only=True)
     schedule = ScheduleRUDSerializer(many=True, allow_null=True)
     establishment_type = EstablishmentTypeGeoSerializer()
+    artisan_category = TagBaseSerializer(many=True, allow_null=True)
 
     class Meta(EstablishmentBaseSerializer.Meta):
         fields = EstablishmentBaseSerializer.Meta.fields + [
             'schedule',
             'establishment_type',
+            'artisan_category',
         ]
 
 
diff --git a/apps/favorites/views.py b/apps/favorites/views.py
index 53a05469..bee25ced 100644
--- a/apps/favorites/views.py
+++ b/apps/favorites/views.py
@@ -29,7 +29,8 @@ class FavoritesEstablishmentListView(generics.ListAPIView):
     def get_queryset(self):
         """Override get_queryset method"""
         return Establishment.objects.filter(favorites__user=self.request.user) \
-            .order_by('-favorites').with_base_related()
+            .order_by('-favorites').with_base_related() \
+            .with_certain_tag_category_related('shop_category', 'artisan_category')
 
 
 class FavoritesProductListView(generics.ListAPIView):