mirror of
https://github.com/netbox-community/netbox.git
synced 2025-07-19 17:59:11 -06:00
95 lines
2.7 KiB
Python
95 lines
2.7 KiB
Python
from __future__ import unicode_literals
|
|
|
|
import django_filters
|
|
|
|
from django.contrib.auth.models import User
|
|
from django.contrib.contenttypes.models import ContentType
|
|
|
|
from dcim.models import Site
|
|
from .models import CF_TYPE_SELECT, CustomField, Graph, ExportTemplate, TopologyMap, UserAction
|
|
|
|
|
|
class CustomFieldFilter(django_filters.Filter):
|
|
"""
|
|
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):
|
|
# Skip filter on empty value
|
|
if not value.strip():
|
|
return queryset
|
|
# Treat 0 as None for Select fields
|
|
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(
|
|
custom_field_values__field__name=self.name,
|
|
custom_field_values__serialized_value__icontains=value,
|
|
)
|
|
|
|
|
|
class CustomFieldFilterSet(django_filters.FilterSet):
|
|
"""
|
|
Dynamically add a Filter for each CustomField applicable to the parent model.
|
|
"""
|
|
|
|
def __init__(self, *args, **kwargs):
|
|
super(CustomFieldFilterSet, self).__init__(*args, **kwargs)
|
|
|
|
obj_type = ContentType.objects.get_for_model(self._meta.model)
|
|
custom_fields = CustomField.objects.filter(obj_type=obj_type, is_filterable=True)
|
|
for cf in custom_fields:
|
|
self.filters['cf_{}'.format(cf.name)] = CustomFieldFilter(name=cf.name, cf_type=cf.type)
|
|
|
|
|
|
class GraphFilter(django_filters.FilterSet):
|
|
|
|
class Meta:
|
|
model = Graph
|
|
fields = ['type', 'name']
|
|
|
|
|
|
class ExportTemplateFilter(django_filters.FilterSet):
|
|
|
|
class Meta:
|
|
model = ExportTemplate
|
|
fields = ['content_type', 'name']
|
|
|
|
|
|
class TopologyMapFilter(django_filters.FilterSet):
|
|
site_id = django_filters.ModelMultipleChoiceFilter(
|
|
name='site',
|
|
queryset=Site.objects.all(),
|
|
label='Site',
|
|
)
|
|
site = django_filters.ModelMultipleChoiceFilter(
|
|
name='site__slug',
|
|
queryset=Site.objects.all(),
|
|
to_field_name='slug',
|
|
label='Site (slug)',
|
|
)
|
|
|
|
class Meta:
|
|
model = TopologyMap
|
|
fields = ['name', 'slug']
|
|
|
|
|
|
class UserActionFilter(django_filters.FilterSet):
|
|
username = django_filters.ModelMultipleChoiceFilter(
|
|
name='user__username',
|
|
queryset=User.objects.all(),
|
|
to_field_name='username',
|
|
)
|
|
|
|
class Meta:
|
|
model = UserAction
|
|
fields = ['user']
|