still trying to order products

This commit is contained in:
Sam
2021-02-24 13:59:42 +00:00
parent d98ee02f32
commit 1532040c2c
3 changed files with 19 additions and 8 deletions

View File

@@ -1,5 +1,7 @@
import django_filters
from rest_framework.filters import BaseFilterBackend
from companies.models import Company
from products.models import Product
@@ -26,10 +28,19 @@ class ProductTagFilter(django_filters.FilterSet):
class Meta:
model = Product
fields = ['name', 'tags', 'category', 'attributes', 'company', 'created']
fields = ['name', 'tags', 'category', 'attributes', 'company', 'created',]
def tag_filter(self, queryset, name, value):
return queryset.filter(**{
name: value,
})
class ProductOrderFilter(BaseFilterBackend):
def filter_queryset(self, request, queryset, view):
order_field = request.GET.get('order', None)
if order_field is not None:
return queryset.order_by(order_field)
else:
return queryset