Pycloud/app/forms.py

34 lines
1.5 KiB
Python
Raw Normal View History

2020-06-23 19:23:57 +02:00
from flask_login import current_user
2020-06-23 16:11:01 +02:00
from app import words
from flask_wtf import FlaskForm
from flask_wtf.file import FileField, FileRequired
2020-06-23 19:23:57 +02:00
from wtforms import SubmitField, PasswordField, StringField, BooleanField, SelectField
2020-06-23 16:11:01 +02:00
from wtforms.validators import DataRequired, EqualTo, ValidationError
from app.models import User
2020-06-27 12:34:40 +02:00
users_list=[]
for users in User.query.all():
users_list.append(users.username)
2020-06-23 16:11:01 +02:00
class UploadForm(FlaskForm):
2020-06-27 12:34:40 +02:00
uploadfile = FileField(words['termUploadFile'], validators=[FileRequired()])
2020-06-23 16:11:01 +02:00
submit = SubmitField(words['termSubmit'])
2020-06-27 12:34:40 +02:00
filename_choose = StringField(words['termFilename'])
folder = SelectField(words['termUsers'], choices=users_list, validate_choice=False)
2020-06-23 16:11:01 +02:00
class LoginForm(FlaskForm):
username = StringField(words['termUsername'], validators=[DataRequired()])
password = PasswordField(words['termPassword'], validators=[DataRequired()])
remember_me = BooleanField(words['termRememberMe'])
submit = SubmitField(words['termSubmit'])
class RegisterForm(FlaskForm):
username = StringField(words['termUsername'], validators=[DataRequired()])
password = PasswordField(words['termPassword'], validators=[DataRequired()])
repeatpassword = PasswordField(words['termRepeatPassword'], validators=[DataRequired(), EqualTo('password')])
isAdmin = BooleanField(words['termIsAdmin'])
submit = SubmitField(words['termSubmit'])
def validate_username(self, username):
user = User.query.filter_by(username=username.data).first()
if user is not None:
raise ValidationError(words['ErrorUsername'])