mirror of
https://github.com/home-assistant/core.git
synced 2025-07-10 06:47:09 +00:00
Add more diagnostic sensors to iRobot (#84995)
Co-authored-by: 930913 <3722064+930913@users.noreply.github.com> Co-authored-by: Xitee <59659167+Xitee1@users.noreply.github.com> Co-authored-by: J. Nick Koston <nick@koston.org> Co-authored-by: Joost Lekkerkerker <joostlek@outlook.com>
This commit is contained in:
parent
dff18b4a16
commit
2d833fd6ea
@ -105,6 +105,18 @@ class IRobotEntity(Entity):
|
|||||||
"""Return the battery level of the vacuum cleaner."""
|
"""Return the battery level of the vacuum cleaner."""
|
||||||
return self.vacuum_state.get("batPct")
|
return self.vacuum_state.get("batPct")
|
||||||
|
|
||||||
|
@property
|
||||||
|
def _run_stats(self):
|
||||||
|
return self.vacuum_state.get("bbrun")
|
||||||
|
|
||||||
|
@property
|
||||||
|
def _mission_stats(self):
|
||||||
|
return self.vacuum_state.get("bbmssn")
|
||||||
|
|
||||||
|
@property
|
||||||
|
def _battery_stats(self):
|
||||||
|
return self.vacuum_state.get("bbchg3")
|
||||||
|
|
||||||
@property
|
@property
|
||||||
def _robot_state(self):
|
def _robot_state(self):
|
||||||
"""Return the state of the vacuum cleaner."""
|
"""Return the state of the vacuum cleaner."""
|
||||||
|
@ -1,7 +1,11 @@
|
|||||||
"""Sensor for checking the battery level of Roomba."""
|
"""Sensor for checking the battery level of Roomba."""
|
||||||
from homeassistant.components.sensor import SensorDeviceClass, SensorEntity
|
from homeassistant.components.sensor import (
|
||||||
|
SensorDeviceClass,
|
||||||
|
SensorEntity,
|
||||||
|
SensorStateClass,
|
||||||
|
)
|
||||||
from homeassistant.config_entries import ConfigEntry
|
from homeassistant.config_entries import ConfigEntry
|
||||||
from homeassistant.const import PERCENTAGE, EntityCategory
|
from homeassistant.const import PERCENTAGE, EntityCategory, UnitOfTime
|
||||||
from homeassistant.core import HomeAssistant
|
from homeassistant.core import HomeAssistant
|
||||||
from homeassistant.helpers.entity_platform import AddEntitiesCallback
|
from homeassistant.helpers.entity_platform import AddEntitiesCallback
|
||||||
|
|
||||||
@ -18,8 +22,31 @@ async def async_setup_entry(
|
|||||||
domain_data = hass.data[DOMAIN][config_entry.entry_id]
|
domain_data = hass.data[DOMAIN][config_entry.entry_id]
|
||||||
roomba = domain_data[ROOMBA_SESSION]
|
roomba = domain_data[ROOMBA_SESSION]
|
||||||
blid = domain_data[BLID]
|
blid = domain_data[BLID]
|
||||||
|
|
||||||
roomba_vac = RoombaBattery(roomba, blid)
|
roomba_vac = RoombaBattery(roomba, blid)
|
||||||
async_add_entities([roomba_vac], True)
|
roomba_battery_cycles = BatteryCycles(roomba, blid)
|
||||||
|
roomba_cleaning_time = CleaningTime(roomba, blid)
|
||||||
|
roomba_average_mission_time = AverageMissionTime(roomba, blid)
|
||||||
|
roomba_total_missions = MissionSensor(roomba, blid, "total", "nMssn")
|
||||||
|
roomba_success_missions = MissionSensor(roomba, blid, "successful", "nMssnOk")
|
||||||
|
roomba_canceled_missions = MissionSensor(roomba, blid, "canceled", "nMssnC")
|
||||||
|
roomba_failed_missions = MissionSensor(roomba, blid, "failed", "nMssnF")
|
||||||
|
roomba_scrubs_count = ScrubsCount(roomba, blid)
|
||||||
|
|
||||||
|
async_add_entities(
|
||||||
|
[
|
||||||
|
roomba_vac,
|
||||||
|
roomba_battery_cycles,
|
||||||
|
roomba_cleaning_time,
|
||||||
|
roomba_average_mission_time,
|
||||||
|
roomba_total_missions,
|
||||||
|
roomba_success_missions,
|
||||||
|
roomba_canceled_missions,
|
||||||
|
roomba_failed_missions,
|
||||||
|
roomba_scrubs_count,
|
||||||
|
],
|
||||||
|
True,
|
||||||
|
)
|
||||||
|
|
||||||
|
|
||||||
class RoombaBattery(IRobotEntity, SensorEntity):
|
class RoombaBattery(IRobotEntity, SensorEntity):
|
||||||
@ -38,3 +65,147 @@ class RoombaBattery(IRobotEntity, SensorEntity):
|
|||||||
def native_value(self):
|
def native_value(self):
|
||||||
"""Return the state of the sensor."""
|
"""Return the state of the sensor."""
|
||||||
return self._battery_level
|
return self._battery_level
|
||||||
|
|
||||||
|
|
||||||
|
class BatteryCycles(IRobotEntity, SensorEntity):
|
||||||
|
"""Class to hold Roomba Sensor basic info."""
|
||||||
|
|
||||||
|
_attr_entity_category = EntityCategory.DIAGNOSTIC
|
||||||
|
_attr_icon = "mdi:counter"
|
||||||
|
|
||||||
|
@property
|
||||||
|
def name(self):
|
||||||
|
"""Return the name of the sensor."""
|
||||||
|
return "Battery cycles"
|
||||||
|
|
||||||
|
@property
|
||||||
|
def unique_id(self):
|
||||||
|
"""Return the ID of this sensor."""
|
||||||
|
return f"battery_cycles_{self._blid}"
|
||||||
|
|
||||||
|
@property
|
||||||
|
def state_class(self):
|
||||||
|
"""Return the state class of this entity, from STATE_CLASSES, if any."""
|
||||||
|
return SensorStateClass.MEASUREMENT
|
||||||
|
|
||||||
|
@property
|
||||||
|
def native_value(self):
|
||||||
|
"""Return the state of the sensor."""
|
||||||
|
return self._battery_stats.get("nLithChrg") or self._battery_stats.get(
|
||||||
|
"nNimhChrg"
|
||||||
|
)
|
||||||
|
|
||||||
|
|
||||||
|
class CleaningTime(IRobotEntity, SensorEntity):
|
||||||
|
"""Class to hold Roomba Sensor basic info."""
|
||||||
|
|
||||||
|
_attr_entity_category = EntityCategory.DIAGNOSTIC
|
||||||
|
_attr_icon = "mdi:clock"
|
||||||
|
_attr_native_unit_of_measurement = UnitOfTime.HOURS
|
||||||
|
|
||||||
|
@property
|
||||||
|
def name(self):
|
||||||
|
"""Return the name of the sensor."""
|
||||||
|
return f"{self._name} cleaning time total"
|
||||||
|
|
||||||
|
@property
|
||||||
|
def unique_id(self):
|
||||||
|
"""Return the ID of this sensor."""
|
||||||
|
return f"total_cleaning_time_{self._blid}"
|
||||||
|
|
||||||
|
@property
|
||||||
|
def native_value(self):
|
||||||
|
"""Return the state of the sensor."""
|
||||||
|
return self._run_stats.get("hr")
|
||||||
|
|
||||||
|
|
||||||
|
class AverageMissionTime(IRobotEntity, SensorEntity):
|
||||||
|
"""Class to hold Roomba Sensor basic info."""
|
||||||
|
|
||||||
|
_attr_entity_category = EntityCategory.DIAGNOSTIC
|
||||||
|
_attr_icon = "mdi:clock"
|
||||||
|
|
||||||
|
@property
|
||||||
|
def name(self):
|
||||||
|
"""Return the name of the sensor."""
|
||||||
|
return "Average mission time"
|
||||||
|
|
||||||
|
@property
|
||||||
|
def unique_id(self):
|
||||||
|
"""Return the ID of this sensor."""
|
||||||
|
return f"average_mission_time_{self._blid}"
|
||||||
|
|
||||||
|
@property
|
||||||
|
def native_unit_of_measurement(self):
|
||||||
|
"""Return the unit_of_measurement of the device."""
|
||||||
|
return UnitOfTime.MINUTES
|
||||||
|
|
||||||
|
@property
|
||||||
|
def native_value(self):
|
||||||
|
"""Return the state of the sensor."""
|
||||||
|
return self._mission_stats.get("aMssnM")
|
||||||
|
|
||||||
|
|
||||||
|
class MissionSensor(IRobotEntity, SensorEntity):
|
||||||
|
"""Class to hold the Roomba missions info."""
|
||||||
|
|
||||||
|
def __init__(self, roomba, blid, mission_type, mission_value_string):
|
||||||
|
"""Initialise iRobot sensor with mission details."""
|
||||||
|
super().__init__(roomba, blid)
|
||||||
|
self._mission_type = mission_type
|
||||||
|
self._mission_value_string = mission_value_string
|
||||||
|
|
||||||
|
_attr_entity_category = EntityCategory.DIAGNOSTIC
|
||||||
|
_attr_icon = "mdi:counter"
|
||||||
|
|
||||||
|
@property
|
||||||
|
def name(self):
|
||||||
|
"""Return the name of the sensor."""
|
||||||
|
return f"Missions {self._mission_type}"
|
||||||
|
|
||||||
|
@property
|
||||||
|
def unique_id(self):
|
||||||
|
"""Return the ID of this sensor."""
|
||||||
|
return f"{self._mission_type}_missions_{self._blid}"
|
||||||
|
|
||||||
|
@property
|
||||||
|
def state_class(self):
|
||||||
|
"""Return the state class of this entity, from STATE_CLASSES, if any."""
|
||||||
|
return SensorStateClass.MEASUREMENT
|
||||||
|
|
||||||
|
@property
|
||||||
|
def native_value(self):
|
||||||
|
"""Return the state of the sensor."""
|
||||||
|
return self._mission_stats.get(self._mission_value_string)
|
||||||
|
|
||||||
|
|
||||||
|
class ScrubsCount(IRobotEntity, SensorEntity):
|
||||||
|
"""Class to hold Roomba Sensor basic info."""
|
||||||
|
|
||||||
|
_attr_entity_category = EntityCategory.DIAGNOSTIC
|
||||||
|
_attr_icon = "mdi:counter"
|
||||||
|
|
||||||
|
@property
|
||||||
|
def name(self):
|
||||||
|
"""Return the name of the sensor."""
|
||||||
|
return "Scrubs count"
|
||||||
|
|
||||||
|
@property
|
||||||
|
def unique_id(self):
|
||||||
|
"""Return the ID of this sensor."""
|
||||||
|
return f"scrubs_count_{self._blid}"
|
||||||
|
|
||||||
|
@property
|
||||||
|
def state_class(self):
|
||||||
|
"""Return the state class of this entity, from STATE_CLASSES, if any."""
|
||||||
|
return SensorStateClass.MEASUREMENT
|
||||||
|
|
||||||
|
@property
|
||||||
|
def entity_registry_enabled_default(self):
|
||||||
|
"""Disable sensor by default."""
|
||||||
|
return False
|
||||||
|
|
||||||
|
@property
|
||||||
|
def native_value(self):
|
||||||
|
"""Return the state of the sensor."""
|
||||||
|
return self._run_stats.get("nScrubs")
|
||||||
|
Loading…
x
Reference in New Issue
Block a user