webvirtcloud/storages/forms.py

import re

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


class AddStgPool(forms.Form):
    name = forms.CharField(
        error_messages={"required": _("No pool name has been entered")}, max_length=20
    )
    stg_type = forms.CharField(max_length=10)
    target = forms.CharField(
        error_messages={"required": _("No path has been entered")},
        max_length=100,
        required=False,
    )
    source = forms.CharField(max_length=100, required=False)
    ceph_user = forms.CharField(required=False)
    ceph_host = forms.CharField(required=False)
    ceph_pool = forms.CharField(required=False)
    secret = forms.CharField(required=False)
    netfs_host = forms.CharField(required=False)
    source_format = forms.CharField(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_target(self):
        storage_type = self.cleaned_data["stg_type"]
        target = self.cleaned_data["target"]
        have_symbol = re.match("^[^-](\/?((?:[0-9a-zA-Z_])|(?:\.)|((?<!\/)-))*)+$", target)
        if storage_type in ["dir", "netfs"] and not have_symbol:
            raise forms.ValidationError(
                _("The target must not contain any special characters")
            )
        if storage_type in ["dir", "netfs"] and not target:
            raise forms.ValidationError(_("No path has been entered"))
        return target

    def clean_source(self):
        storage_type = self.cleaned_data["stg_type"]
        source = self.cleaned_data["source"]
        have_symbol = re.match("^[a-zA-Z0-9\/]+$", source)
        if storage_type in ["logical", "netfs"]:
            if not source:
                raise forms.ValidationError(_("No device or path has been entered"))
            if not have_symbol:
                raise forms.ValidationError(
                    _("The disk source must not contain any special characters")
                )
        return source


class CreateVolumeForm(forms.Form):
    name = forms.CharField(max_length=120)
    format = forms.ChoiceField(
        required=True,
        choices=(("qcow2", "qcow2 (recommended)"), ("qcow", "qcow"), ("raw", "raw")),
    )
    size = forms.IntegerField()
    meta_prealloc = 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 image name must not contain any special characters")
            )
        return name


class CloneImage(forms.Form):
    name = forms.CharField(max_length=120)
    image = forms.CharField(max_length=120)
    convert = forms.BooleanField(required=False)
    format = forms.ChoiceField(
        required=False,
        choices=(("qcow2", "qcow2 (recommended)"), ("qcow", "qcow"), ("raw", "raw")),
    )
    meta_prealloc = 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 image name must not contain any special characters")
            )
        elif len(name) > 120:
            raise forms.ValidationError(
                _("The image name must not exceed 120 characters")
            )
        return name