Update aws.py

kevoreilly · Sep 15, 2024 · 57ba7f1 · 57ba7f1
1 parent ed2065e
commit 57ba7f1
Showing 1 changed file with 3 additions and 2 deletions.
diff --git 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"]