Overhauled the way in which user details are passed to the frontend
This commit is contained in:
@@ -1,5 +1,7 @@
|
||||
from flask import redirect
|
||||
from flask import Blueprint
|
||||
from models.users.user import User
|
||||
from controllers.database.user import UserController
|
||||
|
||||
from flask import redirect, Blueprint, session
|
||||
|
||||
from . import user
|
||||
from . import product
|
||||
@@ -9,6 +11,24 @@ blueprint = Blueprint('main', __name__)
|
||||
blueprint.register_blueprint(user.blueprint)
|
||||
blueprint.register_blueprint(product.blueprint)
|
||||
|
||||
|
||||
### CONTEXTS ###
|
||||
|
||||
# Function that returns a given user class based on the ID in the session
|
||||
@blueprint.context_processor
|
||||
def get_user() -> dict[User|None]:
|
||||
# Get the user based on the user ID
|
||||
user_id = session.get('user_id')
|
||||
user = None
|
||||
|
||||
if user_id != None:
|
||||
db = UserController()
|
||||
user = db.read_id(user_id)
|
||||
print(user)
|
||||
|
||||
return dict(user=user)
|
||||
|
||||
|
||||
# Function responsible for displaying the main landing page of the site
|
||||
@blueprint.route('/')
|
||||
def index():
|
||||
|
||||
Reference in New Issue
Block a user