mirror of
https://github.com/netbox-community/netbox.git
synced 2025-07-16 20:22:53 -06:00
06a206ee33
This can easily be extended (as in the case of LocationFilterSet) by calling super() and ORing a filter to the queryset that is returned. See: https://docs.djangoproject.com/en/5.1/ref/models/querysets/#or
119 lines
3.6 KiB
Python
119 lines
3.6 KiB
Python
import django_filters
|
|
from django.db.models import Q
|
|
|
|
from dcim.choices import LinkStatusChoices
|
|
from dcim.base_filtersets import ScopedFilterSet
|
|
from dcim.models import Interface
|
|
from ipam.models import VLAN
|
|
from netbox.filtersets import NestedGroupModelFilterSet, NetBoxModelFilterSet
|
|
from tenancy.filtersets import TenancyFilterSet
|
|
from utilities.filters import TreeNodeMultipleChoiceFilter
|
|
from .choices import *
|
|
from .models import *
|
|
|
|
__all__ = (
|
|
'WirelessLANFilterSet',
|
|
'WirelessLANGroupFilterSet',
|
|
'WirelessLinkFilterSet',
|
|
)
|
|
|
|
|
|
class WirelessLANGroupFilterSet(NestedGroupModelFilterSet):
|
|
parent_id = django_filters.ModelMultipleChoiceFilter(
|
|
queryset=WirelessLANGroup.objects.all()
|
|
)
|
|
parent = django_filters.ModelMultipleChoiceFilter(
|
|
field_name='parent__slug',
|
|
queryset=WirelessLANGroup.objects.all(),
|
|
to_field_name='slug'
|
|
)
|
|
ancestor_id = TreeNodeMultipleChoiceFilter(
|
|
queryset=WirelessLANGroup.objects.all(),
|
|
field_name='parent',
|
|
lookup_expr='in'
|
|
)
|
|
ancestor = TreeNodeMultipleChoiceFilter(
|
|
queryset=WirelessLANGroup.objects.all(),
|
|
field_name='parent',
|
|
lookup_expr='in',
|
|
to_field_name='slug'
|
|
)
|
|
|
|
class Meta:
|
|
model = WirelessLANGroup
|
|
fields = ('id', 'name', 'slug', 'description')
|
|
|
|
|
|
class WirelessLANFilterSet(NetBoxModelFilterSet, ScopedFilterSet, TenancyFilterSet):
|
|
group_id = TreeNodeMultipleChoiceFilter(
|
|
queryset=WirelessLANGroup.objects.all(),
|
|
field_name='group',
|
|
lookup_expr='in'
|
|
)
|
|
group = TreeNodeMultipleChoiceFilter(
|
|
queryset=WirelessLANGroup.objects.all(),
|
|
field_name='group',
|
|
lookup_expr='in',
|
|
to_field_name='slug'
|
|
)
|
|
status = django_filters.MultipleChoiceFilter(
|
|
choices=WirelessLANStatusChoices
|
|
)
|
|
vlan_id = django_filters.ModelMultipleChoiceFilter(
|
|
queryset=VLAN.objects.all()
|
|
)
|
|
interface_id = django_filters.ModelMultipleChoiceFilter(
|
|
queryset=Interface.objects.all(),
|
|
field_name='interfaces'
|
|
)
|
|
auth_type = django_filters.MultipleChoiceFilter(
|
|
choices=WirelessAuthTypeChoices
|
|
)
|
|
auth_cipher = django_filters.MultipleChoiceFilter(
|
|
choices=WirelessAuthCipherChoices
|
|
)
|
|
|
|
class Meta:
|
|
model = WirelessLAN
|
|
fields = ('id', 'ssid', 'auth_psk', 'scope_id', 'description')
|
|
|
|
def search(self, queryset, name, value):
|
|
if not value.strip():
|
|
return queryset
|
|
qs_filter = (
|
|
Q(ssid__icontains=value) |
|
|
Q(description__icontains=value)
|
|
)
|
|
return queryset.filter(qs_filter)
|
|
|
|
|
|
class WirelessLinkFilterSet(NetBoxModelFilterSet, TenancyFilterSet):
|
|
interface_a_id = django_filters.ModelMultipleChoiceFilter(
|
|
queryset=Interface.objects.all()
|
|
)
|
|
interface_b_id = django_filters.ModelMultipleChoiceFilter(
|
|
queryset=Interface.objects.all()
|
|
)
|
|
status = django_filters.MultipleChoiceFilter(
|
|
choices=LinkStatusChoices
|
|
)
|
|
auth_type = django_filters.MultipleChoiceFilter(
|
|
choices=WirelessAuthTypeChoices
|
|
)
|
|
auth_cipher = django_filters.MultipleChoiceFilter(
|
|
choices=WirelessAuthCipherChoices
|
|
)
|
|
|
|
class Meta:
|
|
model = WirelessLink
|
|
fields = ('id', 'ssid', 'auth_psk', 'distance', 'distance_unit', 'description')
|
|
|
|
def search(self, queryset, name, value):
|
|
if not value.strip():
|
|
return queryset
|
|
qs_filter = (
|
|
Q(ssid__icontains=value) |
|
|
Q(description__icontains=value)
|
|
)
|
|
return queryset.filter(qs_filter)
|