typing improvement for ComInterface

src/tmtccmd/com/__init__.py

@@ -61,7 +61,7 @@ def close(self, args: Any = 0):
         """
 
     @abstractmethod
-    def send(self, data: bytes):
+    def send(self, data: bytes | bytearray):
         """Send raw data.
 
         :raises SendError: Sending failed for some reason.

src/tmtccmd/com/dummy.py

@@ -136,6 +136,6 @@ def data_available(self, timeout: float = 0, parameters: any = 0):
     def receive(self, parameters: any = 0) -> List[bytes]:
         return self.dummy_handler.receive_reply_package()
 
-    def send(self, data: bytes):
+    def send(self, data: bytes | bytearray):
         if data is not None:
             self.dummy_handler.insert_telecommand(data)

src/tmtccmd/com/serial_cobs.py

@@ -49,7 +49,7 @@ def close(self, args: any = None) -> None:
         self.__reception_thread.join(0.4)
         super().close_port()
 
-    def send(self, data: bytes):
+    def send(self, data: bytes | bytearray):
         encoded = bytearray([0])
         encoded.extend(cobs.encode(data))
         encoded.append(0)

src/tmtccmd/com/serial_dle.py

@@ -87,7 +87,7 @@ def close(self, args: any = None) -> None:
         self.__reception_thread.join(0.4)
         super().close_port()
 
-    def send(self, data: bytes):
+    def send(self, data: bytes | bytearray):
         encoded_data = self.__encoder.encode(source_packet=data, add_stx_etx=True)
         self.serial.write(encoded_data)
 

src/tmtccmd/com/tcp.py

@@ -123,7 +123,7 @@ def close(self, args: any = None) -> None:
                 self.__connected = False
         self.__tcp_socket = None
 
-    def send(self, data: bytes):
+    def send(self, data: bytes | bytearray):
         self.__tc_queue.put(data)
 
     def receive(self, poll_timeout: float = 0) -> List[bytes]:

src/tmtccmd/com/udp.py

@@ -64,7 +64,7 @@ def close(self, args: any = None) -> None:
         if self.udp_socket is not None:
             self.udp_socket.close()
 
-    def send(self, data: bytes):
+    def send(self, data: bytes | bytearray):
         if self.udp_socket is None:
             return
         bytes_sent = self.udp_socket.sendto(data, self.send_address.to_tuple)