uncloud/matrixhosting/forms.py

import tldextract

from django import forms
from django.forms import ModelForm
from django.utils.translation import get_language, ugettext_lazy as _
from django.core.exceptions import ValidationError
from .validators import domain_name_validator
from uncloud_pay.models import BillingAddress


class DomainNameField(forms.CharField):
    description = 'Domain name form field'
    default_validators = [domain_name_validator, ]

    def __init__(self,  *args, **kwargs):
        super(DomainNameField, self).__init__(*args, **kwargs)

class MainModelForm(ModelForm):
    def __init__(self, *args, **kwargs):
        super(MainModelForm, self).__init__(*args, **kwargs)
        for visible in self.visible_fields():
            visible.field.widget.attrs['class'] = 'form-control'

class MainForm(forms.Form):
    def __init__(self, *args, **kwargs):
        super(MainForm, self).__init__(*args, **kwargs)
        for visible in self.visible_fields():
            if (isinstance(visible.field.widget, forms.TextInput)):
                visible.field.widget.attrs['class'] = 'form-control'
            elif (isinstance(visible.field.widget, forms.CheckboxInput)):
                visible.field.widget.attrs['class'] = 'custom-control-input'


class InitialRequestForm(MainForm):
    cores = forms.IntegerField(label='CPU', min_value=1, max_value=48, initial=1)
    memory = forms.IntegerField(label='RAM', min_value=2, max_value=200, initial=2)
    storage = forms.IntegerField(label='Storage', min_value=100, max_value=10000, initial=100)
    pricing_name = forms.CharField(required=True)

class RequestHostedVMForm(InitialRequestForm):
    matrix_domain = DomainNameField(required=True, widget=forms.TextInput(attrs={'placeholder': 'Matrix Domain *'}))
    homeserver_domain = DomainNameField(required=True, widget=forms.TextInput(attrs={'placeholder': 'Homeserver Domain *'}))
    webclient_domain = DomainNameField(required=True, widget=forms.TextInput(attrs={'placeholder': 'Webclient Domain *'}))
    is_open_registration = forms.BooleanField(required=False, initial=False)

    def clean(self):
        homeserver_domain = self.cleaned_data.get('homeserver_domain', False)
        webclient_domain = self.cleaned_data.get('webclient_domain', False)
        if homeserver_domain and webclient_domain:
            # Homserver-Domain and Webclient-Domain cannot be below the same second level domain (i.e. homeserver.abc.ch and webclient.def.cloud are ok, 
            # homeserver.abc.ch and webclient.abc.ch are not ok
            homeserver_base = tldextract.extract(homeserver_domain).domain
            webclient_base = tldextract.extract(webclient_domain).domain
            if homeserver_base == webclient_base:
                self._errors['webclient_domain'] = self.error_class([
                    'Homserver-Domain and Webclient-Domain cannot be below the same second level domain'])
        return self.cleaned_data


class BillingAddressForm(MainModelForm):
    class Meta:
        model = BillingAddress
        fields = ['full_name', 'street',
                  'city', 'postal_code', 'country', 'vat_number', 'active', 'owner']