gva/gnuviechadmin/websites/forms.py

"""
This module defines form classes for website editing.

"""
from __future__ import absolute_import

from crispy_forms.bootstrap import AppendedText
from crispy_forms.helper import FormHelper
from crispy_forms.layout import Layout, Submit
from django import forms
from django.urls import reverse
from django.utils.translation import gettext as _

from domains.forms import relative_domain_validator

from .models import Website


class AddWebsiteForm(forms.ModelForm):
    """
    This form is used to create new Website instances.

    """

    class Meta:
        model = Website
        fields = ["subdomain", "wildcard"]

    def __init__(self, *args, **kwargs):
        self.hosting_package = kwargs.pop("hostingpackage")
        self.hosting_domain = kwargs.pop("domain")
        super(AddWebsiteForm, self).__init__(*args, **kwargs)
        self.fields["subdomain"].validators.append(relative_domain_validator)
        if Website.objects.filter(wildcard=True, domain=self.hosting_domain).exists():
            self.fields["wildcard"].widget = forms.HiddenInput()

        self.helper = FormHelper()
        self.helper.form_action = reverse(
            "add_website",
            kwargs={
                "package": self.hosting_package.id,
                "domain": self.hosting_domain.domain,
            },
        )
        self.helper.layout = Layout(
            AppendedText("subdomain", "." + self.hosting_domain.domain),
            "wildcard",
            Submit("submit", _("Add website")),
        )

    def clean_subdomain(self):
        data = self.cleaned_data["subdomain"]
        if Website.objects.filter(domain=self.hosting_domain, subdomain=data).exists():
            raise forms.ValidationError(
                _("There is already a website for this subdomain")
            )
        relative_domain_validator("{0}.{1}".format(data, self.hosting_domain.domain))
        return data

    def save(self, commit=True):
        self.instance.domain = self.hosting_domain
        self.instance.osuser = self.hosting_package.osuser
        return super(AddWebsiteForm, self).save(commit)