|
|
@@ -1,6 +1,7 @@
|
|
|
from __future__ import unicode_literals
|
|
|
|
|
|
import django_filters
|
|
|
+from netaddr import EUI
|
|
|
from netaddr.core import AddrFormatError
|
|
|
|
|
|
from django.contrib.auth.models import User
|
|
|
@@ -8,7 +9,7 @@ from django.db.models import Q
|
|
|
|
|
|
from extras.filters import CustomFieldFilterSet
|
|
|
from tenancy.models import Tenant
|
|
|
-from utilities.filters import NullableModelMultipleChoiceFilter, NumericInFilter
|
|
|
+from utilities.filters import NullableCharFieldFilter, NullableModelMultipleChoiceFilter, NumericInFilter
|
|
|
from .models import (
|
|
|
ConsolePort, ConsolePortTemplate, ConsoleServerPort, ConsoleServerPortTemplate, Device, DeviceBay,
|
|
|
DeviceBayTemplate, DeviceRole, DeviceType, STATUS_CHOICES, IFACE_FF_LAG, Interface, InterfaceConnection,
|
|
|
@@ -113,6 +114,7 @@ class RackFilter(CustomFieldFilterSet, django_filters.FilterSet):
|
|
|
method='search',
|
|
|
label='Search',
|
|
|
)
|
|
|
+ facility_id = NullableCharFieldFilter()
|
|
|
site_id = django_filters.ModelMultipleChoiceFilter(
|
|
|
queryset=Site.objects.all(),
|
|
|
label='Site (ID)',
|
|
|
@@ -156,7 +158,7 @@ class RackFilter(CustomFieldFilterSet, django_filters.FilterSet):
|
|
|
|
|
|
class Meta:
|
|
|
model = Rack
|
|
|
- fields = ['facility_id', 'type', 'width', 'u_height', 'desc_units']
|
|
|
+ fields = ['type', 'width', 'u_height', 'desc_units']
|
|
|
|
|
|
def search(self, queryset, name, value):
|
|
|
if not value.strip():
|
|
|
@@ -383,6 +385,8 @@ class DeviceFilter(CustomFieldFilterSet, django_filters.FilterSet):
|
|
|
to_field_name='slug',
|
|
|
label='Platform (slug)',
|
|
|
)
|
|
|
+ name = NullableCharFieldFilter()
|
|
|
+ asset_tag = NullableCharFieldFilter()
|
|
|
site_id = django_filters.ModelMultipleChoiceFilter(
|
|
|
queryset=Site.objects.all(),
|
|
|
label='Site (ID)',
|
|
|
@@ -439,25 +443,33 @@ class DeviceFilter(CustomFieldFilterSet, django_filters.FilterSet):
|
|
|
|
|
|
class Meta:
|
|
|
model = Device
|
|
|
- fields = ['name', 'serial', 'asset_tag']
|
|
|
+ fields = ['serial']
|
|
|
|
|
|
def search(self, queryset, name, value):
|
|
|
if not value.strip():
|
|
|
return queryset
|
|
|
- return queryset.filter(
|
|
|
+ qs_filter = (
|
|
|
Q(name__icontains=value) |
|
|
|
Q(serial__icontains=value.strip()) |
|
|
|
Q(inventory_items__serial__icontains=value.strip()) |
|
|
|
Q(asset_tag=value.strip()) |
|
|
|
Q(comments__icontains=value)
|
|
|
- ).distinct()
|
|
|
+ )
|
|
|
+ # If the query value looks like a MAC address, search interfaces as well.
|
|
|
+ try:
|
|
|
+ mac = EUI(value.strip())
|
|
|
+ qs_filter |= Q(interfaces__mac_address=mac)
|
|
|
+ except AddrFormatError:
|
|
|
+ pass
|
|
|
+ return queryset.filter(qs_filter).distinct()
|
|
|
|
|
|
def _mac_address(self, queryset, name, value):
|
|
|
value = value.strip()
|
|
|
if not value:
|
|
|
return queryset
|
|
|
try:
|
|
|
- return queryset.filter(interfaces__mac_address=value).distinct()
|
|
|
+ mac = EUI(value.strip())
|
|
|
+ return queryset.filter(interfaces__mac_address=mac).distinct()
|
|
|
except AddrFormatError:
|
|
|
return queryset.none()
|
|
|
|
|
|
@@ -569,7 +581,8 @@ class InterfaceFilter(django_filters.FilterSet):
|
|
|
if not value:
|
|
|
return queryset
|
|
|
try:
|
|
|
- return queryset.filter(mac_address=value)
|
|
|
+ mac = EUI(value.strip())
|
|
|
+ return queryset.filter(mac_address=mac)
|
|
|
except AddrFormatError:
|
|
|
return queryset.none()
|
|
|
|
|
|
@@ -596,10 +609,11 @@ class InventoryItemFilter(DeviceComponentFilterSet):
|
|
|
to_field_name='slug',
|
|
|
label='Manufacturer (slug)',
|
|
|
)
|
|
|
+ asset_tag = NullableCharFieldFilter()
|
|
|
|
|
|
class Meta:
|
|
|
model = InventoryItem
|
|
|
- fields = ['name', 'part_id', 'serial', 'asset_tag', 'discovered']
|
|
|
+ fields = ['name', 'part_id', 'serial', 'discovered']
|
|
|
|
|
|
|
|
|
class ConsoleConnectionFilter(django_filters.FilterSet):
|