From ed455cb995ebf7ac069c3a915244f9c100de208e Mon Sep 17 00:00:00 2001 From: Casu Al Snek Date: Sat, 11 Jan 2025 17:13:14 +0545 Subject: [PATCH] Add user and session to request globals --- backend/utils/auth.py | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) diff --git a/backend/utils/auth.py b/backend/utils/auth.py index 1829726..a8bc72e 100644 --- a/backend/utils/auth.py +++ b/backend/utils/auth.py @@ -1,5 +1,5 @@ from functools import wraps -from flask import request, jsonify +from flask import request, jsonify, g from sqlalchemy import select, and_ from ..db.model import User, Session, db from ..constants import UserRole @@ -23,9 +23,12 @@ def requires_role(roles=None): ).scalar() if not session: return jsonify({'error': 'Invalid or expired session'}), 401 - user = session.user + + user: User = session.user if not user: return jsonify({'error': 'User not found for the Access token'}), 401 + g.current_session = session + g.current_user = user # If no roles specified, allow access if not roles: return f(*args, **kwargs)