mirror of
https://github.com/Doctorado-ML/beflask.git
synced 2025-08-17 08:05:52 +00:00
admin blueprint
This commit is contained in:
53
app/forms.py
53
app/forms.py
@@ -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")
|
||||
|
Reference in New Issue
Block a user