Added filtering chosen tags by country

apps/tag/filters.py

@@ -40,3 +40,32 @@ class TagCategoryFilterSet(filters.FilterSet):
     # todo: filter by establishment type
     def by_establishment_type(self, queryset, name, value):
         return queryset.by_establishment_type(value)
+
+
+class TagsFilterSet(filters.FilterSet):
+    """Chosen tags filterset."""
+
+    # Object type choices
+    NEWS = 'news'
+    ESTABLISHMENT = 'establishment'
+
+    TYPE_CHOICES = (
+        (NEWS, 'News'),
+        (ESTABLISHMENT, 'Establishment'),
+    )
+
+    type = filters.MultipleChoiceFilter(choices=TYPE_CHOICES,
+                                        method='filter_by_type')
+
+    class Meta:
+        """Meta class."""
+
+        model = models.Tag
+        fields = ('type',)
+
+    def filter_by_type(self, queryset, name, value):
+        if self.NEWS in value:
+            queryset = queryset.for_news()
+        if self.ESTABLISHMENT in value:
+            queryset = queryset.for_establishments()
+        return queryset

apps/tag/migrations/0008_auto_20191101_1244.py

@@ -0,0 +1,37 @@
+# Generated by Django 2.2.4 on 2019-11-01 12:44
+
+from django.db import migrations, models
+import django.db.models.deletion
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ('location', '0020_merge_20191030_1714'),
+        ('tag', '0007_auto_20191030_1514'),
+    ]
+
+    operations = [
+        migrations.RemoveField(
+            model_name='tag',
+            name='priority',
+        ),
+        migrations.CreateModel(
+            name='ChosenTagSettings',
+            fields=[
+                ('id', models.AutoField(auto_created=True, primary_key=True, serialize=False, verbose_name='ID')),
+                ('priority', models.IntegerField(default=0)),
+                ('country', models.ForeignKey(on_delete=django.db.models.deletion.PROTECT, to='location.Country')),
+                ('tag', models.ForeignKey(on_delete=django.db.models.deletion.PROTECT, to='tag.Tag')),
+            ],
+            options={
+                'verbose_name': 'Tag',
+                'verbose_name_plural': 'Tags',
+            },
+        ),
+        migrations.AddField(
+            model_name='tag',
+            name='chosen_tag_settings',
+            field=models.ManyToManyField(through='tag.ChosenTagSettings', to='location.Country'),
+        ),
+    ]

apps/tag/models.py

@@ -1,16 +1,26 @@
 """Tag app models."""
+from django.contrib.contenttypes.models import ContentType
 from django.db import models
 from django.utils.translation import gettext_lazy as _
+
 from configuration.models import TranslationSettings
+from location.models import Country
 from utils.models import TJSONField, TranslatedFieldsMixin
 
 
 class TagQuerySet(models.QuerySet):
-    def filter_chosen(self):
+
-        return self.exclude(priority__isnull=True)
+    def for_news(self):
+        """Select chosen tags for news."""
+        return self.filter(category__news_types__isnull=False)
+
+    def for_establishments(self):
+        """Select chosen tags for establishments."""
+        return self.filter(models.Q(category__establishment_types__isnull=False) |
+                           models.Q(category__establishment_subtypes__isnull=False))
 
     def order_by_priority(self):
-        return self.order_by('priority')
+        return self.order_by('chosentagsettings__priority')
 
 
 class Tag(TranslatedFieldsMixin, models.Model):
@@ -24,7 +34,8 @@ class Tag(TranslatedFieldsMixin, models.Model):
     category = models.ForeignKey('TagCategory', on_delete=models.CASCADE,
                                  null=True, related_name='tags',
                                  verbose_name=_('Category'))
-    priority = models.IntegerField(unique=True, null=True, default=None)
+
+    chosen_tag_settings = models.ManyToManyField(Country, through='ChosenTagSettings')
 
     objects = TagQuerySet.as_manager()
 
@@ -35,11 +46,29 @@ class Tag(TranslatedFieldsMixin, models.Model):
         verbose_name_plural = _('Tags')
 
     def __str__(self):
-        label = 'None'
+        return f'Tag (id = {self.id}, label_translated = {self.label_translated})'
-        lang = TranslationSettings.get_solo().default_language
+
-        if self.label and lang in self.label:
+
-            label = self.label[lang]
+class ChosenTagSettingsQuerySet(models.QuerySet):
-        return f'id:{self.id}-{label}'
+
+    def by_country_code(self, country_code):
+        return self.filter(country__code=country_code)
+
+
+class ChosenTagSettings(models.Model):
+    """Tag model."""
+
+    tag = models.ForeignKey(Tag, on_delete=models.PROTECT)
+    country = models.ForeignKey(Country, on_delete=models.PROTECT)
+    priority = models.IntegerField(null=False, default=0)
+
+    objects = ChosenTagSettingsQuerySet.as_manager()
+
+    class Meta:
+        """Meta class."""
+
+        verbose_name = _('Tag')
+        verbose_name_plural = _('Tags')
 
 
 class TagCategoryQuerySet(models.QuerySet):

apps/tag/views.py

@@ -10,10 +10,14 @@ class ChosenTagsView(generics.ListAPIView, viewsets.GenericViewSet):
     pagination_class = None
     permission_classes = (permissions.AllowAny,)
     serializer_class = serializers.TagBaseSerializer
+    filterset_class = filters.TagsFilterSet
 
     def get_queryset(self):
+        result_tags_ids = models.ChosenTagSettings.objects \
+            .by_country_code(self.request.country_code) \
+            .values('tag_id')
         return models.Tag.objects \
-            .filter_chosen() \
+            .filter(id__in=result_tags_ids) \
             .order_by_priority()