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 += '
\n' html += str(f.label_tag()) + "\n" html += str(f) + "\n" if f.errors: html += str(f.errors) + "\n" html += "
\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)