From 472486acd60e1e4c37bbba24eb3e967c95d690c0 Mon Sep 17 00:00:00 2001
From: Saria Hajjar <shajjar@mc-lon-mb11797.local>
Date: Thu, 9 Jan 2020 16:26:11 +0000
Subject: [PATCH] Changed to `q` filter

---
 netbox/ipam/forms.py                        | 10 +++-------
 netbox/ipam/views.py                        | 12 +++++-------
 netbox/templates/ipam/ipaddress_assign.html |  5 ++---
 3 files changed, 10 insertions(+), 17 deletions(-)

diff --git a/netbox/ipam/forms.py b/netbox/ipam/forms.py
index f4bcfb2e6..c3387a5aa 100644
--- a/netbox/ipam/forms.py
+++ b/netbox/ipam/forms.py
@@ -933,7 +933,7 @@ class IPAddressBulkEditForm(BootstrapMixin, AddRemoveTagsForm, CustomFieldBulkEd
 
 
 class IPAddressAssignForm(BootstrapMixin, forms.Form):
-    vrf = forms.ModelChoiceField(
+    vrf_id = forms.ModelChoiceField(
         queryset=VRF.objects.all(),
         required=False,
         label='VRF',
@@ -942,13 +942,9 @@ class IPAddressAssignForm(BootstrapMixin, forms.Form):
             api_url="/api/ipam/vrfs/"
         )
     )
-    address = forms.CharField(
-        label='IP Address',
-        required=False,
-    )
-    dns_name = forms.CharField(
-        label='DNS Name',
+    q = forms.CharField(
         required=False,
+        label='Search',
     )
 
 
diff --git a/netbox/ipam/views.py b/netbox/ipam/views.py
index d6cada398..084cf1205 100644
--- a/netbox/ipam/views.py
+++ b/netbox/ipam/views.py
@@ -749,14 +749,12 @@ class IPAddressAssignView(PermissionRequiredMixin, View):
 
         if form.is_valid():
 
-            queryset = IPAddress.objects.prefetch_related(
+            addresses = IPAddress.objects.prefetch_related(
                 'vrf', 'tenant', 'interface__device', 'interface__virtual_machine'
-            ).filter(
-                vrf=form.cleaned_data['vrf'],
-                address__istartswith=form.cleaned_data['address'],
-                dns_name__icontains=form.cleaned_data['dns_name'],
-            )[:100]  # Limit to 100 results
-            table = tables.IPAddressAssignTable(queryset)
+            )
+            # Limit to 100 results
+            addresses = filters.IPAddressFilter(request.POST, addresses).qs[:100]
+            table = tables.IPAddressAssignTable(addresses)
 
         return render(request, 'ipam/ipaddress_assign.html', {
             'form': form,
diff --git a/netbox/templates/ipam/ipaddress_assign.html b/netbox/templates/ipam/ipaddress_assign.html
index 349c06919..ab163533f 100644
--- a/netbox/templates/ipam/ipaddress_assign.html
+++ b/netbox/templates/ipam/ipaddress_assign.html
@@ -24,9 +24,8 @@
             <div class="panel panel-default">
                 <div class="panel-heading"><strong>Select IP Address</strong></div>
                 <div class="panel-body">
-                    {% render_field form.vrf %}
-                    {% render_field form.address %}
-                    {% render_field form.dns_name %}
+                    {% render_field form.vrf_id %}
+                    {% render_field form.q %}
                 </div>
             </div>
             </div>