ccvpn/lambdainst/forms.py

import base64
from django import forms
from django.contrib.auth.models import User
from django.utils.translation import ugettext_lazy as _
from django.utils.safestring import mark_safe


class FormPureRender:
    def as_pure_aligned(self):
        html = ''
        for f in self:
            html += '<div class="pure-control-group">\n'
            html += str(f.label_tag()) + '\n'
            html += str(f) + '\n'
            if f.errors:
                html += str(f.errors) + '\n'
            html += '</div>\n'
        return mark_safe(html)


class UserField(forms.RegexField):
    def clean(self, value):
        super(UserField, self).clean(value)
        try:
            User.objects.get(username=value)
            raise forms.ValidationError(_("Username taken."))
        except User.DoesNotExist:
            return value


class SignupForm(forms.Form, FormPureRender):
    username = UserField(
        label=_("Username"), min_length=2, max_length=16, regex='^[a-zA-Z0-9_-]+$',
        widget=forms.TextInput(attrs={'required': 'true',
                                      'pattern': '[a-zA-Z0-9_-]{2,32}',
                                      'placeholder': _("Username"),
                                      'autofocus': 'true'})
    )
    password = forms.CharField(
        label=_("Password"),
        widget=forms.PasswordInput(attrs={'placeholder': _("Anything")})
    )
    password2 = forms.CharField(
        label=_("Repeat"),
        widget=forms.PasswordInput(attrs={'placeholder': _("Same Anything")})
    )
    email = forms.EmailField(
        label=_("E-Mail"),
        widget=forms.EmailInput(attrs={'placeholder': _("E-Mail")}),
        required=False,
    )

    def clean_password(self):
        if self.data['password'] != self.data['password2']:
            raise forms.ValidationError(_("Passwords are not the same"))
        return self.data['password']


class ReqEmailForm(forms.Form, FormPureRender):
    email = forms.EmailField(
        label=_("E-Mail"),
        widget=forms.EmailInput(attrs={'placeholder': _("E-Mail")}),
    )


def wg_pk_validator(s):
    try:
        data = base64.b64decode(s, validate=True)
    except:
        raise forms.ValidationError("Invalid public key format")
    if len(data) != 32:
        raise forms.ValidationError("Invalid public key length")


class WgPeerForm(forms.Form):
    public_key = forms.CharField(min_length=3, max_length=100, strip=True, required=False, validators=[
        wg_pk_validator
    ])
    name = forms.CharField(max_length=21, required=False)