From 0e912da615ca876be9c9bde6b01fe92d81a25d1f Mon Sep 17 00:00:00 2001
From: Sanjoy Ghosh <c503ghosh@gmail.com>
Date: Sat, 25 May 2024 19:18:38 +0530
Subject: [PATCH] Controller battery category changed to diagnostic

---
 custom_components/dirigera_platform/sensor.py | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/custom_components/dirigera_platform/sensor.py b/custom_components/dirigera_platform/sensor.py
index 7a2727b..a4ccd99 100644
--- a/custom_components/dirigera_platform/sensor.py
+++ b/custom_components/dirigera_platform/sensor.py
@@ -11,6 +11,7 @@
 from homeassistant.components.sensor import SensorDeviceClass, SensorEntity
 from homeassistant.const import CONF_IP_ADDRESS, CONF_TOKEN
 from homeassistant.core import HomeAssistantError
+from homeassistant.helpers.entity import EntityCategory
 
 from .const import DOMAIN
 from .base_classes import ikea_base_device, ikea_base_device_sensor
@@ -204,6 +205,10 @@ def __init__(self,hass:core.HomeAssistant, hub:Hub, json_data:Controller):
         logger.debug("ikea_controller ctor...")
         super().__init__(hass , hub, json_data, hub.get_controller_by_id)
 
+    @property
+    def entity_category(self):
+        return EntityCategory.DIAGNOSTIC
+    
     @property
     def icon(self):
         return "mdi:battery"