82a440f5af
Make it possible to specify multiple MAC addresses and use a textarea field for entering them. Change-Id: I1a2241d591f4174e08e7c7cc560f4e43facc6457
105 lines
3.2 KiB
Python
105 lines
3.2 KiB
Python
#
|
|
# Licensed under the Apache License, Version 2.0 (the "License"); you may
|
|
# not use this file except in compliance with the License. You may obtain
|
|
# a copy of the License at
|
|
#
|
|
# http://www.apache.org/licenses/LICENSE-2.0
|
|
#
|
|
# Unless required by applicable law or agreed to in writing, software
|
|
# distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
|
|
# WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
|
|
# License for the specific language governing permissions and limitations
|
|
# under the License.
|
|
|
|
import re
|
|
|
|
from django import forms
|
|
from django.utils.translation import ugettext_lazy as _
|
|
import netaddr
|
|
|
|
|
|
SEPARATOR_RE = re.compile('[\s,;|]+', re.UNICODE)
|
|
|
|
|
|
def fieldset(self, *args, **kwargs):
|
|
"""A helper function for grouping fields based on their names."""
|
|
|
|
prefix = kwargs.pop('prefix', None)
|
|
names = args or self.fields.keys()
|
|
for name in names:
|
|
if prefix is None or name.startswith(prefix):
|
|
yield forms.forms.BoundField(self, self.fields[name], name)
|
|
|
|
|
|
class MACDialect(netaddr.mac_eui48):
|
|
"""For validating MAC addresses. Same validation as Nova uses."""
|
|
word_fmt = '%.02x'
|
|
word_sep = ':'
|
|
|
|
|
|
def normalize_MAC(value):
|
|
try:
|
|
return str(netaddr.EUI(
|
|
value.strip(), version=48, dialect=MACDialect)).upper()
|
|
except (netaddr.AddrFormatError, TypeError):
|
|
raise ValueError('Invalid MAC address')
|
|
|
|
|
|
class NumberInput(forms.widgets.TextInput):
|
|
"""A form input for numbers."""
|
|
input_type = 'number'
|
|
|
|
|
|
class NumberPickerInput(NumberInput):
|
|
"""A form input that is rendered as a big number picker."""
|
|
|
|
def __init__(self, attrs=None):
|
|
default_attrs = {'hr-number-picker': '', 'ng-cloak': '', }
|
|
if attrs:
|
|
default_attrs.update(attrs)
|
|
super(NumberPickerInput, self).__init__(default_attrs)
|
|
|
|
|
|
class MACField(forms.fields.Field):
|
|
"""A form field for entering a single MAC address."""
|
|
|
|
def clean(self, value):
|
|
value = super(MACField, self).clean(value)
|
|
try:
|
|
return normalize_MAC(value)
|
|
except ValueError:
|
|
raise forms.ValidationError(_(u'Enter a valid MAC address.'))
|
|
|
|
|
|
class MultiMACField(forms.fields.Field):
|
|
"""A form field for entering multiple MAC addresses.
|
|
|
|
The individual MAC addresses can be separated by any whitespace,
|
|
commas, semicolons or pipe characters.
|
|
|
|
Gives a string of normalized MAC addresses separated by spaces.
|
|
"""
|
|
|
|
def clean(self, value):
|
|
value = super(MultiMACField, self).clean(value)
|
|
try:
|
|
macs = []
|
|
for mac in SEPARATOR_RE.split(value):
|
|
if mac:
|
|
macs.append(normalize_MAC(mac))
|
|
return ' '.join(macs)
|
|
except ValueError:
|
|
raise forms.ValidationError(
|
|
_(u'%r is not a valid MAC address.') % mac)
|
|
|
|
|
|
class NetworkField(forms.fields.Field):
|
|
"""A form field for entering a network specification with a mask."""
|
|
|
|
def clean(self, value):
|
|
value = super(NetworkField, self).clean(value)
|
|
try:
|
|
return str(netaddr.IPNetwork(value, version=4))
|
|
except netaddr.AddrFormatError:
|
|
raise forms.ValidationError(_("Enter valid IPv4 network address."))
|