mirror of
https://gitlab.ub.uni-bielefeld.de/sfb1288inf/nopaque.git
synced 2024-11-15 17:25:44 +00:00
87 lines
3.5 KiB
Python
87 lines
3.5 KiB
Python
from flask import current_app
|
|
from flask_login import current_user
|
|
from flask_wtf import FlaskForm
|
|
from wtforms import (BooleanField, PasswordField, SelectField, StringField,
|
|
SubmitField, ValidationError)
|
|
from wtforms.validators import DataRequired, Email, EqualTo, Length, Regexp
|
|
|
|
|
|
class ChangePasswordForm(FlaskForm):
|
|
password = PasswordField('Old password', validators=[DataRequired()])
|
|
new_password = PasswordField(
|
|
'New password',
|
|
validators=[DataRequired(), EqualTo('password_confirmation',
|
|
message='Passwords must match.')]
|
|
)
|
|
new_password2 = PasswordField(
|
|
'Confirm new password', validators=[DataRequired()])
|
|
submit = SubmitField('Change password')
|
|
|
|
def __init__(self, user=current_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 EditGeneralSettingsForm(FlaskForm):
|
|
dark_mode = BooleanField('Dark mode')
|
|
email = StringField('E-Mail',
|
|
validators=[DataRequired(), Length(1, 254), Email()])
|
|
username = StringField(
|
|
'Benutzername',
|
|
validators=[DataRequired(),
|
|
Length(1, 64),
|
|
Regexp(current_app.config['ALLOWED_USERNAME_REGEX'],
|
|
message='Usernames must have only letters, numbers,'
|
|
' dots or underscores')]
|
|
)
|
|
submit = SubmitField('Submit')
|
|
|
|
def __init__(self, user=current_user, *args, **kwargs):
|
|
super().__init__(*args, **kwargs)
|
|
self.user = user
|
|
self.email.data = self.email.data or user.email
|
|
self.dark_mode.data = self.dark_mode.data or user.setting_dark_mode
|
|
self.username.data = self.username.data or user.username
|
|
|
|
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.')
|
|
|
|
|
|
class EditNotificationSettingsForm(FlaskForm):
|
|
job_status_mail_notifications = SelectField(
|
|
'Job status mail notifications',
|
|
choices=[('', 'Choose your option'),
|
|
('all', 'Notify on all status changes'),
|
|
('end', 'Notify only when a job ended'),
|
|
('none', 'No status update notifications')],
|
|
validators=[DataRequired()])
|
|
job_status_site_notifications = SelectField(
|
|
'Job status site notifications',
|
|
choices=[('', 'Choose your option'),
|
|
('all', 'Notify on all status changes'),
|
|
('end', 'Notify only when a job ended'),
|
|
('none', 'No status update notifications')],
|
|
validators=[DataRequired()])
|
|
submit = SubmitField('Save settings')
|
|
|
|
def __init__(self, user=current_user, *args, **kwargs):
|
|
super().__init__(*args, **kwargs)
|
|
self.user = user
|
|
self.job_status_mail_notifications.data = \
|
|
self.job_status_mail_notifications.data \
|
|
or user.setting_job_status_mail_notifications
|
|
self.job_status_site_notifications.data = \
|
|
self.job_status_site_notifications.data \
|
|
or user.setting_job_status_site_notifications
|