open-law/app/forms/user.py

71 lines
2.1 KiB
Python
Raw Normal View History

2023-04-20 13:10:16 +00:00
from flask_wtf import FlaskForm
from wtforms import (
StringField,
2023-05-04 11:18:49 +00:00
FileField,
2023-04-20 13:10:16 +00:00
PasswordField,
SubmitField,
ValidationError,
BooleanField,
)
2023-04-21 12:58:47 +00:00
from wtforms.validators import DataRequired, Length, EqualTo
2023-05-04 11:18:49 +00:00
from flask_login import current_user
2023-04-20 13:10:16 +00:00
from app import models as m
class UserForm(FlaskForm):
next_url = StringField("next_url")
user_id = StringField("user_id", [DataRequired()])
activated = BooleanField("activated")
username = StringField("Username", [DataRequired()])
password = PasswordField("Password", validators=[DataRequired(), Length(6, 30)])
password_confirmation = PasswordField(
"Confirm Password",
validators=[
DataRequired(),
EqualTo("password", message="Password do not match."),
],
)
submit = SubmitField("Save")
def validate_username(self, field):
if (
m.User.query.filter_by(username=field.data)
.filter(m.User.id != int(self.user_id.data))
.first()
is not None
):
raise ValidationError("This username is taken.")
class NewUserForm(FlaskForm):
activated = BooleanField("activated")
username = StringField("Username", [DataRequired()])
password = PasswordField("Password", validators=[DataRequired(), Length(6, 30)])
password_confirmation = PasswordField(
"Confirm Password",
validators=[
DataRequired(),
EqualTo("password", message="Password do not match."),
],
)
submit = SubmitField("Save")
def validate_username(self, field):
if m.User.query.filter_by(username=field.data).first() is not None:
raise ValidationError("This username is taken.")
2023-05-04 11:18:49 +00:00
class EditUserForm(FlaskForm):
name = StringField("Name", [DataRequired()])
avatar_img = FileField("Avatar file (max 200x200px)")
submit = SubmitField("Save")
def validate_username(self, field):
if (
m.User.query.filter_by(username=field.data)
.filter(m.User.id != current_user.id)
.first()
):
raise ValidationError("This username is taken.")