that was annoying

eive_tmtc/pus_tm/tm_tcp_server.py

@@ -0,0 +1,90 @@
+import socket
+from typing import Optional
+import json
+import base64
+
+from tmtccmd.logging import get_console_logger
+from tmtccmd.util.obj_id import ObjectIdU32
+from dle_encoder import DleEncoder
+
+LOGGER = get_console_logger()
+
+
+class TmTcpServer:
+    def __init__(self, ip_address: str, port: int):
+
+        self.server_socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
+
+        self.server_socket.bind((ip_address, port))
+
+        # for now, only accept one connection
+        self.server_socket.listen(0)
+
+        self.server_socket.setblocking(False)
+
+        self.client_connection: Optional[socket.socket] = None
+
+        self.dle_encoder = DleEncoder()
+
+    def __del__(self):
+        try:
+            self.close()
+        except:
+            LOGGER.warning("Could not close sockets!")
+
+    def close(self):
+        self.server_socket.close()
+        if self.client_connection is not None:
+            self.client_connection.close()
+
+    def _send_dictionary_over_socket(self, dictionary):
+        # keep listeners current
+        if self.client_connection is None:
+            # no running connection, see if a client wants to connect
+            try:
+                (self.client_connection, _) = self.server_socket.accept()
+                self.client_connection.setblocking(False)
+                print("Client connected")
+            except:
+                # no client waiting
+                return
+
+        data_json_bytes = json.dumps(dictionary).encode()
+
+        # dle encode the bytes
+        # adding a newline because someone might want to look at it in a console
+        data_json_bytes = self.dle_encoder.encode(data_json_bytes + b"\n")
+
+        try:
+            sent_length = self.client_connection.send(data_json_bytes)
+        except:
+            self.client_connection = None
+            return
+        if sent_length == 0:
+            self.client_connection.close()
+            self.client_connection = None
+
+    def report_raw_hk_data(self, object_id: ObjectIdU32, set_id: int, hk_data: bytes):
+
+        data_dict = {
+            "type": "TM",
+            "tmType": "Raw HK",
+            "objectId": object_id.as_hex_string,
+            "setId": set_id,
+            "rawData": base64.b64encode(hk_data).decode(),
+        }
+
+        self._send_dictionary_over_socket(data_dict)
+
+    def report_parsed_hk_data(
+        self, object_id: ObjectIdU32, set_id: int, data_dictionary
+    ):
+        data_dict = {
+            "type": "TM",
+            "tmType": "Parsed HK",
+            "objectId": object_id.as_hex_string,
+            "setId": set_id,
+            "content": data_dictionary,
+        }
+
+        self._send_dictionary_over_socket(data_dict)