From dab30f50d31ac2a7eedade519488c55df7a50206 Mon Sep 17 00:00:00 2001 From: Jeremy Stretch 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.