From d3ab910fba2b9b61d8e6394818e95e605f084131 Mon Sep 17 00:00:00 2001
From: Kuroshini <o.hayatov@id-east.ru>
Date: Wed, 16 Oct 2019 14:11:07 +0300
Subject: [PATCH] Filter recipes news

---
 apps/news/filters.py | 21 +++++++++++++++++----
 apps/news/models.py  |  6 ++++++
 2 files changed, 23 insertions(+), 4 deletions(-)

diff --git a/apps/news/filters.py b/apps/news/filters.py
index c4462189..084c570f 100644
--- a/apps/news/filters.py
+++ b/apps/news/filters.py
@@ -1,14 +1,21 @@
 """Filters from application News"""
-import django_filters
+from django_filters import rest_framework as filters
+from django.utils.translation import gettext_lazy as _
 
 from news import models
 
 
-class NewsListFilterSet(django_filters.FilterSet):
+class NewsListFilterSet(filters.FilterSet):
     """FilterSet for News list"""
 
-    is_highlighted = django_filters.BooleanFilter()
-    title = django_filters.CharFilter(method='by_title')
+    is_highlighted = filters.BooleanFilter()
+    title = filters.CharFilter(method='by_title')
+    tag_group = filters.ChoiceFilter(
+        choices=(
+            (models.News.RECIPES_TAG_VALUE, _('Recipes')),
+        ),
+        method='by_tag_group'
+    )
 
     class Meta:
         """Meta class"""
@@ -16,8 +23,14 @@ class NewsListFilterSet(django_filters.FilterSet):
         fields = (
             'title',
             'is_highlighted',
+            'tag_group',
         )
 
+    def by_tag_group(self, queryset, name, value):
+        if value == models.News.RECIPES_TAG_VALUE:
+            queryset = queryset.recipe_news()
+        return queryset
+
     def by_title(self, queryset, name, value):
         """Crappy search by title according to locale"""
         if value:
diff --git a/apps/news/models.py b/apps/news/models.py
index 73a8c369..9cf385d2 100644
--- a/apps/news/models.py
+++ b/apps/news/models.py
@@ -53,6 +53,10 @@ class NewsQuerySet(TranslationQuerysetMixin):
         """Filter collection by country code."""
         return self.filter(country__code=code)
 
+    def recipe_news(self):
+        """Returns news with tag 'cook' qs."""
+        return self.filter(tags__value=News.RECIPES_TAG_VALUE)
+
     def published(self):
         """Return only published news"""
         now = timezone.now()
@@ -127,6 +131,8 @@ class News(BaseAttributes, TranslatedFieldsMixin):
         (PUBLISHED_EXCLUSIVE, _('Published exclusive')),
     )
 
+    RECIPES_TAG_VALUE = 'cook'
+
     old_id = models.PositiveIntegerField(_('old id'), blank=True, null=True, default=None)
     news_type = models.ForeignKey(NewsType, on_delete=models.PROTECT,
                                   verbose_name=_('news type'))