1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
|
from __future__ import unicode_literals
from collections import OrderedDict
from django_filters import (FilterSet, MultipleChoiceFilter, RangeFilter,
OrderingFilter)
from django.forms import CheckboxSelectMultiple
from django.utils.translation import pgettext_lazy
from django_prices.models import PriceField
from .models import Product, ProductAttribute
DEFAULT_SORT = 'name'
SORT_BY_FIELDS = [{'value': 'name',
'label': pgettext_lazy('Sort by filter', 'name')},
{'value': 'price',
'label': pgettext_lazy('Sort by filter', 'price')}]
class ProductFilter(FilterSet):
def __init__(self, *args, **kwargs):
self.category = kwargs.pop('category')
super(ProductFilter, self).__init__(*args, **kwargs)
self.product_attributes, self.variant_attributes = (
self._get_attributes())
self.filters.update(self._add_product_attributes_filters())
self.filters.update(self._add_product_variants_attributes_filters())
self.filters = OrderedDict(sorted(self.filters.items()))
sort_by = OrderingFilter(
label='Sort by',
fields=[(field['value'], field['value']) for field in SORT_BY_FIELDS]
)
class Meta:
model = Product
fields = ['price']
filter_overrides = {
PriceField: {
'filter_class': RangeFilter
}
}
def _get_attributes(self):
product_attributes = (
ProductAttribute.objects.all()
.prefetch_related('values')
.filter(products_class__products__categories=self.category)
.distinct())
variant_attributes = (
ProductAttribute.objects.all()
.prefetch_related('values')
.filter(product_variants_class__products__categories=self.category)
.distinct())
return product_attributes, variant_attributes
def _add_product_attributes_filters(self):
filters = {}
for attribute in self.product_attributes:
filters[attribute.slug] = MultipleChoiceFilter(
name='attributes__%s' % attribute.pk,
label=attribute.name,
widget=CheckboxSelectMultiple,
choices=self._get_attribute_choices(attribute))
return filters
def _add_product_variants_attributes_filters(self):
filters = {}
for attribute in self.variant_attributes:
filters[attribute.slug] = MultipleChoiceFilter(
name='variants__attributes__%s' % attribute.pk,
label=attribute.name,
widget=CheckboxSelectMultiple,
choices=self._get_attribute_choices(attribute))
return filters
def _get_attribute_choices(self, attribute):
return [(choice.pk, choice.name) for choice in attribute.values.all()]
|
