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

review filter by product id

Browse Source
This commit is contained in:
alex 2019-11-20 12:56:57 +03:00
parent d3faacba3b
commit 61e049f5c3
1 changed files with 7 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
apps/review/filters.py
View File

@ -8,6 +8,7 @@ class ReviewFilter(filters.FilterSet):
"""Review filter set.""" """Review filter set."""
establishment_id = filters.NumberFilter(field_name='object_id', ) establishment_id = filters.NumberFilter(field_name='object_id', )
product_id = filters.NumberFilter(field_name='object_id', )
class Meta: class Meta:
"""Meta class.""" """Meta class."""
@ -15,9 +16,15 @@ class ReviewFilter(filters.FilterSet):
model = models.Review model = models.Review
fields = ( fields = (
'establishment_id', 'establishment_id',
'product_id',
) )
def by_establishment_id(self, queryset, name, value): def by_establishment_id(self, queryset, name, value):
if value not in EMPTY_VALUES: if value not in EMPTY_VALUES:
return queryset.by_establishment_id(value, content_type='establishment') return queryset.by_establishment_id(value, content_type='establishment')
return queryset return queryset
def by_product_id(self, queryset, name, value):
if value not in EMPTY_VALUES:
return queryset.by_establishment_id(value, content_type='product')
return queryset