From dab30f50d31ac2a7eedade519488c55df7a50206 Mon Sep 17 00:00:00 2001
From: Jeremy Stretch <stretch@packetlife.net>
Date: Tue, 13 Aug 2019 09:48:51 -0400
Subject: [PATCH] Add IPNetworkVar

---
 netbox/extras/scripts.py | 13 +++++++++++++
 1 file changed, 13 insertions(+)

diff --git a/netbox/extras/scripts.py b/netbox/extras/scripts.py
index c5bb674f5..121502f82 100644
--- a/netbox/extras/scripts.py
+++ b/netbox/extras/scripts.py
@@ -7,6 +7,7 @@ from django.conf import settings
 from django.core.validators import RegexValidator
 from django.db import transaction
 
+from ipam.formfields import IPFormField
 from utilities.exceptions import AbortTransaction
 from .constants import LOG_DEFAULT, LOG_FAILURE, LOG_INFO, LOG_SUCCESS, LOG_WARNING
 from .forms import ScriptForm
@@ -18,6 +19,7 @@ __all__ = [
     'IntegerVar',
     'BooleanVar',
     'ObjectVar',
+    'IPNetworkVar',
 ]
 
 
@@ -116,6 +118,17 @@ class ObjectVar(ScriptVariable):
         self.field_attrs['queryset'] = queryset
 
 
+class IPNetworkVar(ScriptVariable):
+    """
+    An IPv4 or IPv6 prefix.
+    """
+    form_field = IPFormField
+
+
+#
+# Scripts
+#
+
 class Script:
     """
     Custom scripts inherit this object.