Update load_user function
This commit is contained in:
9
app.py
9
app.py
@@ -1,11 +1,11 @@
|
|||||||
from datetime import date
|
from datetime import date
|
||||||
import os
|
import os
|
||||||
from flask import Flask, abort, render_template, redirect, request, url_for
|
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
|
import jinja_partials
|
||||||
from jinja2_fragments import render_block
|
from jinja2_fragments import render_block
|
||||||
from decorators import validate_person, validate_topset, validate_workout
|
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 extensions import db
|
||||||
from utils import convert_str_to_date, generate_plot
|
from utils import convert_str_to_date, generate_plot
|
||||||
from flask_htmx import HTMX
|
from flask_htmx import HTMX
|
||||||
@@ -29,10 +29,7 @@ login_manager.login_message_category = 'info'
|
|||||||
|
|
||||||
@login_manager.user_loader
|
@login_manager.user_loader
|
||||||
def load_user(person_id):
|
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)
|
return get_person_by_id(person_id)
|
||||||
if row:
|
|
||||||
return Person(row['person_id'], row['name'], row['email'], row['password_hash'])
|
|
||||||
return None
|
|
||||||
|
|
||||||
app.register_blueprint(auth, url_prefix='/auth')
|
app.register_blueprint(auth, url_prefix='/auth')
|
||||||
|
|
||||||
|
|||||||
Reference in New Issue
Block a user