webvirtcloud/networks/forms.py

import re
from django import forms
from django.utils.translation import ugettext_lazy as _


class AddNetPool(forms.Form):
    name = forms.CharField(error_messages={'required': _('No pool name has been entered')},
                           max_length=20)
    subnet = forms.CharField(error_messages={'required': _('No IPv4 subnet has been entered')},
                             max_length=20, required=False)
    subnet6 = forms.CharField(error_messages={'required': _('No IPv6 subnet has been entered')},
                              max_length=42, required=False)
    forward = forms.CharField(max_length=100)
    dhcp4 = forms.BooleanField(required=False)
    dhcp6 = forms.BooleanField(required=False)
    fixed = forms.BooleanField(required=False)
    bridge_name = forms.CharField(max_length=20, required=False)
    openvswitch = forms.BooleanField(required=False)

    def clean_name(self):
        name = self.cleaned_data['name']
        have_symbol = re.match('^[a-zA-Z0-9\.\_\-]+$', name)
        if not have_symbol:
            raise forms.ValidationError(_('The pool name must not contain any special characters'))
        elif len(name) > 20:
            raise forms.ValidationError(_('The pool name must not exceed 20 characters'))
        return name

    def clean_subnet(self):
        subnet = self.cleaned_data['subnet']
        have_symbol = re.match('^[0-9./]+$', subnet if subnet else ".")
        if not have_symbol:
            raise forms.ValidationError(_('The IPv4 subnet must not contain any special characters'))
        elif len(subnet) > 20:
            raise forms.ValidationError(_('The IPv4 subnet must not exceed 20 characters'))
        return subnet

    def clean_subnet6(self):
        subnet = self.cleaned_data['subnet6']
        have_symbol = re.match('^[0-9a-fA-F:/]+$', subnet if subnet else ":")
        if not have_symbol:
            raise forms.ValidationError(_('The IPv6 subnet must not contain any special characters'))
        elif len(subnet) > 42:
            raise forms.ValidationError(_('The IPv6 subnet must not exceed 42 characters'))
        return subnet

    def clean_bridge_name(self):
        bridge_name = self.cleaned_data['bridge_name']
        if self.cleaned_data['forward'] in ['bridge', 'macvtap']:
            have_symbol = re.match('^[a-zA-Z0-9\.\_\:\-]+$', bridge_name)
            if not have_symbol:
                raise forms.ValidationError(_('The pool bridge name must not contain any special characters'))
            elif len(bridge_name) > 20:
                raise forms.ValidationError(_('The pool bridge name must not exceed 20 characters'))
            return bridge_name