From be4c5e73126365fb47edc874796f23bbed8415ba Mon Sep 17 00:00:00 2001 From: Kekoa Kaaikala Date: Wed, 21 Sep 2022 17:42:52 +0000 Subject: [PATCH] Agent: Fix mypy issues in propagation credentials Updated aggregating_propagation_credentials_repository.py --- .../aggregating_propagation_credentials_repository.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/monkey/infection_monkey/credential_repository/aggregating_propagation_credentials_repository.py b/monkey/infection_monkey/credential_repository/aggregating_propagation_credentials_repository.py index f7c86c30d..46cf1b9fe 100644 --- a/monkey/infection_monkey/credential_repository/aggregating_propagation_credentials_repository.py +++ b/monkey/infection_monkey/credential_repository/aggregating_propagation_credentials_repository.py @@ -1,5 +1,5 @@ import logging -from typing import Any, Iterable +from typing import Any, Dict, Iterable, Sequence from common.credentials import Credentials, LMHash, NTHash, Password, SSHKeypair, Username from common.credentials.credentials import Identity, Secret @@ -21,7 +21,7 @@ class AggregatingPropagationCredentialsRepository(IPropagationCredentialsReposit """ def __init__(self, control_channel: IControlChannel): - self._stored_credentials = { + self._stored_credentials: Dict = { "exploit_user_list": set(), "exploit_password_list": set(), "exploit_lm_hash_list": set(), @@ -72,7 +72,7 @@ class AggregatingPropagationCredentialsRepository(IPropagationCredentialsReposit return self._stored_credentials - def _set_attribute(self, attribute_to_be_set: str, credentials_values: Iterable[Any]): + def _set_attribute(self, attribute_to_be_set: str, credentials_values: Sequence[Any]): if not credentials_values: return