Streamline message class. (#2133)

Author: janiversen

pymodbus/framer/ascii_framer.py

@@ -39,7 +39,7 @@ def __init__(self, decoder, client=None):
         self._hsize = 0x02
         self._start = b":"
         self._end = b"\r\n"
-        self.message_handler = MessageAscii([0], True)
+        self.message_handler = MessageAscii()
 
     def decode_data(self, data):
         """Decode data."""

pymodbus/framer/rtu_framer.py

@@ -61,7 +61,7 @@ def __init__(self, decoder, client=None):
         self._end = b"\x0d\x0a"
         self._min_frame_size = 4
         self.function_codes = decoder.lookup.keys() if decoder else {}
-        self.message_handler = MessageRTU([0], True)
+        self.message_handler = MessageRTU()
 
     def decode_data(self, data):
         """Decode data."""

pymodbus/framer/socket_framer.py

@@ -43,7 +43,7 @@ def __init__(self, decoder, client=None):
         """
         super().__init__(decoder, client)
         self._hsize = 0x07
-        self.message_handler = MessageSocket([0], True)
+        self.message_handler = MessageSocket()
 
     def decode_data(self, data):
         """Decode data."""

pymodbus/framer/tls_framer.py

@@ -35,7 +35,7 @@ def __init__(self, decoder, client=None):
         """
         super().__init__(decoder, client)
         self._hsize = 0x0
-        self.message_handler = MessageTLS([0], True)
+        self.message_handler = MessageTLS()
 
     def decode_data(self, data):
         """Decode data."""

pymodbus/message/base.py

@@ -14,23 +14,9 @@ class MessageBase:
 
     EMPTY = b''
 
-    def __init__(
-        self,
-        device_ids: list[int],
-        is_server: bool,
-    ) -> None:
-        """Initialize a message instance.
-
-        :param device_ids: list of device id to accept (server only), None for all.
-        """
-        self.device_ids = device_ids
-        self.is_server = is_server
-        self.broadcast: bool = (0 in device_ids)
-
+    def __init__(self) -> None:
+        """Initialize a message instance."""
 
-    def validate_device_id(self, dev_id: int) -> bool:
-        """Check if device id is expected."""
-        return self.broadcast or (dev_id in self.device_ids)
 
     @abstractmethod
     def decode(self, _data: bytes) -> tuple[int, int, int, bytes]:

pymodbus/message/message.py

@@ -65,15 +65,21 @@ def __init__(self,
         """
         super().__init__(params, is_server)
         self.device_ids = device_ids
-        self.message_type = message_type
+        self.broadcast: bool = (0 in device_ids)
         self.msg_handle: MessageBase = {
-            MessageType.RAW: MessageRaw(device_ids, is_server),
-            MessageType.ASCII: MessageAscii(device_ids, is_server),
-            MessageType.RTU: MessageRTU(device_ids, is_server),
-            MessageType.SOCKET: MessageSocket(device_ids, is_server),
-            MessageType.TLS: MessageTLS(device_ids, is_server),
+            MessageType.RAW: MessageRaw(),
+            MessageType.ASCII: MessageAscii(),
+            MessageType.RTU: MessageRTU(),
+            MessageType.SOCKET: MessageSocket(),
+            MessageType.TLS: MessageTLS(),
         }[message_type]
 
+
+    def validate_device_id(self, dev_id: int) -> bool:
+        """Check if device id is expected."""
+        return self.broadcast or (dev_id in self.device_ids)
+
+
     def callback_data(self, data: bytes, addr: tuple | None = None) -> int:
         """Handle received data."""
         tot_len = len(data)

pymodbus/message/raw.py

@@ -23,9 +23,6 @@ def decode(self, data: bytes) -> tuple[int, int, int, bytes]:
             return 0, 0, 0, self.EMPTY
         dev_id = int(data[0])
         tid = int(data[1])
-        if not self.validate_device_id(dev_id):
-            Log.debug("Device id: {} in frame {} unknown, skipping.", dev_id, data, ":hex")
-
         return len(data), dev_id, tid, data[2:]
 
     def encode(self, data: bytes, device_id: int, tid: int) -> bytes:

pymodbus/message/rtu.py

@@ -177,7 +177,7 @@ def callback(result):
             nonlocal resp
             resp = result
 
-        self._legacy_decode(callback, self.device_ids)
+        self._legacy_decode(callback, [0])
         return 0, 0, 0, b''
 
 

test/message/conftest.py

@@ -21,6 +21,7 @@ def __init__(self,
         """Initialize a message instance."""
         super().__init__(message_type, params, is_server, device_ids)
         self.send = mock.Mock()
+        self.message_type = message_type
 
     def callback_new_connection(self) -> ModbusProtocol:
         """Call when listener receive new connection request."""

test/message/test_ascii.py

@@ -11,7 +11,7 @@ class TestMessageAscii:
     @pytest.fixture(name="frame")
     def prepare_frame():
         """Return message object."""
-        return MessageAscii([1], False)
+        return MessageAscii()
 
 
     @pytest.mark.parametrize(