tools: check gateway comms before starting

For tools that communicate through the gateway scripts, ensure the
script is running before the tool attempts to start.

Signed-off-by: Jordan Yates <jordan@embeint.com>

Author: JordanYates

src/infuse_iot/tools/audio_record.py

@@ -5,6 +5,7 @@
 __author__ = "Jordan Yates"
 __copyright__ = "Copyright 2024, Embeint Holdings Pty Ltd"
 
+import sys
 import time
 import wave
 from contextlib import ExitStack
@@ -107,6 +108,9 @@ def handle_connection(self):
                         self.handle_channel("right", stack, tdf)
 
     def run(self):
+        if not self._client.comms_check():
+            sys.exit("No communications gateway detected (infuse gateway/bt_native)")
+
         try:
             types = GatewayRequestConnectionRequest.DataType.DATA
             Console.log_info(f"Connecting to 0x{self._id:016x}")

src/infuse_iot/tools/bt_log.py

@@ -5,6 +5,7 @@
 __author__ = "Jordan Yates"
 __copyright__ = "Copyright 2024, Embeint Holdings Pty Ltd"
 
+import sys
 
 from infuse_iot.commands import InfuseCommand
 from infuse_iot.common import InfuseType
@@ -41,6 +42,9 @@ def add_parser(cls, parser):
         )
 
     def run(self):
+        if not self._client.comms_check():
+            sys.exit("No communications gateway detected (infuse gateway/bt_native)")
+
         try:
             types = GatewayRequestConnectionRequest.DataType.LOGGING
             if self._data:

src/infuse_iot/tools/data_logger_sync.py

@@ -9,6 +9,7 @@
 import glob
 import os
 import pathlib
+import sys
 
 from rich.live import Live
 from rich.progress import (
@@ -176,6 +177,9 @@ def handle_sync(self, live: Live, device_id: int, state: DeviceState):
             self.state_update(live, "Scanning")
 
     def run(self):
+        if not self._client.comms_check():
+            sys.exit("No communications gateway detected (infuse gateway/bt_native)")
+
         with Live(self.progress_table(), refresh_per_second=4) as live:
             for source, announce in self._client.observe_announce():
                 self.state_update(live, "Scanning")

src/infuse_iot/tools/localhost.py

@@ -7,6 +7,7 @@
 
 import asyncio
 import pathlib
+import sys
 import threading
 import time
 from typing import Any
@@ -240,6 +241,9 @@ def recv_thread(self) -> None:
         self._data_lock.release()
 
     def run(self):
+        if not self._client.comms_check():
+            sys.exit("No communications gateway detected (infuse gateway/bt_native)")
+
         Console.init()
         app = web.Application()
         # Route for serving the HTML file

src/infuse_iot/tools/ota_upgrade.py

@@ -199,6 +199,9 @@ def run_file_copy(self, live: Live, mtu: int, source: HopReceived):
             self._pending[source.infuse_id] = time.time() + 60
 
     def run(self):
+        if not self._client.comms_check():
+            sys.exit("No communications gateway detected (infuse gateway/bt_native)")
+
         if self._single_diff:
             self.gateway_diff_load()
 

src/infuse_iot/tools/rpc.py

@@ -9,6 +9,7 @@
 import importlib
 import pkgutil
 import random
+import sys
 
 import infuse_iot.rpc_wrappers as wrappers
 from infuse_iot.commands import InfuseCommand, InfuseRpcCommand
@@ -74,6 +75,9 @@ def rx_handler(self, pkt: ClientNotification):
             print(pkt.epacket.payload.decode("utf-8"), end="")
 
     def run(self):
+        if not self._client.comms_check():
+            sys.exit("No communications gateway detected (infuse gateway/bt_native)")
+
         try:
             types = GatewayRequestConnectionRequest.DataType.COMMAND
             if self._args.conn_log:

src/infuse_iot/tools/serial_throughput.py

@@ -6,6 +6,7 @@
 __copyright__ = "Copyright 2024, Embeint Holdings Pty Ltd"
 
 import random
+import sys
 import time
 
 from infuse_iot.commands import InfuseCommand
@@ -81,6 +82,9 @@ def run_send_test(self, num, size, queue_size):
         print(msg)
 
     def run(self):
+        if not self._client.comms_check():
+            sys.exit("No communications gateway detected (infuse gateway/bt_native)")
+
         # No queuing
         print(f"Averaged across {self._iterations} packets with no queuing:")
         self.run_send_test(self._iterations, 4, 1)

src/infuse_iot/tools/tdf_csv.py

@@ -6,6 +6,7 @@
 __copyright__ = "Copyright 2024, Embeint Holdings Pty Ltd"
 
 import os
+import sys
 import time
 
 from infuse_iot.commands import InfuseCommand
@@ -38,6 +39,9 @@ def __init__(self, args):
         self.args = args
 
     def run(self):
+        if not self._client.comms_check():
+            sys.exit("No communications gateway detected (infuse gateway/bt_native)")
+
         files = {}
 
         while True:

src/infuse_iot/tools/tdf_list.py

@@ -5,6 +5,7 @@
 __author__ = "Jordan Yates"
 __copyright__ = "Copyright 2024, Embeint Holdings Pty Ltd"
 
+import sys
 import time
 
 import tabulate
@@ -97,6 +98,9 @@ def append_readings(self, table: list[tuple[str | None, str | None, str, str, st
             self.append_tdf(table, tdf_name, time_str, t)
 
     def run(self) -> None:
+        if not self._client.comms_check():
+            sys.exit("No communications gateway detected (infuse gateway/bt_native)")
+
         while True:
             msg = self._client.receive()
             if msg is None: