From e92665a4278630a6c72a118db2671755fd455033 Mon Sep 17 00:00:00 2001 From: Daniel Goldberg Date: Wed, 27 Nov 2019 13:39:47 +0200 Subject: [PATCH] Random PEP8 cleanups --- .../blackbox/island_client/monkey_island_requests.py | 3 ++- monkey/infection_monkey/network/tools.py | 2 +- .../infection_monkey/system_info/windows_info_collector.py | 6 +++++- 3 files changed, 8 insertions(+), 3 deletions(-) diff --git a/envs/monkey_zoo/blackbox/island_client/monkey_island_requests.py b/envs/monkey_zoo/blackbox/island_client/monkey_island_requests.py index 8ac53996b..8ff3d3667 100644 --- a/envs/monkey_zoo/blackbox/island_client/monkey_island_requests.py +++ b/envs/monkey_zoo/blackbox/island_client/monkey_island_requests.py @@ -27,10 +27,11 @@ class MonkeyIslandRequests(object): @classmethod def refresh_jwt_token(cls, request_function): @functools.wraps(request_function) - def request_function_wrapper(self, *args,**kwargs): + def request_function_wrapper(self, *args, **kwargs): self.token = self.try_get_jwt_from_server() # noinspection PyArgumentList return request_function(self, *args, **kwargs) + return request_function_wrapper def get_jwt_from_server(self): diff --git a/monkey/infection_monkey/network/tools.py b/monkey/infection_monkey/network/tools.py index 7a029e5dd..5e95e20be 100644 --- a/monkey/infection_monkey/network/tools.py +++ b/monkey/infection_monkey/network/tools.py @@ -308,4 +308,4 @@ def get_interface_to_target(dst): return None paths.sort() ret = paths[-1][1] - return ret[1] \ No newline at end of file + return ret[1] diff --git a/monkey/infection_monkey/system_info/windows_info_collector.py b/monkey/infection_monkey/system_info/windows_info_collector.py index 1419478db..857b42303 100644 --- a/monkey/infection_monkey/system_info/windows_info_collector.py +++ b/monkey/infection_monkey/system_info/windows_info_collector.py @@ -3,11 +3,15 @@ import logging import sys sys.coinit_flags = 0 # needed for proper destruction of the wmi python module - +# noinspection PyPep8 import infection_monkey.config +# noinspection PyPep8 from infection_monkey.system_info.mimikatz_collector import MimikatzCollector +# noinspection PyPep8 from infection_monkey.system_info import InfoCollector +# noinspection PyPep8 from infection_monkey.system_info.wmi_consts import WMI_CLASSES +# noinspection PyPep8 from common.utils.wmi_utils import WMIUtils LOG = logging.getLogger(__name__)