tools: auto_activate: added

scripts/reboot_count_reset.py

@@ -21,7 +21,7 @@
 
 class RebootCountResetter:
     def __init__(self, args: argparse.Namespace):
-        self.app_id = args.app
+        self.app_ids = args.app
         self.count = args.count
         self.client = LocalClient(default_multicast_address(), 1.0)
         self.decoder = TDF()
@@ -49,7 +49,7 @@ def state_update(self, live: Live, state: str):
         live.update(self.progress_table())
 
     def announce_observed(self, live: Live, infuse_id: int, pkt: readings.announce | readings.announce_v2):
-        if pkt.application != self.app_id:
+        if pkt.application not in self.app_ids:
             return
         if pkt.reboots == self.count:
             if (infuse_id not in self.already) and (infuse_id not in self.updated):
@@ -87,7 +87,7 @@ def run(self):
 if __name__ == "__main__":
     parser = argparse.ArgumentParser("Reset reboot counters to a common value")
     addr_group = parser.add_mutually_exclusive_group(required=True)
-    addr_group.add_argument("--app", type=lambda x: int(x, 0), help="Application ID to reset")
+    addr_group.add_argument("--app", type=lambda x: int(x, 0), action="append", help="Application ID to reset")
     parser.add_argument("--count", type=int, default=0, help="Value to reset count to")
 
     args = parser.parse_args()

src/infuse_iot/tools/auto_activate.py

@@ -0,0 +1,103 @@
+#!/usr/bin/env python3
+
+"""Automatically activate/deactivate observed devices"""
+
+__author__ = "Jordan Yates"
+__copyright__ = "Copyright 2026, Embeint Holdings Pty Ltd"
+
+
+from rich.live import Live
+from rich.status import Status
+from rich.table import Table
+
+import infuse_iot.definitions.kv as kv
+import infuse_iot.definitions.rpc as rpc
+from infuse_iot.commands import InfuseCommand
+from infuse_iot.epacket.packet import Auth
+from infuse_iot.rpc_client import RpcClient
+from infuse_iot.socket_comms import (
+    GatewayRequestConnectionRequest,
+    LocalClient,
+    default_multicast_address,
+)
+from infuse_iot.tdf import TDF
+
+
+class SubCommand(InfuseCommand):
+    NAME = "auto_activate"
+    HELP = "Automatically activate/deactivate observed devices"
+    DESCRIPTION = "Automatically activate/deactivate observed devices"
+
+    def __init__(self, args):
+        self.app_ids = args.app
+        self.active = args.active or False
+        self.client = LocalClient(default_multicast_address(), 1.0)
+        self.decoder = TDF()
+        self.state = "Scanning"
+        self.name = "Active" if args.active else "Inactive"
+
+        self.correct: set[int] = set()
+        self.updated: set[int] = set()
+
+    @classmethod
+    def add_parser(cls, parser):
+        addr_group = parser.add_mutually_exclusive_group(required=True)
+        addr_group.add_argument("--app", type=lambda x: int(x, 0), action="append", help="Application ID to control")
+        mode_group = parser.add_mutually_exclusive_group(required=True)
+        mode_group.add_argument("--active", action="store_true", help="Move all devices to active state")
+        mode_group.add_argument("--inactive", action="store_true", help="Move all devices to inactive state")
+
+    def progress_table(self):
+        table = Table()
+        table.add_column()
+        table.add_column("Count")
+        table.add_row(self.name, str(len(self.correct)))
+        table.add_row("Updated", str(len(self.updated)))
+
+        meta = Table(box=None)
+        meta.add_column()
+        meta.add_row(table)
+        meta.add_row(Status(self.state))
+
+        return meta
+
+    def state_update(self, live: Live, state: str):
+        self.state = state
+        live.update(self.progress_table())
+
+    def update_active_state(self, live: Live, infuse_id: int, active: bool):
+        try:
+            self.state_update(live, f"Connecting to {infuse_id:016X}")
+            with self.client.connection(infuse_id, GatewayRequestConnectionRequest.DataType.COMMAND) as mtu:
+                rpc_client = RpcClient(self.client, mtu, infuse_id)
+
+                key_val = bytes(kv.slots.application_active(1 if active else 0))
+                all_vals = (
+                    bytes(rpc.rpc_struct_kv_store_value(kv.slots.application_active.BASE_ID, len(key_val))) + key_val
+                )
+                params = bytes(rpc.kv_write.request(1)) + all_vals
+
+                hdr, _ = rpc_client.run_standard_cmd(
+                    rpc.kv_write.COMMAND_ID, Auth.DEVICE, params, rpc.kv_write.response.from_buffer_copy
+                )
+                if hdr is None:
+                    return
+                if hdr.return_code == 0:
+                    self.updated.add(infuse_id)
+
+        except ConnectionRefusedError:
+            self.state_update(live, "Scanning")
+            return
+        self.state_update(live, "Scanning")
+
+    def run(self):
+        with Live(self.progress_table(), refresh_per_second=4) as live:
+            for source, announce in self.client.observe_announce():
+                if announce.application not in self.app_ids:
+                    continue
+                application_active = not (announce.flags & 0x80)
+                if self.active == application_active:
+                    self.correct.add(source.infuse_id)
+                else:
+                    self.update_active_state(live, source.infuse_id, self.active)
+                live.update(self.progress_table())

src/infuse_iot/zephyr/lora.py

@@ -0,0 +1,41 @@
+#!/usr/bin/env python3
+
+import enum
+
+
+class Bandwidth(enum.IntEnum):
+    """
+    LoRa Bandwidth
+    Higher bandwidths result in faster datarates but lower range
+    """
+
+    BW_125_KHZ = 0
+    BW_250_KHZ = 1
+    BW_500_KHZ = 2
+
+
+class SpreadingFactor(enum.IntEnum):
+    """
+    LoRa Spreading Factor
+    Higher spreading factors result in lower datarates but higher range
+    """
+
+    SF6 = 6
+    SF7 = 7
+    SF8 = 8
+    SF9 = 9
+    SF10 = 10
+    SF11 = 11
+    SF12 = 12
+
+
+class CodingRate(enum.IntEnum):
+    """
+    LoRa Coding Rate
+    Higher coding rates improve error recovery but increase packet overhead
+    """
+
+    CR_4_5 = 1
+    CR_4_6 = 2
+    CR_4_7 = 3
+    CR_4_8 = 4