phzhik/gault-millau
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Wine color for wine tags only

Browse Source
This commit is contained in:
Kuroshini 2019-10-16 14:11:07 +03:00
parent a305c6ba83
commit d9e46fdfad
1 changed files with 5 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
apps/tag/filters.py
View File

@ -3,6 +3,7 @@ from django_filters import rest_framework as filters
from establishment.models import EstablishmentType
from django.conf import settings
from tag import models
from product import models as product_models
class TagsBaseFilterSet(filters.FilterSet):
@ -42,7 +43,10 @@ class TagCategoryFilterSet(TagsBaseFilterSet):
'product_type', )
def by_product_type(self, queryset, name, value):
return queryset.by_product_type(value)
if value == product_models.ProductType.WINE:
queryset = queryset.filter(index_name='wine-color')
queryset = queryset.by_product_type(value)
return queryset
# todo: filter by establishment type
def by_establishment_type(self, queryset, name, value):