Merge pull request #36 from cpaillet/snmp

feat(snmp): add device search

Author: cpaillet

netbox_cmdb/netbox_cmdb/admin.py

@@ -185,6 +185,8 @@ class SNMPAdmin(BaseAdmin):
         "contact",
     )
 
+    search_fields = ("device__name", "location")
+
     def community_list_display(self, obj):
         return ", ".join([str(community) for community in obj.community_list.all()])
 

netbox_cmdb/netbox_cmdb/filtersets.py

@@ -1,5 +1,6 @@
 import django_filters
 from django.db.models import Q
+from netbox_cmdb.models.snmp import SNMP
 from tenancy.filtersets import TenancyFilterSet
 from utilities.filters import MultiValueCharFilter
 
@@ -179,3 +180,21 @@ def search(self, queryset, name, value):
         return queryset.filter(
             Q(device__name__icontains=value) | Q(name__icontains=value)
         ).distinct()
+
+
+class SNMPFilterSet(ChangeLoggedModelFilterSet):
+    """AS number filterset."""
+
+    q = django_filters.CharFilter(
+        method="search",
+        label="Search",
+    )
+
+    class Meta:
+        model = SNMP
+        fields = ["device"]
+
+    def search(self, queryset, name, value):
+        if not value.strip():
+            return queryset
+        return queryset.filter(Q(device__name__icontains=value)).distinct()

netbox_cmdb/netbox_cmdb/views.py

@@ -11,6 +11,7 @@
     ASNFilterSet,
     BGPPeerGroupFilterSet,
     BGPSessionFilterSet,
+    SNMPFilterSet,
 )
 from netbox_cmdb.forms import (
     ASNForm,
@@ -124,6 +125,7 @@ class BGPPeerGroupView(ObjectView):
 ## Snmp groups views
 class SNMPListView(ObjectListView):
     queryset = SNMP.objects.all()
+    filterset = SNMPFilterSet
     table = SNMPTable