318 lines
9.0 KiB
Python
318 lines
9.0 KiB
Python
import django_filters
|
|
|
|
from django.db.models import Q
|
|
|
|
from .models import Site, RackGroup, Rack, Manufacturer, DeviceType, DeviceRole, Device, ConsolePort, \
|
|
ConsoleServerPort, Platform, PowerPort, PowerOutlet, Interface, InterfaceConnection
|
|
|
|
|
|
class RackGroupFilter(django_filters.FilterSet):
|
|
site_id = django_filters.ModelMultipleChoiceFilter(
|
|
name='site',
|
|
queryset=Site.objects.all(),
|
|
label='Site (ID)',
|
|
)
|
|
site = django_filters.ModelMultipleChoiceFilter(
|
|
name='site',
|
|
queryset=Site.objects.all(),
|
|
to_field_name='slug',
|
|
label='Site (slug)',
|
|
)
|
|
|
|
class Meta:
|
|
model = RackGroup
|
|
fields = ['site_id', 'site']
|
|
|
|
|
|
class RackFilter(django_filters.FilterSet):
|
|
q = django_filters.MethodFilter(
|
|
action='search',
|
|
label='Search',
|
|
)
|
|
site_id = django_filters.ModelMultipleChoiceFilter(
|
|
name='site',
|
|
queryset=Site.objects.all(),
|
|
label='Site (ID)',
|
|
)
|
|
site = django_filters.ModelMultipleChoiceFilter(
|
|
name='site',
|
|
queryset=Site.objects.all(),
|
|
to_field_name='slug',
|
|
label='Site (slug)',
|
|
)
|
|
group_id = django_filters.ModelMultipleChoiceFilter(
|
|
name='group',
|
|
queryset=RackGroup.objects.all(),
|
|
label='Group (ID)',
|
|
)
|
|
group = django_filters.ModelMultipleChoiceFilter(
|
|
name='group',
|
|
queryset=RackGroup.objects.all(),
|
|
to_field_name='slug',
|
|
label='Group (slug)',
|
|
)
|
|
|
|
class Meta:
|
|
model = Rack
|
|
fields = ['q', 'site_id', 'site', 'u_height']
|
|
|
|
def search(self, queryset, value):
|
|
value = value.strip()
|
|
return queryset.filter(
|
|
Q(name__icontains=value) |
|
|
Q(facility_id__icontains=value)
|
|
)
|
|
|
|
|
|
class DeviceTypeFilter(django_filters.FilterSet):
|
|
manufacturer_id = django_filters.ModelChoiceFilter(
|
|
name='manufacturer',
|
|
queryset=Manufacturer.objects.all(),
|
|
label='Manufacturer (ID)',
|
|
)
|
|
manufacturer = django_filters.ModelChoiceFilter(
|
|
name='manufacturer',
|
|
queryset=Manufacturer.objects.all(),
|
|
to_field_name='slug',
|
|
label='Manufacturer (slug)',
|
|
)
|
|
|
|
class Meta:
|
|
model = DeviceType
|
|
fields = ['manufacturer_id', 'manufacturer', 'model', 'u_height', 'is_console_server', 'is_pdu',
|
|
'is_network_device']
|
|
|
|
|
|
class DeviceFilter(django_filters.FilterSet):
|
|
q = django_filters.MethodFilter(
|
|
action='search',
|
|
label='Search',
|
|
)
|
|
site_id = django_filters.ModelMultipleChoiceFilter(
|
|
name='rack__site',
|
|
queryset=Site.objects.all(),
|
|
label='Site (ID)',
|
|
)
|
|
site = django_filters.ModelMultipleChoiceFilter(
|
|
name='rack__site',
|
|
queryset=Site.objects.all(),
|
|
to_field_name='slug',
|
|
label='Site name (slug)',
|
|
)
|
|
rack_id = django_filters.ModelMultipleChoiceFilter(
|
|
name='rack',
|
|
queryset=Rack.objects.all(),
|
|
label='Rack (ID)',
|
|
)
|
|
role_id = django_filters.ModelMultipleChoiceFilter(
|
|
name='device_role',
|
|
queryset=DeviceRole.objects.all(),
|
|
label='Role (ID)',
|
|
)
|
|
role = django_filters.ModelMultipleChoiceFilter(
|
|
name='device_role',
|
|
queryset=DeviceRole.objects.all(),
|
|
to_field_name='slug',
|
|
label='Role (slug)',
|
|
)
|
|
device_type = django_filters.ModelMultipleChoiceFilter(
|
|
name='device_type',
|
|
queryset=DeviceType.objects.all(),
|
|
label='Device type (ID)',
|
|
)
|
|
manufacturer_id = django_filters.ModelMultipleChoiceFilter(
|
|
name='device_type__manufacturer',
|
|
queryset=Manufacturer.objects.all(),
|
|
label='Manufacturer (ID)',
|
|
)
|
|
manufacturer = django_filters.ModelMultipleChoiceFilter(
|
|
name='device_type__manufacturer',
|
|
queryset=Manufacturer.objects.all(),
|
|
to_field_name='slug',
|
|
label='Manufacturer (slug)',
|
|
)
|
|
model = django_filters.ModelMultipleChoiceFilter(
|
|
name='device_type',
|
|
queryset=DeviceType.objects.all(),
|
|
to_field_name='slug',
|
|
label='Device model (slug)',
|
|
)
|
|
platform_id = django_filters.ModelMultipleChoiceFilter(
|
|
name='platform',
|
|
queryset=Platform.objects.all(),
|
|
label='Platform (ID)',
|
|
)
|
|
platform = django_filters.ModelMultipleChoiceFilter(
|
|
name='platform',
|
|
queryset=Platform.objects.all(),
|
|
to_field_name='slug',
|
|
label='Platform (slug)',
|
|
)
|
|
is_console_server = django_filters.BooleanFilter(
|
|
name='device_type__is_console_server',
|
|
label='Is a console server',
|
|
)
|
|
is_pdu = django_filters.BooleanFilter(
|
|
name='device_type__is_pdu',
|
|
label='Is a PDU',
|
|
)
|
|
is_network_device = django_filters.BooleanFilter(
|
|
name='device_type__is_network_device',
|
|
label='Is a network device',
|
|
)
|
|
|
|
class Meta:
|
|
model = Device
|
|
fields = ['q', 'name', 'site_id', 'site', 'rack_id', 'role_id', 'role', 'device_type', 'manufacturer_id',
|
|
'manufacturer', 'model', 'platform_id', 'platform', 'is_console_server', 'is_pdu',
|
|
'is_network_device']
|
|
|
|
def search(self, queryset, value):
|
|
value = value.strip()
|
|
return queryset.filter(
|
|
Q(name__icontains=value) |
|
|
Q(serial__icontains=value) |
|
|
Q(modules__serial__icontains=value)
|
|
).distinct()
|
|
|
|
|
|
class ConsolePortFilter(django_filters.FilterSet):
|
|
device_id = django_filters.ModelMultipleChoiceFilter(
|
|
name='device',
|
|
queryset=Device.objects.all(),
|
|
label='Device (ID)',
|
|
)
|
|
device = django_filters.ModelMultipleChoiceFilter(
|
|
name='device',
|
|
queryset=Device.objects.all(),
|
|
to_field_name='name',
|
|
label='Device (name)',
|
|
)
|
|
|
|
class Meta:
|
|
model = ConsolePort
|
|
fields = ['device_id', 'device', 'name']
|
|
|
|
|
|
class ConsoleServerPortFilter(django_filters.FilterSet):
|
|
device_id = django_filters.ModelMultipleChoiceFilter(
|
|
name='device',
|
|
queryset=Device.objects.all(),
|
|
label='Device (ID)',
|
|
)
|
|
device = django_filters.ModelMultipleChoiceFilter(
|
|
name='device',
|
|
queryset=Device.objects.all(),
|
|
to_field_name='name',
|
|
label='Device (name)',
|
|
)
|
|
|
|
class Meta:
|
|
model = ConsoleServerPort
|
|
fields = ['device_id', 'device', 'name']
|
|
|
|
|
|
class PowerPortFilter(django_filters.FilterSet):
|
|
device_id = django_filters.ModelMultipleChoiceFilter(
|
|
name='device',
|
|
queryset=Device.objects.all(),
|
|
label='Device (ID)',
|
|
)
|
|
device = django_filters.ModelMultipleChoiceFilter(
|
|
name='device',
|
|
queryset=Device.objects.all(),
|
|
to_field_name='name',
|
|
label='Device (name)',
|
|
)
|
|
|
|
class Meta:
|
|
model = PowerPort
|
|
fields = ['device_id', 'device', 'name']
|
|
|
|
|
|
class PowerOutletFilter(django_filters.FilterSet):
|
|
device_id = django_filters.ModelMultipleChoiceFilter(
|
|
name='device',
|
|
queryset=Device.objects.all(),
|
|
label='Device (ID)',
|
|
)
|
|
device = django_filters.ModelMultipleChoiceFilter(
|
|
name='device',
|
|
queryset=Device.objects.all(),
|
|
to_field_name='name',
|
|
label='Device (name)',
|
|
)
|
|
|
|
class Meta:
|
|
model = PowerOutlet
|
|
fields = ['device_id', 'device', 'name']
|
|
|
|
|
|
class InterfaceFilter(django_filters.FilterSet):
|
|
device_id = django_filters.ModelMultipleChoiceFilter(
|
|
name='device',
|
|
queryset=Device.objects.all(),
|
|
label='Device (ID)',
|
|
)
|
|
device = django_filters.ModelMultipleChoiceFilter(
|
|
name='device',
|
|
queryset=Device.objects.all(),
|
|
to_field_name='name',
|
|
label='Device (name)',
|
|
)
|
|
|
|
class Meta:
|
|
model = Interface
|
|
fields = ['device_id', 'device', 'name']
|
|
|
|
|
|
class ConsoleConnectionFilter(django_filters.FilterSet):
|
|
site = django_filters.MethodFilter(
|
|
action='filter_site',
|
|
label='Site (slug)',
|
|
)
|
|
|
|
class Meta:
|
|
model = ConsoleServerPort
|
|
|
|
def filter_site(self, queryset, value):
|
|
value = value.strip()
|
|
if not value:
|
|
return queryset
|
|
return queryset.filter(cs_port__device__rack__site__slug=value)
|
|
|
|
|
|
class PowerConnectionFilter(django_filters.FilterSet):
|
|
site = django_filters.MethodFilter(
|
|
action='filter_site',
|
|
label='Site (slug)',
|
|
)
|
|
|
|
class Meta:
|
|
model = PowerOutlet
|
|
|
|
def filter_site(self, queryset, value):
|
|
value = value.strip()
|
|
if not value:
|
|
return queryset
|
|
return queryset.filter(power_outlet__device__rack__site__slug=value)
|
|
|
|
|
|
class InterfaceConnectionFilter(django_filters.FilterSet):
|
|
site = django_filters.MethodFilter(
|
|
action='filter_site',
|
|
label='Site (slug)',
|
|
)
|
|
|
|
class Meta:
|
|
model = InterfaceConnection
|
|
|
|
def filter_site(self, queryset, value):
|
|
value = value.strip()
|
|
if not value:
|
|
return queryset
|
|
return queryset.filter(
|
|
Q(interface_a__device__rack__site__slug=value) |
|
|
Q(interface_b__device__rack__site__slug=value)
|
|
)
|