From 1a7135e13f934f4588b32abad01c2ffa7a3cf035 Mon Sep 17 00:00:00 2001 From: Mike Salvatore Date: Mon, 13 Dec 2021 08:37:00 -0500 Subject: [PATCH] Agent: Improve Callback type hint in IPScanner --- monkey/infection_monkey/master/ip_scanner.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/monkey/infection_monkey/master/ip_scanner.py b/monkey/infection_monkey/master/ip_scanner.py index 1c273fa22..b54adfb4a 100644 --- a/monkey/infection_monkey/master/ip_scanner.py +++ b/monkey/infection_monkey/master/ip_scanner.py @@ -11,7 +11,9 @@ from .threading_utils import create_daemon_thread logger = logging.getLogger() -Callback = Callable[[str, PingScanData, Dict[int, PortScanData]], None] +IP = str +Port = int +Callback = Callable[[IP, PingScanData, Dict[Port, PortScanData]], None] class IPScanner: