zach
/
eufy-robovac
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Fix the missing error key

This commit is contained in:
Dennis Melzer 2023-12-19 10:03:22 +01:00 committed by Luke Bonaccorsi
parent 19aefa8e65
commit 8e6e311bfc
1 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
custom_components/robovac/vacuum.py
View File

@ -359,11 +359,13 @@ class RoboVacEntity(StateVacuumEntity):
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."""