# -*- coding: utf-8 -*-
import re
from flask import g, current_app, session, Markup, url_for
from flask.ext.wtf import Form
from flask.ext.login import current_user
from wtforms import TextField, PasswordField, ValidationError, BooleanField,\
validators
from wtforms.form import FormMeta
from utils import NotRegexp
_username_re = re.compile(r'^[a-zA-Z][a-zA-Z0-9-]{1,15}$')
_username_exclude_re = re.compile(r'^(admin|root)')
class RegisterForm(Form):
username = TextField('Benutzername', [
validators.Regexp(_username_re, message=u'Benutzername darf nur aus '
u'a-z, Zahlen und - bestehen (2-16 Zeichen, am Anfang nur a-z).'),
NotRegexp(_username_exclude_re, message=u'Dieser Benutzername ist nicht erlaubt.'),
])
mail = TextField('E-Mail-Adresse', [validators.Email(), validators.Length(min=6, max=50)])
question = TextField('Hauptstadt von Deutschland?', [validators.AnyOf(('Berlin', 'berlin'),
message=u'Bitte beantworte die Frage.')])
def validate_username(form, field):
try:
current_app.user_backend.get_by_uid(field.data)
except current_app.user_backend.NoSuchUserError:
if current_app.username_blacklist:
if field.data.lower() in current_app.username_blacklist:
raise ValidationError(Markup(u'Dieser Benutzername ist momentan nicht erlaubt. '
u'Weitere Informationen' % url_for('default.about')))
else:
raise ValidationError(u'Dieser Benutzername ist schon vergeben.')
def validate_mail(form, field):
try:
current_app.user_backend.get_by_mail(field.data)
except current_app.user_backend.NoSuchUserError:
pass
else:
raise ValidationError(Markup(u'Ein Benutzername mit dieser Adresse existiert bereits. '
u'Falls du deinen Benutzernamen vergessen hast, kannst du die '
u'Passwort-vergessen-Funktion benutzen.'
% url_for('default.lost_password')))
class AdminCreateAccountForm(RegisterForm):
def validate_username(form, field):
try:
current_app.user_backend.get_by_uid(field.data)
except current_app.user_backend.NoSuchUserError:
return
else:
raise ValidationError(u'Dieser Benutzername ist schon vergeben')
question = None
class RegisterCompleteForm(Form):
password = PasswordField('Passwort', [validators.Required(),
validators.EqualTo('password_confirm', message=u'Passwörter stimmen nicht überein')])
password_confirm = PasswordField(u'Passwort bestätigen')
# n.b. this form is also used in lost_password_complete
class LostPasswordForm(Form):
username_or_mail = TextField(u'Benutzername oder E-Mail')
def validate_username_or_mail(form, field):
if '@' not in field.data:
try:
form.user = current_app.user_backend.get_by_uid(field.data)
except NoSuchUserError:
raise ValidationError(u'Es gibt keinen Benutzer mit diesem Namen.')
else:
try:
form.user = current_app.user_backend.get_by_mail(field.data)
except NoSuchUserError:
raise ValidationError(u'Es gibt keinen Benutzer mit dieser Adresse.')
class SettingsMeta(FormMeta):
def __call__(cls, *args, **kwargs):
for service in current_app.all_services:
setattr(cls, 'password_%s' % service.id, PasswordField(
u'Passwort für %s' % service.name, [
validators.Optional(),
validators.EqualTo(
'password_confirm_%s' % service.id,
message=u'Passwörter stimmen nicht überein'),
]))
setattr(cls, 'password_confirm_%s' % service.id, PasswordField(
u'Passwort für %s (Bestätigung)' % service.name))
setattr(cls, 'delete_%s' % service.id, BooleanField(
u'Passwort für %s löschen' % service.name))
return super(SettingsMeta, cls).__call__(*args, **kwargs)
class SettingsForm(Form):
__metaclass__ = SettingsMeta
old_password = PasswordField('Altes Passwort')
password = PasswordField('Neues Passwort', [validators.Optional(),
validators.EqualTo('password_confirm', message=u'Passwörter stimmen nicht überein')])
password_confirm = PasswordField(u'Passwort bestätigen')
mail = TextField('E-Mail-Adresse', [validators.Optional(), validators.Email(), validators.Length(min=6, max=50)])
def validate_old_password(form, field):
if form.password.data:
if not field.data:
raise ValidationError(u'Gib bitte dein altes Passwort ein, um ein neues zu setzen.')
if field.data != current_user.password:
raise ValidationError(u'Altes Passwort ist falsch.')
def validate_mail(form, field):
results = current_app.user_backend.find_by_mail(field.data)
for user in results:
if user.uid != current_user.uid:
raise ValidationError(u'Diese E-Mail-Adresse wird schon von einem anderen Account benutzt!')
def get_servicepassword(self, service_id):
return getattr(self, 'password_%s' % service_id)
def get_servicepasswordconfirm(self, service_id):
return getattr(self, 'password_confirm_%s' % service_id)
def get_servicedelete(self, service_id):
return getattr(self, 'delete_%s' % service_id)
class AdminDisableAccountForm(Form):
username = TextField(u'Benutzername')
def validate_username(form, field):
try:
form.user = current_app.user_backend.get_by_uid(field.data)
except current_app.user_backend.NoSuchUserError:
raise ValidationError(u'Dieser Benutzername existiert nicht')