From 57ba7f1e4b3cbd6f8f40a14a7dc181ecb0e21bee Mon Sep 17 00:00:00 2001 From: doomedraven Date: Sun, 15 Sep 2024 09:12:31 +0200 Subject: [PATCH] Update aws.py --- modules/machinery/aws.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/modules/machinery/aws.py b/modules/machinery/aws.py index f6e3a2f021e..853afa336e8 100644 --- a/modules/machinery/aws.py +++ b/modules/machinery/aws.py @@ -7,7 +7,7 @@ try: import boto3 except ImportError: - sys.exit("Missed boto3 dependency: pip3 install boto3") + sys.exit("Missed boto3 dependency: poetry run pip3 install boto3") from lib.cuckoo.common.abstracts import Machinery from lib.cuckoo.common.config import Config @@ -16,6 +16,7 @@ logging.getLogger("boto3").setLevel(logging.CRITICAL) logging.getLogger("botocore").setLevel(logging.CRITICAL) log = logging.getLogger(__name__) +cfg_resultserver_ip = Config().get("resultserver").get("ip") class AWS(Machinery): @@ -111,7 +112,7 @@ def _allocate_new_machine(self): resultserver_ip = ( autoscale_options["resultserver_ip"] if autoscale_options.get("resultserver_ip") - else Config().get("resultserver").get("ip") + else cfg_resultserver_ip ) if autoscale_options.get("resultserver_port"): resultserver_port = autoscale_options["resultserver_port"]