Agent: Fix mypy issues in propagation credentials

Updated aggregating_propagation_credentials_repository.py
This commit is contained in:
Kekoa Kaaikala 2022-09-21 17:42:52 +00:00
parent 60c1eb9cc1
commit be4c5e7312
1 changed files with 3 additions and 3 deletions

View File

@ -1,5 +1,5 @@
import logging 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 import Credentials, LMHash, NTHash, Password, SSHKeypair, Username
from common.credentials.credentials import Identity, Secret from common.credentials.credentials import Identity, Secret
@ -21,7 +21,7 @@ class AggregatingPropagationCredentialsRepository(IPropagationCredentialsReposit
""" """
def __init__(self, control_channel: IControlChannel): def __init__(self, control_channel: IControlChannel):
self._stored_credentials = { self._stored_credentials: Dict = {
"exploit_user_list": set(), "exploit_user_list": set(),
"exploit_password_list": set(), "exploit_password_list": set(),
"exploit_lm_hash_list": set(), "exploit_lm_hash_list": set(),
@ -72,7 +72,7 @@ class AggregatingPropagationCredentialsRepository(IPropagationCredentialsReposit
return self._stored_credentials 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: if not credentials_values:
return return