Fix the missing error key

CodeFoodPixels · Jan 13, 2024 · 8e6e311 · 8e6e311
1 parent 19aefa8
commit 8e6e311
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/custom_components/robovac/vacuum.py b/custom_components/robovac/vacuum.py
@@ -359,11 +359,13 @@ def update_entity_values(self):
                     CONSUMABLE_CODE in self.tuyastatus
                     and self.tuyastatus.get(CONSUMABLE_CODE) is not None
                 ):
-                    self._attr_consumables = ast.literal_eval(
+                    consumables = ast.literal_eval(
                         base64.b64decode(self.tuyastatus.get(CONSUMABLE_CODE)).decode(
                             "ascii"
                         )
-                    )["consumable"]["duration"]
+                    )
+                    if "consumable" in consumables and "duration" in consumables:
+                        self._attr_consumables = consumables["consumable"]["duration"]
 
     async def async_locate(self, **kwargs):
         """Locate the vacuum cleaner."""