Add IPNetworkVar

This commit is contained in:
Jeremy Stretch 2019-08-13 09:48:51 -04:00
parent f1e12213e7
commit a0d181728a

View File

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