admin blueprint

This commit is contained in:
2023-06-08 19:58:07 +02:00
parent f7064811d7
commit bc57f6f747
12 changed files with 288 additions and 253 deletions

View File

@@ -9,11 +9,7 @@ from wtforms import (
from wtforms.validators import (
DataRequired,
Length,
EqualTo,
Email,
ValidationError,
)
from app.models import User
class LoginForm(FlaskForm):
@@ -27,54 +23,5 @@ class LoginForm(FlaskForm):
submit = SubmitField()
class UserForm(FlaskForm):
username = StringField(
"Username", validators=[DataRequired(), Length(1, 20)]
)
email = StringField("Email", validators=[DataRequired(), Email()])
password = PasswordField(
"Password", validators=[DataRequired(), Length(4, 150)]
)
password2 = PasswordField(
"Password",
validators=[DataRequired(), Length(4, 150), EqualTo("password")],
)
admin = BooleanField("Admin")
benchmark_id = SelectField("Benchmark", coerce=int)
submit = SubmitField()
def validate_username(self, username):
user = User.query.filter_by(username=username.data).first()
message = "Already taken. Please use a different one."
if user is not None:
if self.user_id is None:
raise ValidationError(message)
else:
if user.id != int(self.user_id):
raise ValidationError(message)
def validate_email(self, email):
user = User.query.filter_by(email=email.data).first()
message = "Already taken. Please use a different one."
if user is not None:
if self.user_id is None:
raise ValidationError(message)
else:
if user.id != int(self.user_id):
raise ValidationError(message)
class UpdatePasswordForm(FlaskForm):
password = PasswordField(
"Password", validators=[DataRequired(), Length(4, 150)]
)
password2 = PasswordField(
"Password",
validators=[DataRequired(), Length(4, 150), EqualTo("password")],
)
submit = SubmitField()
class BenchmarkSelect(FlaskForm):
submit = SubmitField("Select")