mirror of
https://github.com/netbox-community/netbox.git
synced 2025-12-21 21:02:23 -06:00
Fixes #1649: Correct fitlering on null values (e.g. ?tenant_id=0) for django-filters v1.1.0+
This commit is contained in:
@@ -4,7 +4,6 @@ import django_filters
|
||||
import itertools
|
||||
|
||||
from django import forms
|
||||
from django.db.models import Q
|
||||
from django.utils.encoding import force_text
|
||||
|
||||
|
||||
@@ -66,51 +65,3 @@ class NullableModelMultipleChoiceField(forms.ModelMultipleChoiceField):
|
||||
stripped_value = value
|
||||
super(NullableModelMultipleChoiceField, self).clean(stripped_value)
|
||||
return value
|
||||
|
||||
|
||||
class NullableModelMultipleChoiceFilter(django_filters.ModelMultipleChoiceFilter):
|
||||
"""
|
||||
This class extends ModelMultipleChoiceFilter to accept an additional value which implies "is null". The default
|
||||
queryset filter argument is:
|
||||
|
||||
.filter(fieldname=value)
|
||||
|
||||
When filtering by the value representing "is null" ('0' by default) the argument is modified to:
|
||||
|
||||
.filter(fieldname__isnull=True)
|
||||
"""
|
||||
field_class = NullableModelMultipleChoiceField
|
||||
|
||||
def __init__(self, *args, **kwargs):
|
||||
self.null_value = kwargs.get('null_value', 0)
|
||||
super(NullableModelMultipleChoiceFilter, self).__init__(*args, **kwargs)
|
||||
|
||||
def filter(self, qs, value):
|
||||
value = value or () # Make sure we have an iterable
|
||||
|
||||
if self.is_noop(qs, value):
|
||||
return qs
|
||||
|
||||
# Even though not a noop, no point filtering if empty
|
||||
if not value:
|
||||
return qs
|
||||
|
||||
q = Q()
|
||||
for v in set(value):
|
||||
# Filtering by "is null"
|
||||
if v == force_text(self.null_value):
|
||||
arg = {'{}__isnull'.format(self.name): True}
|
||||
# Filtering by a related field (e.g. slug)
|
||||
elif self.field.to_field_name is not None:
|
||||
arg = {'{}__{}'.format(self.name, self.field.to_field_name): v}
|
||||
# Filtering by primary key (default)
|
||||
else:
|
||||
arg = {self.name: v}
|
||||
if self.conjoined:
|
||||
qs = self.get_method(qs)(**arg)
|
||||
else:
|
||||
q |= Q(**arg)
|
||||
if self.distinct:
|
||||
return self.get_method(qs)(q).distinct()
|
||||
|
||||
return self.get_method(qs)(q)
|
||||
|
||||
Reference in New Issue
Block a user