mirror of
https://gitlab.ub.uni-bielefeld.de/sfb1288inf/nopaque.git
synced 2024-12-26 03:14:19 +00:00
143 lines
3.7 KiB
Python
143 lines
3.7 KiB
Python
from flask_wtf import FlaskForm
|
|
from wtforms import (
|
|
BooleanField,
|
|
FileField,
|
|
PasswordField,
|
|
SelectField,
|
|
StringField,
|
|
SubmitField,
|
|
TextAreaField,
|
|
ValidationError
|
|
)
|
|
from wtforms.validators import (
|
|
DataRequired,
|
|
InputRequired,
|
|
Email,
|
|
EqualTo,
|
|
Length,
|
|
Regexp
|
|
)
|
|
from app.models import User, UserSettingJobStatusMailNotificationLevel
|
|
from app.auth import USERNAME_REGEX
|
|
|
|
|
|
class ChangePasswordForm(FlaskForm):
|
|
password = PasswordField('Old password', validators=[DataRequired()])
|
|
new_password = PasswordField(
|
|
'New password',
|
|
validators=[
|
|
DataRequired(),
|
|
EqualTo('new_password_2', message='Passwords must match')
|
|
]
|
|
)
|
|
new_password_2 = PasswordField(
|
|
'New password confirmation',
|
|
validators=[
|
|
DataRequired(),
|
|
EqualTo('new_password', message='Passwords must match')
|
|
]
|
|
)
|
|
submit = SubmitField()
|
|
|
|
def __init__(self, user, *args, **kwargs):
|
|
super().__init__(*args, **kwargs)
|
|
self.user = user
|
|
|
|
def validate_current_password(self, field):
|
|
if not self.user.verify_password(field.data):
|
|
raise ValidationError('Invalid password')
|
|
|
|
|
|
class EditProfileSettingsForm(FlaskForm):
|
|
user_avatar = FileField(
|
|
'Image File'
|
|
)
|
|
email = StringField(
|
|
'E-Mail',
|
|
validators=[InputRequired(), Length(max=254), Email()]
|
|
)
|
|
username = StringField(
|
|
'Username',
|
|
validators=[
|
|
InputRequired(),
|
|
Length(max=64),
|
|
Regexp(
|
|
USERNAME_REGEX,
|
|
message=(
|
|
'Usernames must have only letters, numbers, dots or '
|
|
'underscores'
|
|
)
|
|
)
|
|
]
|
|
)
|
|
full_name = StringField(
|
|
'Full name',
|
|
validators=[Length(max=128)]
|
|
)
|
|
about_me = TextAreaField(
|
|
'About me',
|
|
validators=[
|
|
Length(max=254)
|
|
]
|
|
)
|
|
website = StringField(
|
|
'Website',
|
|
validators=[
|
|
Length(max=254)
|
|
]
|
|
)
|
|
organization = StringField(
|
|
'Organization',
|
|
validators=[
|
|
Length(max=128)
|
|
]
|
|
)
|
|
location = StringField(
|
|
'Location',
|
|
validators=[
|
|
Length(max=128)
|
|
]
|
|
)
|
|
|
|
submit = SubmitField()
|
|
|
|
def __init__(self, user, *args, **kwargs):
|
|
super().__init__(*args, **kwargs)
|
|
self.user = user
|
|
|
|
def validate_email(self, field):
|
|
if (field.data != self.user.email
|
|
and User.query.filter_by(email=field.data).first()):
|
|
raise ValidationError('Email already registered')
|
|
|
|
def validate_username(self, field):
|
|
if (field.data != self.user.username
|
|
and User.query.filter_by(username=field.data).first()):
|
|
raise ValidationError('Username already in use')
|
|
|
|
def validate_image_file(self, field):
|
|
if not field.data.filename.lower().endswith('.jpg' or '.png' or '.jpeg'):
|
|
raise ValidationError('only .jpg, .png and .jpeg!')
|
|
|
|
|
|
class EditNotificationSettingsForm(FlaskForm):
|
|
job_status_mail_notification_level = SelectField(
|
|
'Job status mail notification level',
|
|
choices=[('', 'Choose your option')],
|
|
validators=[DataRequired()]
|
|
)
|
|
submit = SubmitField()
|
|
|
|
def __init__(self, *args, **kwargs):
|
|
super().__init__(*args, **kwargs)
|
|
self.job_status_mail_notification_level.choices += [
|
|
(x.name, x.name.capitalize())
|
|
for x in UserSettingJobStatusMailNotificationLevel
|
|
]
|
|
|
|
class EditPrivacySettingsForm(FlaskForm):
|
|
public_profile = BooleanField(
|
|
'Public profile'
|
|
)
|
|
submit = SubmitField()
|