update battery commands #864 (#981)

mrlt8 · Sep 2, 2023 · e1ce8db · e1ce8db
1 parent 0cc6e8b
commit e1ce8db
Show file tree

Hide file tree

Showing 2 changed files with 23 additions and 2 deletions.
diff --git a/app/wyzebridge/wyze_commands.py b/app/wyzebridge/wyze_commands.py
@@ -18,7 +18,8 @@
     "motion_tracking": "K11020GetMotionTracking",
     "motion_tagging": "K10290GetMotionTagging",
     "camera_info": "K10020CheckCameraInfo",
-    "battery": "K10448GetBatteryUsage",
+    "battery": "K10050GetPowerLevel",
+    "battery_usage": "K10448GetBatteryUsage",
     "rtsp": "K10604GetRtspParam",
     "param_info": "K10020CheckCameraParams",  # Requires a Payload
 }

diff --git a/app/wyzecam/tutk/tutk_protocol.py b/app/wyzecam/tutk/tutk_protocol.py
@@ -453,6 +453,18 @@ def encode(self) -> bytes:
         return encode(self.code, bytes([self.status]))
 
 
+class K10050GetPowerLevel(TutkWyzeProtocolMessage):
+    def __init__(self):
+        super().__init__(10050)
+
+    def parse_response(self, resp_data):
+        data = json.loads(resp_data)
+        try:
+            return data["camerainfo"]["powerlevel"]
+        except KeyError:
+            return 0
+
+
 class K10056SetResolvingBit(TutkWyzeProtocolMessage):
     """
     A message used to set the resolution and bitrate of the camera.
@@ -846,7 +858,15 @@ def __init__(self):
         super().__init__(10448)
 
     def parse_response(self, resp_data):
-        return json.loads(resp_data)
+        data = json.loads(resp_data)
+        return {
+            "last_charge": data["0"],
+            "live_streaming": data["1"],
+            "events_uploaded": data["2"],
+            "events_filtered": data["3"],
+            "sd_recordings": data["4"],
+            "5": data["5"],
+        }
 
 
 class K10600SetRtspSwitch(TutkWyzeProtocolMessage):