From 0ba2434ed1a1dca443506e3a4204807502bf41f6 Mon Sep 17 00:00:00 2001 From: stilbruch Date: Mon, 21 Mar 2022 19:09:56 -0500 Subject: More database work --- app/login.py | 18 ++++++++++++++++++ 1 file changed, 18 insertions(+) (limited to 'app/login.py') diff --git a/app/login.py b/app/login.py index 7e441b2..6d4b4a2 100644 --- a/app/login.py +++ b/app/login.py @@ -1,7 +1,25 @@ from app import app from database import database_get from flask_login import LoginManager, UserMixin, current_user +from flask_wtf import FlaskForm +from wtforms import StringField, PasswordField, BooleanField +from wtforms.validators import DataRequired, Email +# https://python.plainenglish.io/implementing-flask-login-with-hash-password-888731c88a99 + +# Forms classes for flask_wtf +class LoginForm(FlaskForm): + username = StringField("username", validators=[DataRequired()]) + password = PasswordField("password", validators=[DataRequired()]) + remember_me = BooleanField() + +class RegisterForm(FlaskForm): + username = StringField("username", validators=[DataRequired()]) + password = PasswordField("password", validators=[DataRequired()]) + name = StringField("name") + email = StringField("email", validators=[DataRequired(), Email()]) + +# User class for flask_login class User(UserMixin): def __init__(self, id, email, password_hash): self.id = unicode(id) -- cgit v1.2.3