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")