From c57ebef076438c83a0e408227341398249566ff1 Mon Sep 17 00:00:00 2001 From: stilbruch Date: Wed, 23 Mar 2022 12:30:23 -0500 Subject: User registration and login now works --- app/login.py | 55 ------------------------------------------------------- 1 file changed, 55 deletions(-) delete mode 100644 app/login.py (limited to 'app/login.py') diff --git a/app/login.py b/app/login.py deleted file mode 100644 index 6d4b4a2..0000000 --- a/app/login.py +++ /dev/null @@ -1,55 +0,0 @@ -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) - self.email = email - self.password_hash = password_hash - self.authenticated = False - - def is_authenticated(self): - return self.authenticated - - def is_active(self): - return True - - def is_anonymous(self): - return False - - def get_id(self): - return self.id - -login_manager = LoginManager() -login_manager.init_app(app) - -@login_manager.user_loader -def load_user(uid): - conn = database_get() - curs = conn.cursor() - curs.execute("SELECT * FROM users WHERE id = (?)", [uid]) - row = curs.fetchone() - - if row is None: - return None - else: - return User(int(row[0]), row[1], row[2]) -- cgit v1.2.3