Update load_user function

This commit is contained in:
Peter Stockings
2025-02-02 17:45:54 +11:00
parent b0296be9a0
commit 116de33df3

9
app.py
View File

@@ -1,11 +1,11 @@
from datetime import date
import os
from flask import Flask, abort, render_template, redirect, request, url_for
from flask_login import LoginManager, current_user
from flask_login import LoginManager
import jinja_partials
from jinja2_fragments import render_block
from decorators import validate_person, validate_topset, validate_workout
from routes.auth import auth, Person
from routes.auth import auth, get_person_by_id
from extensions import db
from utils import convert_str_to_date, generate_plot
from flask_htmx import HTMX
@@ -29,10 +29,7 @@ login_manager.login_message_category = 'info'
@login_manager.user_loader
def load_user(person_id):
row = db.execute("""SELECT person_id, name, email, password_hash from Person WHERE person_id=%s""", [person_id], one=True)
if row:
return Person(row['person_id'], row['name'], row['email'], row['password_hash'])
return None
return get_person_by_id(person_id)
app.register_blueprint(auth, url_prefix='/auth')