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

Tags for artisans. Chosen & Category

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

8
apps/tag/filters.py
View File

@ -46,7 +46,11 @@ class TagCategoryFilterSet(TagsBaseFilterSet):
# 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):
return queryset.by_establishment_type(value) if value == EstablishmentType.ARTISAN:
qs = models.TagCategory.objects.filter(index_name='shop_category')
else:
qs = queryset.by_establishment_type(value)
return qs
class TagsFilterSet(TagsBaseFilterSet): class TagsFilterSet(TagsBaseFilterSet):
@ -65,7 +69,7 @@ class TagsFilterSet(TagsBaseFilterSet):
def by_establishment_type(self, queryset, name, value): def by_establishment_type(self, queryset, name, value):
if value == EstablishmentType.ARTISAN: if value == EstablishmentType.ARTISAN:
return models.Tag.objects.by_category_index_name('shop_category') return models.Tag.objects.by_category_index_name('shop_category')[0:8]
return queryset.by_establishment_type(value) return queryset.by_establishment_type(value)
# TMP TODO remove it later # TMP TODO remove it later