|
@@ -11,6 +11,10 @@ from .models import RIR, Aggregate, VRF, Prefix, IPAddress, VLAN, VLANGroup, Rol
|
|
|
|
|
|
|
|
|
class VRFFilter(django_filters.FilterSet):
|
|
|
+ q = django_filters.MethodFilter(
|
|
|
+ action='search',
|
|
|
+ label='Search',
|
|
|
+ )
|
|
|
name = django_filters.CharFilter(
|
|
|
name='name',
|
|
|
lookup_type='icontains',
|
|
@@ -28,12 +32,23 @@ class VRFFilter(django_filters.FilterSet):
|
|
|
label='Tenant (slug)',
|
|
|
)
|
|
|
|
|
|
+ def search(self, queryset, value):
|
|
|
+ return queryset.filter(
|
|
|
+ Q(name__icontains=value) |
|
|
|
+ Q(rd__icontains=value) |
|
|
|
+ Q(description__icontains=value)
|
|
|
+ )
|
|
|
+
|
|
|
class Meta:
|
|
|
model = VRF
|
|
|
fields = ['name', 'rd']
|
|
|
|
|
|
|
|
|
class AggregateFilter(django_filters.FilterSet):
|
|
|
+ q = django_filters.MethodFilter(
|
|
|
+ action='search',
|
|
|
+ label='Search',
|
|
|
+ )
|
|
|
rir_id = django_filters.ModelMultipleChoiceFilter(
|
|
|
name='rir',
|
|
|
queryset=RIR.objects.all(),
|
|
@@ -50,6 +65,15 @@ class AggregateFilter(django_filters.FilterSet):
|
|
|
model = Aggregate
|
|
|
fields = ['family', 'rir_id', 'rir', 'date_added']
|
|
|
|
|
|
+ def search(self, queryset, value):
|
|
|
+ qs_filter = Q(description__icontains=value)
|
|
|
+ try:
|
|
|
+ prefix = str(IPNetwork(value.strip()).cidr)
|
|
|
+ qs_filter |= Q(prefix__net_contains_or_equals=prefix)
|
|
|
+ except AddrFormatError:
|
|
|
+ pass
|
|
|
+ return queryset.filter(qs_filter)
|
|
|
+
|
|
|
|
|
|
class PrefixFilter(django_filters.FilterSet):
|
|
|
q = django_filters.MethodFilter(
|
|
@@ -114,12 +138,13 @@ class PrefixFilter(django_filters.FilterSet):
|
|
|
fields = ['family', 'site_id', 'site', 'vrf', 'vrf_id', 'vlan_id', 'vlan_vid', 'status', 'role_id', 'role']
|
|
|
|
|
|
def search(self, queryset, value):
|
|
|
- value = value.strip()
|
|
|
+ qs_filter = Q(description__icontains=value)
|
|
|
try:
|
|
|
- query = str(IPNetwork(value).cidr)
|
|
|
- return queryset.filter(prefix__net_contains_or_equals=query)
|
|
|
+ prefix = str(IPNetwork(value.strip()).cidr)
|
|
|
+ qs_filter |= Q(prefix__net_contains_or_equals=prefix)
|
|
|
except AddrFormatError:
|
|
|
- return queryset.none()
|
|
|
+ pass
|
|
|
+ return queryset.filter(qs_filter)
|
|
|
|
|
|
def search_by_parent(self, queryset, value):
|
|
|
value = value.strip()
|
|
@@ -205,12 +230,13 @@ class IPAddressFilter(django_filters.FilterSet):
|
|
|
fields = ['q', 'family', 'vrf_id', 'vrf', 'device_id', 'device', 'interface_id']
|
|
|
|
|
|
def search(self, queryset, value):
|
|
|
- value = value.strip()
|
|
|
+ qs_filter = Q(description__icontains=value)
|
|
|
try:
|
|
|
- query = str(IPNetwork(value))
|
|
|
- return queryset.filter(address__net_host=query)
|
|
|
+ ipaddress = str(IPNetwork(value.strip()))
|
|
|
+ qs_filter |= Q(address__net_host=ipaddress)
|
|
|
except AddrFormatError:
|
|
|
- return queryset.none()
|
|
|
+ pass
|
|
|
+ return queryset.filter(qs_filter)
|
|
|
|
|
|
def _vrf(self, queryset, value):
|
|
|
if str(value) == '':
|
|
@@ -261,6 +287,10 @@ class VLANGroupFilter(django_filters.FilterSet):
|
|
|
|
|
|
|
|
|
class VLANFilter(django_filters.FilterSet):
|
|
|
+ q = django_filters.MethodFilter(
|
|
|
+ action='search',
|
|
|
+ label='Search',
|
|
|
+ )
|
|
|
site_id = django_filters.ModelMultipleChoiceFilter(
|
|
|
name='site',
|
|
|
queryset=Site.objects.all(),
|
|
@@ -318,3 +348,11 @@ class VLANFilter(django_filters.FilterSet):
|
|
|
class Meta:
|
|
|
model = VLAN
|
|
|
fields = ['site_id', 'site', 'vid', 'name', 'status', 'role_id', 'role']
|
|
|
+
|
|
|
+ def search(self, queryset, value):
|
|
|
+ qs_filter = Q(name__icontains=value) | Q(description__icontains=value)
|
|
|
+ try:
|
|
|
+ qs_filter |= Q(vid=int(value))
|
|
|
+ except ValueError:
|
|
|
+ pass
|
|
|
+ return queryset.filter(qs_filter)
|