diff --git a/apps/review/filters.py b/apps/review/filters.py index 4867d1df..5b215b8e 100644 --- a/apps/review/filters.py +++ b/apps/review/filters.py @@ -8,6 +8,7 @@ class ReviewFilter(filters.FilterSet): """Review filter set.""" establishment_id = filters.NumberFilter(field_name='object_id', ) + product_id = filters.NumberFilter(field_name='object_id', ) class Meta: """Meta class.""" @@ -15,9 +16,15 @@ class ReviewFilter(filters.FilterSet): model = models.Review fields = ( 'establishment_id', + 'product_id', ) def by_establishment_id(self, queryset, name, value): if value not in EMPTY_VALUES: return queryset.by_establishment_id(value, content_type='establishment') 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