From ae8604d4291f5ffde569ed604afc9b5dc81d0338 Mon Sep 17 00:00:00 2001 From: Shay Levy Date: Sat, 30 Apr 2022 17:46:27 +0300 Subject: [PATCH] Fix Litterrobot available property (#71102) --- homeassistant/components/litterrobot/vacuum.py | 10 ++++++---- 1 file changed, 6 insertions(+), 4 deletions(-) diff --git a/homeassistant/components/litterrobot/vacuum.py b/homeassistant/components/litterrobot/vacuum.py index 96908c1fa9d..0d9a2b1dc3d 100644 --- a/homeassistant/components/litterrobot/vacuum.py +++ b/homeassistant/components/litterrobot/vacuum.py @@ -18,7 +18,7 @@ from homeassistant.components.vacuum import ( VacuumEntityFeature, ) from homeassistant.config_entries import ConfigEntry -from homeassistant.const import STATE_OFF, STATE_UNAVAILABLE +from homeassistant.const import STATE_OFF from homeassistant.core import HomeAssistant from homeassistant.helpers import config_validation as cv, entity_platform from homeassistant.helpers.entity_platform import AddEntitiesCallback @@ -96,12 +96,14 @@ class LitterRobotCleaner(LitterRobotControlEntity, StateVacuumEntity): | VacuumEntityFeature.TURN_ON ) + @property + def available(self) -> bool: + """Return True if the cleaner has been seen recently.""" + return self.robot.last_seen > datetime.now(timezone.utc) - UNAVAILABLE_AFTER + @property def state(self) -> str: """Return the state of the cleaner.""" - if self.robot.last_seen < datetime.now(timezone.utc) - UNAVAILABLE_AFTER: - return STATE_UNAVAILABLE - return LITTER_BOX_STATUS_STATE_MAP.get(self.robot.status, STATE_ERROR) @property