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