webvirtcloud/interfaces/forms.py

import re

from django import forms
from django.utils.translation import gettext_lazy as _


class AddInterface(forms.Form):
    name = forms.CharField(max_length=10, required=True)
    itype = forms.ChoiceField(
        required=True, choices=(("bridge", "bridge"), ("ethernet", "ethernet"))
    )
    start_mode = forms.ChoiceField(
        required=True,
        choices=(("none", "none"), ("onboot", "onboot"), ("hotplug", "hotplug")),
    )
    netdev = forms.CharField(max_length=15, required=True)
    ipv4_type = forms.ChoiceField(
        required=True,
        choices=(("dhcp", "dhcp"), ("static", "static"), ("none", "none")),
    )
    ipv4_addr = forms.CharField(max_length=18, required=False)
    ipv4_gw = forms.CharField(max_length=15, required=False)
    ipv6_type = forms.ChoiceField(
        required=True,
        choices=(("dhcp", "dhcp"), ("static", "static"), ("none", "none")),
    )
    ipv6_addr = forms.CharField(max_length=100, required=False)
    ipv6_gw = forms.CharField(max_length=100, required=False)
    stp = forms.ChoiceField(required=False, choices=(("on", "on"), ("off", "off")))
    delay = forms.IntegerField(required=False)

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

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

    def clean_ipv6_addr(self):
        ipv6_addr = self.cleaned_data["ipv6_addr"]
        have_symbol = re.match("^[0-9a-f./:]+|^$", ipv6_addr)
        if not have_symbol:
            raise forms.ValidationError(
                _("The IPv6 address must not contain any special characters")
            )
        elif len(ipv6_addr) > 100:
            raise forms.ValidationError(
                _("The IPv6 address must not exceed 100 characters")
            )
        return ipv6_addr

    def clean_ipv6_gw(self):
        ipv6_gw = self.cleaned_data["ipv6_gw"]
        have_symbol = re.match("^[0-9a-f./:]+|^$", ipv6_gw)
        if not have_symbol:
            raise forms.ValidationError(
                _("The IPv6 gateway must not contain any special characters")
            )
        elif len(ipv6_gw) > 100:
            raise forms.ValidationError(
                _("The IPv6 gateway must not exceed 100 characters")
            )
        return ipv6_gw

    def clean_name(self):
        name = self.cleaned_data["name"]
        have_symbol = re.match("^[a-z0-9.]+$", name)
        if not have_symbol:
            raise forms.ValidationError(
                _("The interface must not contain any special characters")
            )
        elif len(name) > 10:
            raise forms.ValidationError(
                _("The interface must not exceed 10 characters")
            )
        return name

    def clean_netdev(self):
        netdev = self.cleaned_data["netdev"]
        have_symbol = re.match("^[a-z0-9.:]+$", netdev)
        if not have_symbol:
            raise forms.ValidationError(
                _("The interface must not contain any special characters")
            )
        elif len(netdev) > 10:
            raise forms.ValidationError(
                _("The interface must not exceed 10 characters")
            )
        return netdev