|
@@ -2,7 +2,7 @@ import django_filters
|
|
|
|
|
|
from django.contrib.contenttypes.models import ContentType
|
|
from django.contrib.contenttypes.models import ContentType
|
|
|
|
|
|
-from .models import CustomField
|
|
+from .models import CF_TYPE_SELECT, CustomField
|
|
|
|
|
|
|
|
|
|
class CustomFieldFilter(django_filters.Filter):
|
|
class CustomFieldFilter(django_filters.Filter):
|
|
@@ -10,9 +10,22 @@ class CustomFieldFilter(django_filters.Filter):
|
|
Filter objects by the presence of a CustomFieldValue. The filter's name is used as the CustomField name.
|
|
Filter objects by the presence of a CustomFieldValue. The filter's name is used as the CustomField name.
|
|
"""
|
|
"""
|
|
|
|
|
|
|
|
+ def __init__(self, cf_type, *args, **kwargs):
|
|
|
|
+ self.cf_type = cf_type
|
|
|
|
+ super(CustomFieldFilter, self).__init__(*args, **kwargs)
|
|
|
|
+
|
|
def filter(self, queryset, value):
|
|
def filter(self, queryset, value):
|
|
|
|
+
|
|
if not value.strip():
|
|
if not value.strip():
|
|
return queryset
|
|
return queryset
|
|
|
|
+
|
|
|
|
+ try:
|
|
|
|
+ if self.cf_type == CF_TYPE_SELECT and int(value) == 0:
|
|
|
|
+ return queryset.exclude(
|
|
|
|
+ custom_field_values__field__name=self.name,
|
|
|
|
+ )
|
|
|
|
+ except ValueError:
|
|
|
|
+ pass
|
|
return queryset.filter(
|
|
return queryset.filter(
|
|
custom_field_values__field__name=self.name,
|
|
custom_field_values__field__name=self.name,
|
|
custom_field_values__serialized_value=value,
|
|
custom_field_values__serialized_value=value,
|
|
@@ -30,4 +43,4 @@ class CustomFieldFilterSet(django_filters.FilterSet):
|
|
obj_type = ContentType.objects.get_for_model(self._meta.model)
|
|
obj_type = ContentType.objects.get_for_model(self._meta.model)
|
|
custom_fields = CustomField.objects.filter(obj_type=obj_type, is_filterable=True)
|
|
custom_fields = CustomField.objects.filter(obj_type=obj_type, is_filterable=True)
|
|
for cf in custom_fields:
|
|
for cf in custom_fields:
|
|
- self.filters['cf_{}'.format(cf.name)] = CustomFieldFilter(name=cf.name)
|
|
+ self.filters['cf_{}'.format(cf.name)] = CustomFieldFilter(name=cf.name, cf_type=cf.type)
|