Fix the missing error key

This commit is contained in:
Dennis Melzer 2023-12-19 10:03:22 +01:00
parent ae189ce422
commit 199e541168
1 changed files with 4 additions and 2 deletions

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."""