|
@@ -13,15 +13,10 @@ from .models import Aggregate, IPAddress, Prefix, RIR, Role, Service, VLAN, VLAN
|
|
|
|
|
|
|
|
|
class VRFFilter(CustomFieldFilterSet, django_filters.FilterSet):
|
|
|
- q = django_filters.MethodFilter(
|
|
|
- action='search',
|
|
|
+ q = django_filters.CharFilter(
|
|
|
+ method='search',
|
|
|
label='Search',
|
|
|
)
|
|
|
- name = django_filters.CharFilter(
|
|
|
- name='name',
|
|
|
- lookup_type='icontains',
|
|
|
- label='Name',
|
|
|
- )
|
|
|
tenant_id = NullableModelMultipleChoiceFilter(
|
|
|
name='tenant',
|
|
|
queryset=Tenant.objects.all(),
|
|
@@ -34,7 +29,9 @@ class VRFFilter(CustomFieldFilterSet, django_filters.FilterSet):
|
|
|
label='Tenant (slug)',
|
|
|
)
|
|
|
|
|
|
- def search(self, queryset, value):
|
|
|
+ def search(self, queryset, name, value):
|
|
|
+ if not value.strip():
|
|
|
+ return queryset
|
|
|
return queryset.filter(
|
|
|
Q(name__icontains=value) |
|
|
|
Q(rd__icontains=value) |
|
|
@@ -43,7 +40,7 @@ class VRFFilter(CustomFieldFilterSet, django_filters.FilterSet):
|
|
|
|
|
|
class Meta:
|
|
|
model = VRF
|
|
|
- fields = ['rd']
|
|
|
+ fields = ['name', 'rd']
|
|
|
|
|
|
|
|
|
class RIRFilter(django_filters.FilterSet):
|
|
@@ -54,8 +51,8 @@ class RIRFilter(django_filters.FilterSet):
|
|
|
|
|
|
|
|
|
class AggregateFilter(CustomFieldFilterSet, django_filters.FilterSet):
|
|
|
- q = django_filters.MethodFilter(
|
|
|
- action='search',
|
|
|
+ q = django_filters.CharFilter(
|
|
|
+ method='search',
|
|
|
label='Search',
|
|
|
)
|
|
|
rir_id = django_filters.ModelMultipleChoiceFilter(
|
|
@@ -74,7 +71,9 @@ class AggregateFilter(CustomFieldFilterSet, django_filters.FilterSet):
|
|
|
model = Aggregate
|
|
|
fields = ['family', 'date_added']
|
|
|
|
|
|
- def search(self, queryset, value):
|
|
|
+ def search(self, queryset, name, value):
|
|
|
+ if not value.strip():
|
|
|
+ return queryset
|
|
|
qs_filter = Q(description__icontains=value)
|
|
|
try:
|
|
|
prefix = str(IPNetwork(value.strip()).cidr)
|
|
@@ -85,12 +84,12 @@ class AggregateFilter(CustomFieldFilterSet, django_filters.FilterSet):
|
|
|
|
|
|
|
|
|
class PrefixFilter(CustomFieldFilterSet, django_filters.FilterSet):
|
|
|
- q = django_filters.MethodFilter(
|
|
|
- action='search',
|
|
|
+ q = django_filters.CharFilter(
|
|
|
+ method='search',
|
|
|
label='Search',
|
|
|
)
|
|
|
- parent = django_filters.MethodFilter(
|
|
|
- action='search_by_parent',
|
|
|
+ parent = django_filters.CharFilter(
|
|
|
+ method='search_by_parent',
|
|
|
label='Parent prefix',
|
|
|
)
|
|
|
vrf_id = NullableModelMultipleChoiceFilter(
|
|
@@ -151,7 +150,9 @@ class PrefixFilter(CustomFieldFilterSet, django_filters.FilterSet):
|
|
|
model = Prefix
|
|
|
fields = ['family', 'status']
|
|
|
|
|
|
- def search(self, queryset, value):
|
|
|
+ def search(self, queryset, name, value):
|
|
|
+ if not value.strip():
|
|
|
+ return queryset
|
|
|
qs_filter = Q(description__icontains=value)
|
|
|
try:
|
|
|
prefix = str(IPNetwork(value.strip()).cidr)
|
|
@@ -160,7 +161,7 @@ class PrefixFilter(CustomFieldFilterSet, django_filters.FilterSet):
|
|
|
pass
|
|
|
return queryset.filter(qs_filter)
|
|
|
|
|
|
- def search_by_parent(self, queryset, value):
|
|
|
+ def search_by_parent(self, queryset, name, value):
|
|
|
value = value.strip()
|
|
|
if not value:
|
|
|
return queryset
|
|
@@ -170,32 +171,14 @@ class PrefixFilter(CustomFieldFilterSet, django_filters.FilterSet):
|
|
|
except AddrFormatError:
|
|
|
return queryset.none()
|
|
|
|
|
|
- def _tenant(self, queryset, value):
|
|
|
- if str(value) == '':
|
|
|
- return queryset
|
|
|
- return queryset.filter(
|
|
|
- Q(tenant__slug=value) |
|
|
|
- Q(tenant__isnull=True, vrf__tenant__slug=value)
|
|
|
- )
|
|
|
-
|
|
|
- def _tenant_id(self, queryset, value):
|
|
|
- try:
|
|
|
- value = int(value)
|
|
|
- except ValueError:
|
|
|
- return queryset.none()
|
|
|
- return queryset.filter(
|
|
|
- Q(tenant__pk=value) |
|
|
|
- Q(tenant__isnull=True, vrf__tenant__pk=value)
|
|
|
- )
|
|
|
-
|
|
|
|
|
|
class IPAddressFilter(CustomFieldFilterSet, django_filters.FilterSet):
|
|
|
- q = django_filters.MethodFilter(
|
|
|
- action='search',
|
|
|
+ q = django_filters.CharFilter(
|
|
|
+ method='search',
|
|
|
label='Search',
|
|
|
)
|
|
|
- parent = django_filters.MethodFilter(
|
|
|
- action='search_by_parent',
|
|
|
+ parent = django_filters.CharFilter(
|
|
|
+ method='search_by_parent',
|
|
|
label='Parent prefix',
|
|
|
)
|
|
|
vrf_id = NullableModelMultipleChoiceFilter(
|
|
@@ -239,9 +222,11 @@ class IPAddressFilter(CustomFieldFilterSet, django_filters.FilterSet):
|
|
|
|
|
|
class Meta:
|
|
|
model = IPAddress
|
|
|
- fields = ['q', 'family', 'status']
|
|
|
+ fields = ['family', 'status']
|
|
|
|
|
|
- def search(self, queryset, value):
|
|
|
+ def search(self, queryset, name, value):
|
|
|
+ if not value.strip():
|
|
|
+ return queryset
|
|
|
qs_filter = Q(description__icontains=value)
|
|
|
try:
|
|
|
ipaddress = str(IPNetwork(value.strip()))
|
|
@@ -250,12 +235,12 @@ class IPAddressFilter(CustomFieldFilterSet, django_filters.FilterSet):
|
|
|
pass
|
|
|
return queryset.filter(qs_filter)
|
|
|
|
|
|
- def search_by_parent(self, queryset, value):
|
|
|
+ def search_by_parent(self, queryset, name, value):
|
|
|
value = value.strip()
|
|
|
if not value:
|
|
|
return queryset
|
|
|
try:
|
|
|
- query = str(IPNetwork(value).cidr)
|
|
|
+ query = str(IPNetwork(value.strip()).cidr)
|
|
|
return queryset.filter(address__net_contained_or_equal=query)
|
|
|
except AddrFormatError:
|
|
|
return queryset.none()
|
|
@@ -276,11 +261,12 @@ class VLANGroupFilter(django_filters.FilterSet):
|
|
|
|
|
|
class Meta:
|
|
|
model = VLANGroup
|
|
|
+ fields = ['name']
|
|
|
|
|
|
|
|
|
class VLANFilter(CustomFieldFilterSet, django_filters.FilterSet):
|
|
|
- q = django_filters.MethodFilter(
|
|
|
- action='search',
|
|
|
+ q = django_filters.CharFilter(
|
|
|
+ method='search',
|
|
|
label='Search',
|
|
|
)
|
|
|
site_id = NullableModelMultipleChoiceFilter(
|
|
@@ -305,15 +291,6 @@ class VLANFilter(CustomFieldFilterSet, django_filters.FilterSet):
|
|
|
to_field_name='slug',
|
|
|
label='Group',
|
|
|
)
|
|
|
- name = django_filters.CharFilter(
|
|
|
- name='name',
|
|
|
- lookup_type='icontains',
|
|
|
- label='Name',
|
|
|
- )
|
|
|
- vid = django_filters.NumberFilter(
|
|
|
- name='vid',
|
|
|
- label='VLAN number (1-4095)',
|
|
|
- )
|
|
|
tenant_id = NullableModelMultipleChoiceFilter(
|
|
|
name='tenant',
|
|
|
queryset=Tenant.objects.all(),
|
|
@@ -339,12 +316,14 @@ class VLANFilter(CustomFieldFilterSet, django_filters.FilterSet):
|
|
|
|
|
|
class Meta:
|
|
|
model = VLAN
|
|
|
- fields = ['status']
|
|
|
+ fields = ['name', 'vid', 'status']
|
|
|
|
|
|
- def search(self, queryset, value):
|
|
|
+ def search(self, queryset, name, value):
|
|
|
+ if not value.strip():
|
|
|
+ return queryset
|
|
|
qs_filter = Q(name__icontains=value) | Q(description__icontains=value)
|
|
|
try:
|
|
|
- qs_filter |= Q(vid=int(value))
|
|
|
+ qs_filter |= Q(vid=int(value.strip()))
|
|
|
except ValueError:
|
|
|
pass
|
|
|
return queryset.filter(qs_filter)
|