category tag tags filter by obj type

apps/tag/serializers.py

@@ -1,9 +1,12 @@
 """Tag serializers."""
 from rest_framework import serializers
+from rest_framework.fields import SerializerMethodField
+
 from establishment.models import (Establishment, EstablishmentType,
                                   EstablishmentSubType)
 from news.models import News, NewsType
 from tag import models
+from tag.models import Tag
 from utils.exceptions import (ObjectAlreadyAdded, BindingObjectNotFound,
                               RemovedBindingObjectNotFound)
 from utils.serializers import TranslatedField
@@ -12,6 +15,9 @@ from utils.serializers import TranslatedField
 class TagBaseSerializer(serializers.ModelSerializer):
     """Serializer for model Tag."""
 
+    def get_extra_kwargs(self):
+        return super().get_extra_kwargs()
+
     label_translated = TranslatedField()
     index_name = serializers.CharField(source='value', read_only=True, allow_null=True)
 
@@ -37,6 +43,7 @@ class TagBackOfficeSerializer(TagBaseSerializer):
             'category'
         )
 
+
 class TagCategoryProductSerializer(serializers.ModelSerializer):
     """SHORT Serializer for TagCategory"""
 
@@ -57,7 +64,7 @@ class TagCategoryBaseSerializer(serializers.ModelSerializer):
     """Serializer for model TagCategory."""
 
     label_translated = TranslatedField()
-    tags = TagBaseSerializer(many=True, read_only=True)
+    tags = SerializerMethodField()
 
     class Meta:
         """Meta class."""
@@ -70,6 +77,22 @@ class TagCategoryBaseSerializer(serializers.ModelSerializer):
             'tags',
         )
 
+    def get_tags(self, obj):
+        query_params = dict(self.context['request'].query_params)
+        if len(query_params) > 1:
+            return None
+
+        tags = Tag.objects.all()
+
+        if 'establishment_type' in query_params:
+            types = query_params['establishment_type']
+            tags = tags.filter(establishments__establishment_type__index_name__in=types).distinct()
+        elif 'product_type' in query_params:
+            types = query_params['product_type']
+            tags = tags.filter(products__product_type__index_name__in=types).distinct()
+
+        return TagBaseSerializer(instance=tags, many=True, read_only=True).data
+
 
 class TagCategoryShortSerializer(serializers.ModelSerializer):
     """Serializer for model TagCategory."""