From a975ae4688be61ccb403837908168cd125b31e64 Mon Sep 17 00:00:00 2001 From: Shreya Malviya Date: Fri, 10 Mar 2023 17:27:11 +0530 Subject: [PATCH] Island: Register AgentOTP resource with other authentication resources --- monkey/monkey_island/cc/app.py | 2 -- .../flask_resources/register_resources.py | 2 ++ 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/monkey/monkey_island/cc/app.py b/monkey/monkey_island/cc/app.py index 29de2eb25c2..3054e471289 100644 --- a/monkey/monkey_island/cc/app.py +++ b/monkey/monkey_island/cc/app.py @@ -26,7 +26,6 @@ ResetAgentConfiguration, TerminateAllAgents, ) -from monkey_island.cc.services.authentication_service.flask_resources import AgentOTP from monkey_island.cc.resources.exploitations.monkey_exploitation import MonkeyExploitation from monkey_island.cc.resources.island_mode import IslandMode from monkey_island.cc.resources.local_run import LocalRun @@ -91,7 +90,6 @@ def init_api_resources(api: FlaskDIWrapper): def init_restful_endpoints(api: FlaskDIWrapper): api.add_resource(Root) - api.add_resource(AgentOTP) api.add_resource(Agents) api.add_resource(LocalRun) diff --git a/monkey/monkey_island/cc/services/authentication_service/flask_resources/register_resources.py b/monkey/monkey_island/cc/services/authentication_service/flask_resources/register_resources.py index f6dba0e3b64..1b3cdf10ef7 100644 --- a/monkey/monkey_island/cc/services/authentication_service/flask_resources/register_resources.py +++ b/monkey/monkey_island/cc/services/authentication_service/flask_resources/register_resources.py @@ -3,6 +3,7 @@ from common import DIContainer from ..authentication_facade import AuthenticationFacade +from .agent_otp import AgentOTP from .login import Login from .logout import Logout from .register import Register @@ -18,3 +19,4 @@ def register_resources(api: flask_restful.Api, container: DIContainer): ) api.add_resource(Login, *Login.urls, resource_class_args=(authentication_facade,)) api.add_resource(Logout, *Logout.urls, resource_class_args=(authentication_facade,)) + api.add_resource(AgentOTP, *AgentOTP.urls)