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

fix: re-add consumables code

This commit is contained in:
Luke Bonaccorsi 2024-03-22 00:35:35 +00:00
parent 6cf2274e50
commit 6843472474
1 changed files with 7 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
custom_components/robovac/vacuum.py
View File

@ -357,11 +357,16 @@ class RoboVacEntity(StateVacuumEntity):
CONSUMABLE_CODE in self.tuyastatus CONSUMABLE_CODE in self.tuyastatus
and self.tuyastatus.get(CONSUMABLE_CODE) is not None 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( base64.b64decode(self.tuyastatus.get(CONSUMABLE_CODE)).decode(
"ascii" "ascii"
) )
)["consumable"]["duration"] )
if (
"consumable" in consumables
and "duration" in consumables["consumable"]
):
self._attr_consumables = consumables["consumable"]["duration"]
async def async_locate(self, **kwargs): async def async_locate(self, **kwargs):
"""Locate the vacuum cleaner.""" """Locate the vacuum cleaner."""