diff --git a/app/routers/bot.py b/app/routers/bot.py index 26fa242..8c36352 100644 --- a/app/routers/bot.py +++ b/app/routers/bot.py @@ -3,6 +3,8 @@ from quart import Blueprint, jsonify, request, abort, current_app from werkzeug.exceptions import HTTPException from internal.types import ActiveClient, DiscordId from internal.db_wrappers import DiscordIdDbController +from quart_jwt_extended import jwt_required +from router.auth import role_required bot_bp = Blueprint('bot', __name__) diff --git a/app/routers/nodes.py b/app/routers/nodes.py index 4c1f1ea..944993e 100644 --- a/app/routers/nodes.py +++ b/app/routers/nodes.py @@ -1,11 +1,13 @@ import json import asyncio # Import asyncio import websockets +import uuid from quart import Blueprint, jsonify, request, abort, current_app from werkzeug.exceptions import HTTPException from enum import Enum from internal.types import ActiveClient, NodeCommands, UserRoles -import uuid # Import uuid for generating unique request IDs +from quart_jwt_extended import jwt_required +from router.auth import role_required nodes_bp = Blueprint('nodes', __name__) diff --git a/app/routers/systems.py b/app/routers/systems.py index 31c00b3..00660f2 100644 --- a/app/routers/systems.py +++ b/app/routers/systems.py @@ -1,6 +1,8 @@ from quart import Blueprint, jsonify, request, abort, current_app from werkzeug.exceptions import HTTPException from internal.types import System +from quart_jwt_extended import jwt_required +from router.auth import role_required systems_bp = Blueprint('systems', __name__)