wasptool: add --battery to check battery level

Signed-off-by: thefenriswolf <stefan.rohrbacher97@gmail.com>

1 files changed, 8 insertions, 0 deletions

diff --git a/tools/wasptool b/tools/wasptool
index befcd98..ccca180 100755
--- a/tools/wasptool
+++ b/tools/wasptool
@@ -189,6 +189,9 @@ def handle_exec(c, fname):
     print_log(log)
     log.close()
 
+def handle_battery_level(c):
+    print(f'Battery: {c.run_command("watch.battery.level()")}%')
+    
 def handle_memory_free(c):
     # wasp namespace to avoid having to import into the global space
     print(f'Before GC: {c.run_command("wasp.gc.mem_free()"):7}')
@@ -329,6 +332,8 @@ if __name__ == '__main__':
             description='Wasp-os command and control client')
     parser.add_argument('--as', dest='upload_as', default=None,
             help="Filename to use on the target (e.g. wasptool --upload docs/main/chrono.py --as main.py")
+    parser.add_argument('--battery', action='store_true',
+            help="Report remaining battery charge")
     parser.add_argument('--bootloader', action='store_true',
             help="Reboot into the bootloader mode for OTA update")
     parser.add_argument('--binary', action='store_true',
@@ -441,5 +446,8 @@ if __name__ == '__main__':
     if args.bootloader:
         handle_reset(console, ota=True)
         sys.exit(0)
+        
+    if args.battery:
+        handle_battery_level(console)
 
     console.unsync()