From 6051cfa29aa4a4e63d5d6bb20eb95cdee8eb872e Mon Sep 17 00:00:00 2001 From: Kekoa Kaaikala Date: Thu, 8 Sep 2022 15:55:50 +0000 Subject: [PATCH] Agent: Fix log call exhausting iterator --- monkey/infection_monkey/monkey.py | 1 + monkey/infection_monkey/network/relay/utils.py | 2 -- 2 files changed, 1 insertion(+), 2 deletions(-) diff --git a/monkey/infection_monkey/monkey.py b/monkey/infection_monkey/monkey.py index b242ad7d2..eea2d6af7 100644 --- a/monkey/infection_monkey/monkey.py +++ b/monkey/infection_monkey/monkey.py @@ -126,6 +126,7 @@ class InfectionMonkey: return opts def _get_server(self): + logger.debug(f"Trying to wake up with servers: {', '.join(self._opts.servers)}") servers_iterator = (s for s in self._opts.servers) server = find_server(servers_iterator) if server: diff --git a/monkey/infection_monkey/network/relay/utils.py b/monkey/infection_monkey/network/relay/utils.py index 7ab38581b..4a5532c30 100644 --- a/monkey/infection_monkey/network/relay/utils.py +++ b/monkey/infection_monkey/network/relay/utils.py @@ -14,8 +14,6 @@ logger = logging.getLogger(__name__) def find_server(servers: Iterable[str]) -> Optional[str]: - logger.debug(f"Trying to wake up with servers: {', '.join(servers)}") - for server in servers: logger.debug(f"Trying to connect to server: {server}")