From 4ee1450001cf6d79b807ce350c0de511108f2b5d Mon Sep 17 00:00:00 2001 From: Shreya Malviya Date: Thu, 4 Aug 2022 13:03:04 +0530 Subject: [PATCH] Island: Inject local_ip_addresses into Root resource's constructor --- monkey/monkey_island/cc/resources/root.py | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) diff --git a/monkey/monkey_island/cc/resources/root.py b/monkey/monkey_island/cc/resources/root.py index 0bfc94a8c..dcf07c9aa 100644 --- a/monkey/monkey_island/cc/resources/root.py +++ b/monkey/monkey_island/cc/resources/root.py @@ -1,4 +1,5 @@ import logging +from typing import Sequence from flask import jsonify, make_response, request @@ -6,7 +7,6 @@ from monkey_island.cc.database import mongo from monkey_island.cc.resources.AbstractResource import AbstractResource from monkey_island.cc.resources.request_authentication import jwt_required from monkey_island.cc.services.infection_lifecycle import get_completed_steps -from monkey_island.cc.services.utils.network_utils import get_local_ip_addresses logger = logging.getLogger(__name__) @@ -15,6 +15,9 @@ class Root(AbstractResource): urls = ["/api"] + def __init__(self, local_ip_addresses: Sequence[str]): + self._local_ips = local_ip_addresses + def get(self, action=None): if not action: action = request.args.get("action") @@ -29,7 +32,7 @@ class Root(AbstractResource): @jwt_required def get_server_info(self): return jsonify( - ip_addresses=get_local_ip_addresses(), + ip_addresses=self._local_ips, mongo=str(mongo.db), completed_steps=get_completed_steps(), )