From 6e6c3f6133e8d6a9b190a0beb7e770d4b18f7064 Mon Sep 17 00:00:00 2001 From: Mike Salvatore Date: Tue, 23 Nov 2021 06:32:31 -0500 Subject: [PATCH] Agent: Fix capitalization of Dict type hints in IPuppet --- monkey/infection_monkey/i_puppet.py | 4 ++-- monkey/infection_monkey/puppet/mock_puppet.py | 6 +++--- 2 files changed, 5 insertions(+), 5 deletions(-) diff --git a/monkey/infection_monkey/i_puppet.py b/monkey/infection_monkey/i_puppet.py index 46181f509..d44f827eb 100644 --- a/monkey/infection_monkey/i_puppet.py +++ b/monkey/infection_monkey/i_puppet.py @@ -62,7 +62,7 @@ class IPuppet(metaclass=abc.ABCMeta): """ @abc.abstractmethod - def exploit_host(self, name: str, host: str, options: dict, interrupt: threading.Event) -> bool: + def exploit_host(self, name: str, host: str, options: Dict, interrupt: threading.Event) -> bool: """ Runs an exploiter against a remote host :param str name: The name of the exploiter to run @@ -74,7 +74,7 @@ class IPuppet(metaclass=abc.ABCMeta): """ @abc.abstractmethod - def run_payload(self, name: str, options: dict, interrupt: threading.Event) -> None: + def run_payload(self, name: str, options: Dict, interrupt: threading.Event) -> None: """ Runs a payload :param str name: The name of the payload to run diff --git a/monkey/infection_monkey/puppet/mock_puppet.py b/monkey/infection_monkey/puppet/mock_puppet.py index b35d168b7..910c3a636 100644 --- a/monkey/infection_monkey/puppet/mock_puppet.py +++ b/monkey/infection_monkey/puppet/mock_puppet.py @@ -141,7 +141,7 @@ class MockPuppet(IPuppet): return {} - def run_pba(self, name: str, options: dict) -> None: + def run_pba(self, name: str, options: Dict) -> None: logger.debug(f"run_pba({name}, {options})") return None @@ -206,13 +206,13 @@ class MockPuppet(IPuppet): return {} - def exploit_host(self, name: str, host: str, options: dict, interrupt: threading.Event) -> bool: + def exploit_host(self, name: str, host: str, options: Dict, interrupt: threading.Event) -> bool: logger.debug(f"exploit_hosts({name}, {host}, {options})") successful_exploiters = {DOT_1: {"PowerShellExploiter"}, DOT_3: {"SSHExploiter"}} return name in successful_exploiters.get(host, {}) - def run_payload(self, name: str, options: dict, interrupt: threading.Event) -> None: + def run_payload(self, name: str, options: Dict, interrupt: threading.Event) -> None: logger.debug(f"run_payload({name}, {options})") return None