PowerStress client approximately works

Author: geeksville

meshtastic/__main__.py

@@ -852,14 +852,14 @@ def setSimpleConfig(modem_preset):
             qr = pyqrcode.create(url)
             print(qr.terminal())
 
-        if args.slog_out:
+        if args.slog_out or args.power_stress:
             # Setup loggers
             global meter
             LogSet(interface, args.slog_out if args.slog_out != 'default' else None, meter)
 
-        if args.power_stress:
-            stress = PowerStress(interface)
-            stress.run()
+            if args.power_stress:
+                stress = PowerStress(interface)
+                stress.run()
 
         if args.listen:
             closeNow = False

meshtastic/powermon/stress.py

@@ -47,7 +47,7 @@ def sendPowerStress(
             self.node_id,
             portnums_pb2.POWERSTRESS_APP,
             wantAck=True,
-            wantResponse=False,
+            wantResponse=True,
             onResponse=onResponse,
             onResponseAckPermitted=True
         )
@@ -64,7 +64,7 @@ def run(self):
         # Send the power stress command
         gotAck = False
 
-        def onResponse(packet, interface):
+        def onResponse(packet: dict):  # pylint: disable=unused-argument
             nonlocal gotAck
             gotAck = True
 

meshtastic/slog/slog.py

@@ -1,11 +1,11 @@
 """code logging power consumption of meshtastic devices."""
 
 import atexit
+import io
 import logging
 import os
 import re
 import threading
-import io
 import time
 from dataclasses import dataclass
 from datetime import datetime
@@ -100,10 +100,17 @@ def __init__(self, client: MeshInterface, dir_path: str) -> None:
         """
         self.client = client
         self.writer = FeatherWriter(f"{dir_path}/slog")
-        self.raw_file: Optional[io.TextIOWrapper] = open(  # pylint: disable=consider-using-with
+        self.raw_file: Optional[
+            io.TextIOWrapper
+        ] = open(  # pylint: disable=consider-using-with
             f"{dir_path}/raw.txt", "w", encoding="utf8"
         )
-        self.listener = pub.subscribe(self._onLogMessage, TOPIC_MESHTASTIC_LOG_LINE)
+
+        # We need a closure here because the subscription API is very strict about exact arg matching
+        def listen_glue(line, interface):  # pylint: disable=unused-argument
+            self._onLogMessage(line)
+
+        self.listener = pub.subscribe(listen_glue, TOPIC_MESHTASTIC_LOG_LINE)
 
     def close(self) -> None:
         """Stop logging."""
@@ -113,9 +120,7 @@ def close(self) -> None:
         self.raw_file = None  # mark that we are shutting down
         f.close()  # Close the raw.txt file
 
-    def _onLogMessage(
-        self, line: str, interface: MeshInterface  # pylint: disable=unused-argument
-    ) -> None:
+    def _onLogMessage(self, line: str) -> None:
         """Handle log messages.
 
         line (str): the line of log output
@@ -126,17 +131,20 @@ def _onLogMessage(
             args = m.group(2)
             args += " "  # append a space so that if the last arg is an empty str it will still be accepted as a match
             logging.debug(f"SLog {src}, reason: {args}")
-            d = log_defs.get(src)
-            if d:
-                r = d.format.parse(args)  # get the values with the correct types
-                if r:
-                    di = r.named
-                    di["time"] = datetime.now()
-                    self.writer.add_row(di)
-                else:
-                    logging.warning(f"Failed to parse slog {line} with {d.format}")
+            if(src != "PM"):
+                logging.warning(f"Not yet handling structured log {src} (FIXME)")
             else:
-                logging.warning(f"Unknown Structured Log: {line}")
+                d = log_defs.get(src)
+                if d:
+                    r = d.format.parse(args)  # get the values with the correct types
+                    if r:
+                        di = r.named
+                        di["time"] = datetime.now()
+                        self.writer.add_row(di)
+                    else:
+                        logging.warning(f"Failed to parse slog {line} with {d.format}")
+                else:
+                    logging.warning(f"Unknown Structured Log: {line}")
         if self.raw_file:
             self.raw_file.write(line + "\n")  # Write the raw log