From 104e0abda9aceb13931d3e1d9d4c2d9e6d40e6e5 Mon Sep 17 00:00:00 2001 From: Kekoa Kaaikala Date: Fri, 7 Oct 2022 13:26:49 +0000 Subject: [PATCH] Agent: Fix mypy issues in smb_tools.py --- monkey/infection_monkey/exploit/tools/smb_tools.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) 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)):