Move and rename a lot of stuff into kaehmy app

kaehmy/forms.py

@@ -0,0 +1,89 @@
+from django import forms
+from django.utils.translation import ugettext_lazy as _
+from django.core.exceptions import ValidationError
+
+from kaehmy.models import PresetRole, CustomRole, Application, Comment
+from webapp.models import BaseRole
+
+class CheckboxSelectMultiple(forms.widgets.CheckboxSelectMultiple):
+    option_template_name = 'checkbox_option.html'
+
+    def create_option(self, name, value, label, selected, index, subindex=None, attrs=None):
+        dic = super(CheckboxSelectMultiple, self).create_option(name, value, label, selected, index, subindex, attrs)
+        description = PresetRole.objects.get(id=value).description
+        dic['description'] = description
+        return dic
+
+    def __init__(self, *args, **kwargs):
+        super(CheckboxSelectMultiple, self).__init__(*args, **kwargs)
+
+
+class ApplicationForm(forms.ModelForm):
+    """Class representing Kaehmy form."""
+
+    class Meta:
+        """Meta for class Application."""
+
+        model = Application
+        fields = ['name', 'email', 'phone_number', 'year',
+                  'preset_roles', 'custom_roles', 'custom_role_name',
+                  'custom_role_is_board', 'text']
+
+    def __init__(self, *args, **kwargs):
+        super(ApplicationForm, self).__init__(*args, **kwargs)
+
+        self.fields["email"].label = _('Email (not public)')
+        self.fields["phone_number"].label = _('Phone number (not public)')
+
+        custom_roles_exist = CustomRole.objects.all().exists()
+        self.fields["custom_roles"].widget = forms.widgets.CheckboxSelectMultiple() if custom_roles_exist else forms.HiddenInput()
+        self.fields["custom_roles"].help_text = ""
+        self.fields["custom_roles"].label = _('Custom roles')
+        self.fields["custom_roles"].queryset = CustomRole.objects.all()
+
+        for cat_id, category in BaseRole.CATEGORIES:
+            key = 'preset_roles_{}'.format(cat_id)
+            qset = PresetRole.objects.filter(category=cat_id).order_by('category', '-is_board')
+            self.fields[key] = forms.ModelMultipleChoiceField(qset)
+            self.fields[key].widget = CheckboxSelectMultiple(attrs={
+                'title': _('Preset roles'),
+                'name': 'preset_roles',
+            })
+            self.fields[key].help_text = ""
+            self.fields[key].queryset = qset
+            self.fields[key].label = _(category)
+            self.fields[key].required = False
+
+    def clean(self):
+        cleaned_data = super(ApplicationForm, self).clean()
+        for key in cleaned_data.keys():
+            if 'preset_roles_' in key:
+                cleaned_data['preset_roles'] = cleaned_data['preset_roles'] | cleaned_data[key]
+
+        return cleaned_data
+
+    def clean_phone_number(self):
+        """Clean phone number field."""
+        number = self.cleaned_data.get('phone_number')
+        if number.isdigit():
+            return number
+        else:
+            raise ValidationError(_('Invalid phone number'))
+
+    def clean_custom_role_name(self):
+        """Check that no other custom role with same name exists."""
+        custom_name = self.cleaned_data.get('custom_role_name')
+        if not CustomRole.objects.filter(name=custom_name).exists():
+            return custom_name
+        else:
+            raise ValidationError(_('Custom role with the same name already exists.'))
+
+    def non_role_fields(self):
+        return [self.fields[k] for k in self.fields.keys() if k not in ["preset_roles", "custom_roles"]]
+
+
+class CommentForm(forms.ModelForm):
+
+    class Meta:
+        model = Comment
+        fields = ['name', 'email', 'message', 'parent']