diff --git a/homeassistant/components/kostal_plenticore/select.py b/homeassistant/components/kostal_plenticore/select.py index 4e594095f7e..1844f0894a5 100644 --- a/homeassistant/components/kostal_plenticore/select.py +++ b/homeassistant/components/kostal_plenticore/select.py @@ -9,7 +9,7 @@ from homeassistant.components.select import SelectEntity from homeassistant.config_entries import ConfigEntry from homeassistant.const import DEVICE_DEFAULT_NAME from homeassistant.core import HomeAssistant -from homeassistant.helpers.entity import DeviceInfo +from homeassistant.helpers.entity import DeviceInfo, EntityCategory from homeassistant.helpers.entity_platform import AddEntitiesCallback from homeassistant.helpers.update_coordinator import CoordinatorEntity @@ -54,6 +54,8 @@ async def async_setup_entry( class PlenticoreDataSelect(CoordinatorEntity, SelectEntity, ABC): """Representation of a Plenticore Select.""" + _attr_entity_category = EntityCategory.CONFIG + def __init__( self, coordinator, diff --git a/homeassistant/components/kostal_plenticore/sensor.py b/homeassistant/components/kostal_plenticore/sensor.py index 15971cec68d..bef47e7c605 100644 --- a/homeassistant/components/kostal_plenticore/sensor.py +++ b/homeassistant/components/kostal_plenticore/sensor.py @@ -10,7 +10,7 @@ from homeassistant.components.sensor import ATTR_STATE_CLASS, SensorEntity from homeassistant.config_entries import ConfigEntry from homeassistant.const import ATTR_DEVICE_CLASS, ATTR_ICON, ATTR_UNIT_OF_MEASUREMENT from homeassistant.core import HomeAssistant -from homeassistant.helpers.entity import DeviceInfo +from homeassistant.helpers.entity import DeviceInfo, EntityCategory from homeassistant.helpers.update_coordinator import CoordinatorEntity from .const import ( @@ -65,6 +65,7 @@ async def async_setup_entry( sensor_data, PlenticoreDataFormatter.get_method(fmt), plenticore.device_info, + None, ) ) @@ -96,6 +97,7 @@ async def async_setup_entry( sensor_data, PlenticoreDataFormatter.get_method(fmt), plenticore.device_info, + EntityCategory.DIAGNOSTIC, ) ) @@ -116,6 +118,7 @@ class PlenticoreDataSensor(CoordinatorEntity, SensorEntity): sensor_data: dict[str, Any], formatter: Callable[[str], Any], device_info: DeviceInfo, + entity_category: EntityCategory, ): """Create a new Sensor Entity for Plenticore process data.""" super().__init__(coordinator) @@ -130,6 +133,8 @@ class PlenticoreDataSensor(CoordinatorEntity, SensorEntity): self._device_info = device_info + self._attr_entity_category = entity_category + @property def available(self) -> bool: """Return if entity is available.""" diff --git a/homeassistant/components/kostal_plenticore/switch.py b/homeassistant/components/kostal_plenticore/switch.py index b3b1ba29e84..ea1f78283f3 100644 --- a/homeassistant/components/kostal_plenticore/switch.py +++ b/homeassistant/components/kostal_plenticore/switch.py @@ -8,7 +8,7 @@ import logging from homeassistant.components.switch import SwitchEntity from homeassistant.config_entries import ConfigEntry from homeassistant.core import HomeAssistant -from homeassistant.helpers.entity import DeviceInfo +from homeassistant.helpers.entity import DeviceInfo, EntityCategory from homeassistant.helpers.entity_platform import AddEntitiesCallback from homeassistant.helpers.update_coordinator import CoordinatorEntity @@ -70,6 +70,8 @@ async def async_setup_entry( class PlenticoreDataSwitch(CoordinatorEntity, SwitchEntity, ABC): """Representation of a Plenticore Switch.""" + _attr_entity_category = EntityCategory.CONFIG + def __init__( self, coordinator,