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 establishment.models import EstablishmentType
from django.conf import settings from django.conf import settings
from tag import models from tag import models
from product import models as product_models
class TagsBaseFilterSet(filters.FilterSet): class TagsBaseFilterSet(filters.FilterSet):
@ -42,7 +43,10 @@ class TagCategoryFilterSet(TagsBaseFilterSet):
'product_type', ) 'product_type', )
def by_product_type(self, queryset, name, value): 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 # todo: filter by establishment type
def by_establishment_type(self, queryset, name, value): def by_establishment_type(self, queryset, name, value):