Update base.py

azerty9971 · Sep 1, 2024 · c83cd1c · c83cd1c
1 parent a59c935
commit c83cd1c
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/homeassistant/components/tuya/base.py b/homeassistant/components/tuya/base.py
@@ -122,9 +122,12 @@ def from_json(cls, data: str) -> Self:
     def from_raw(cls, data: str) -> Self:
         """Decode base64 string and return a ElectricityTypeData object."""
         raw = base64.b64decode(data)
-        voltage = struct.unpack(">H", raw[0:2])[0] / 10.0
-        electriccurrent = struct.unpack(">L", b"\x00" + raw[2:5])[0] / 1000.0
-        power = struct.unpack(">L", b"\x00" + raw[5:8])[0] / 1000.0
+        try:
+            voltage = struct.unpack(">H", raw[0:2])[0] / 10.0
+            electriccurrent = struct.unpack(">L", b"\x00" + raw[2:5])[0] / 1000.0
+            power = struct.unpack(">L", b"\x00" + raw[5:8])[0] / 1000.0
+        except:
+            LOGGER.warning(f"Failed UNPACK: {data}")
         return cls(
             electriccurrent=str(electriccurrent), power=str(power), voltage=str(voltage)
         )