src/judas_client/connector.py

@@ -5,16 +5,20 @@ import logging as lg
 import selectors
 import socket
 import time
-from typing import Callable
+from typing import Callable, TYPE_CHECKING
 
 from judas_protocol import Category, ControlAction, Message
 
+if TYPE_CHECKING:
+    from judas_client.client import Client
+
 
 class Connector:
     """Connector class for managing TCP connection and message exchange."""
 
     def __init__(
         self,
+        client: Client,
         mac_address: str,
         host: str,
         port: int,
@@ -34,6 +38,8 @@ class Connector:
         )
         self.logger.debug("Initializing Connector...")
 
+        self.client: Client = client
+
         self.host: str = host
         self.port: int = port