WMGZON/controllers/web/endpoints.py

from models.users.user import User
from controllers.database.user import UserController

from flask import redirect, Blueprint, session

from . import user
from . import product

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)
        
    return dict(user=user)


# Function responsible for displaying the main landing page of the site
@blueprint.route('/')
def index():
    return redirect("/products")