Skip to content

Commit f3e7976

Browse files
committed
Agent: Format config log messages so they are readable
1 parent 3fb8c06 commit f3e7976

File tree

2 files changed

+7
-7
lines changed

2 files changed

+7
-7
lines changed

monkey/infection_monkey/control.py

+4-3
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,7 @@
11
import json
22
import logging
33
import platform
4+
from pprint import pformat
45
from socket import gethostname
56
from urllib.parse import urljoin
67

@@ -206,10 +207,10 @@ def load_control_config():
206207

207208
try:
208209
unknown_variables = WormConfiguration.from_kv(reply.json().get("config"))
209-
LOG.info(
210-
"New configuration was loaded from server: %r"
211-
% (WormConfiguration.hide_sensitive_info(WormConfiguration.as_dict()),)
210+
formatted_config = pformat(
211+
WormConfiguration.hide_sensitive_info(WormConfiguration.as_dict())
212212
)
213+
LOG.info(f"New configuration was loaded from server:\n{formatted_config}")
213214
except Exception as exc:
214215
# we don't continue with default conf here because it might be dangerous
215216
LOG.error(

monkey/infection_monkey/main.py

+3-4
Original file line numberDiff line numberDiff line change
@@ -6,6 +6,7 @@
66
import sys
77
import traceback
88
from multiprocessing import freeze_support
9+
from pprint import pformat
910

1011
# dummy import for pyinstaller
1112
# noinspection PyUnresolvedReferences
@@ -76,10 +77,8 @@ def main():
7677
"default" % (config_file,)
7778
)
7879

79-
print(
80-
"Loaded Configuration: %r"
81-
% WormConfiguration.hide_sensitive_info(WormConfiguration.as_dict())
82-
)
80+
formatted_config = pformat(WormConfiguration.hide_sensitive_info(WormConfiguration.as_dict()))
81+
print(f"Loaded Configuration:\n{formatted_config}")
8382

8483
# Make sure we're not in a machine that has the kill file
8584
kill_path = (

0 commit comments

Comments
 (0)