diff --git a/monkey/infection_monkey/exploit/tools/smb_tools.py b/monkey/infection_monkey/exploit/tools/smb_tools.py index 5a4ab5042..5593a27d1 100644 --- a/monkey/infection_monkey/exploit/tools/smb_tools.py +++ b/monkey/infection_monkey/exploit/tools/smb_tools.py @@ -3,7 +3,7 @@ import ntpath import pprint from io import BytesIO from pathlib import PurePath -from typing import Optional +from typing import Any, Dict, Optional, Tuple from impacket.dcerpc.v5 import srvs, transport from impacket.smb3structs import SMB2_DIALECT_002, SMB2_DIALECT_21 @@ -79,8 +79,8 @@ class SmbTools(object): resp = resp["InfoStruct"]["ShareInfo"]["Level2"]["Buffer"] - high_priority_shares = () - low_priority_shares = () + high_priority_shares: Tuple[Tuple[str, Dict[str, Any]], ...] = () + low_priority_shares: Tuple[Tuple[str, Dict[str, Any]], ...] = () file_name = dst_path.name for i in range(len(resp)):