Add entity category to roomba (#58362)
This commit is contained in:
parent
d05127cb71
commit
1a261f7802
1 changed files with 7 additions and 1 deletions
|
@ -1,7 +1,11 @@
|
|||
"""Sensor for checking the battery level of Roomba."""
|
||||
from homeassistant.components.sensor import SensorEntity
|
||||
from homeassistant.components.vacuum import STATE_DOCKED
|
||||
from homeassistant.const import DEVICE_CLASS_BATTERY, PERCENTAGE
|
||||
from homeassistant.const import (
|
||||
DEVICE_CLASS_BATTERY,
|
||||
ENTITY_CATEGORY_DIAGNOSTIC,
|
||||
PERCENTAGE,
|
||||
)
|
||||
from homeassistant.helpers.icon import icon_for_battery_level
|
||||
|
||||
from .const import BLID, DOMAIN, ROOMBA_SESSION
|
||||
|
@ -20,6 +24,8 @@ async def async_setup_entry(hass, config_entry, async_add_entities):
|
|||
class RoombaBattery(IRobotEntity, SensorEntity):
|
||||
"""Class to hold Roomba Sensor basic info."""
|
||||
|
||||
_attr_entity_category = ENTITY_CATEGORY_DIAGNOSTIC
|
||||
|
||||
@property
|
||||
def name(self):
|
||||
"""Return the name of the sensor."""
|
||||
|
|
Loading…
Add table
Reference in a new issue