mirror of https://github.com/logos-co/open-law.git
75 lines
2.2 KiB
Python
75 lines
2.2 KiB
Python
from flask_wtf import FlaskForm
|
|
from wtforms import (
|
|
StringField,
|
|
FileField,
|
|
PasswordField,
|
|
SubmitField,
|
|
ValidationError,
|
|
BooleanField,
|
|
)
|
|
from wtforms.validators import DataRequired, Length, EqualTo
|
|
from flask_login import current_user
|
|
|
|
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.")
|
|
|
|
|
|
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.")
|
|
|
|
|
|
class ReactivateUserForm(FlaskForm):
|
|
submit = SubmitField("Save")
|