feat(connector.py): add on_message handler function argument to init
This commit is contained in:
@@ -4,6 +4,7 @@ from __future__ import annotations
|
||||
import logging as lg
|
||||
import socket
|
||||
import time
|
||||
from typing import Callable
|
||||
|
||||
from judas_protocol import Message
|
||||
|
||||
@@ -17,6 +18,7 @@ class Connector:
|
||||
*,
|
||||
connect_timeout: float = 5.0,
|
||||
ack_timeout: float | None = None,
|
||||
on_message: Callable[[Message], None],
|
||||
) -> None:
|
||||
self.logger: lg.Logger = lg.getLogger(
|
||||
f"{__name__}.{self.__class__.__name__}"
|
||||
@@ -35,6 +37,8 @@ class Connector:
|
||||
|
||||
self.mac_address: str = mac_address
|
||||
|
||||
self.on_message: Callable[[Message], None] = on_message
|
||||
|
||||
def _send_ack(self) -> None:
|
||||
self.logger.debug("[>] Sending ACK...")
|
||||
try:
|
||||
|
||||
Reference in New Issue
Block a user