From fa8b4d6d979d613d67423eee3b7c42ada1398567 Mon Sep 17 00:00:00 2001 From: Kuroshini Date: Thu, 9 Jan 2020 13:48:38 +0300 Subject: [PATCH] chosen tags for recipes --- apps/tag/filters.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/apps/tag/filters.py b/apps/tag/filters.py index a731272e..28c3db33 100644 --- a/apps/tag/filters.py +++ b/apps/tag/filters.py @@ -10,10 +10,12 @@ class TagsBaseFilterSet(filters.FilterSet): # Object type choices NEWS = 'news' ESTABLISHMENT = 'establishment' + RECIPES = 'recipe' TYPE_CHOICES = ( (NEWS, 'News'), (ESTABLISHMENT, 'Establishment'), + (RECIPES, 'Recipe'), ) type = filters.MultipleChoiceFilter(choices=TYPE_CHOICES, @@ -91,5 +93,7 @@ class TagsFilterSet(TagsBaseFilterSet): if self.ESTABLISHMENT in value: queryset = queryset.for_establishments().filter(category__value_type='list').filter(value__in=settings.ESTABLISHMENT_CHOSEN_TAGS).distinct( 'value') + if self.RECIPES in value: + queryset = queryset.for_news().filter(value__in=settings.RECIPES_CHOSEN_TAGS).distinct('value') return queryset