diff options
| author | stilbruch <stilbruch@protonmail.com> | 2022-03-23 12:30:23 -0500 |
|---|---|---|
| committer | stilbruch <stilbruch@protonmail.com> | 2022-03-23 12:30:23 -0500 |
| commit | c57ebef076438c83a0e408227341398249566ff1 (patch) | |
| tree | d4e943e771a127d285baaee14867f48040f23c8d /app/login.py | |
| parent | 43755d0b6d1f65fd65862454be81bd0f5d7ff7dd (diff) | |
| download | Strengthy-c57ebef076438c83a0e408227341398249566ff1.tar.xz Strengthy-c57ebef076438c83a0e408227341398249566ff1.zip | |
User registration and login now works
Diffstat (limited to 'app/login.py')
| -rw-r--r-- | app/login.py | 55 |
1 files changed, 0 insertions, 55 deletions
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]) |
