diff --git a/backend/app/utils.py b/backend/app/utils.py index d08e8fb0..e74a6615 100644 --- a/backend/app/utils.py +++ b/backend/app/utils.py @@ -1,6 +1,6 @@ from functools import wraps from flask import jsonify -from flask_jwt_extended import get_jwt_identity +from app.auth.quart_jwt import get_jwt_identity from app.models.user import User def admin_required(f): diff --git a/backend/app/websocket_manager.py b/backend/app/websocket_manager.py index 50cb6408..fe0217d1 100644 --- a/backend/app/websocket_manager.py +++ b/backend/app/websocket_manager.py @@ -14,7 +14,7 @@ from datetime import datetime from flask import request, current_app from flask_socketio import emit, join_room, leave_room, disconnect from .extensions import socketio_server as socketio # Import singleton SocketIO instance -from flask_jwt_extended import decode_token +from app.auth.quart_jwt import decode_token from functools import wraps import json from queue import Queue diff --git a/backend/app/websocket_manager_async.py b/backend/app/websocket_manager_async.py index 154999c8..21a1cacb 100644 --- a/backend/app/websocket_manager_async.py +++ b/backend/app/websocket_manager_async.py @@ -11,7 +11,7 @@ import asyncio from typing import Dict, Set, Any, Optional from datetime import datetime from .extensions import socketio_server as sio -from flask_jwt_extended import decode_token +from app.auth.quart_jwt import decode_token # Set up logging logger = logging.getLogger(__name__)