Update entity descriptions

Breina · Feb 22, 2024 · 8aa3734 · 8aa3734
1 parent 75a9c76
commit 8aa3734
Show file tree

Hide file tree

Showing 3 changed files with 14 additions and 2 deletions.
diff --git a/custom_components/idrac_power/binary_sensor.py b/custom_components/idrac_power/binary_sensor.py
@@ -59,6 +59,8 @@ def __init__(self, hass, rest: IdracRest, device_info, unique_id, name):
             name=name,
             icon='mdi:power',
             device_class=BinarySensorDeviceClass.RUNNING,
+            device_info=device_info,
+            unique_id=unique_id,
         )
 
         self._attr_device_info = device_info

diff --git a/custom_components/idrac_power/button.py b/custom_components/idrac_power/button.py
@@ -53,6 +53,8 @@ def __init__(self, hass, rest: IdracRest, device_info, unique_id, name):
             name=name,
             icon='mdi:power',
             device_class=ButtonDeviceClass.UPDATE,
+            device_info=device_info,
+            unique_id = unique_id,
         )
 
         self._attr_device_info = device_info
@@ -79,6 +81,8 @@ def __init__(self, hass, rest: IdracRest, device_info, unique_id, name):
             name=name,
             icon='mdi:power',
             device_class=ButtonDeviceClass.UPDATE,
+            device_info=device_info,
+            unique_id=unique_id,
         )
 
         self._attr_device_info = device_info

diff --git a/custom_components/idrac_power/sensor.py b/custom_components/idrac_power/sensor.py
@@ -80,7 +80,9 @@ def __init__(self, hass, rest: IdracRest, device_info, unique_id, name):
             icon='mdi:lightning-bolt',
             native_unit_of_measurement='W',
             device_class=SensorDeviceClass.POWER,
-            state_class=SensorStateClass.MEASUREMENT
+            state_class=SensorStateClass.MEASUREMENT,
+            device_info=device_info,
+            unique_id=unique_id,
         )
 
         self._attr_device_info = device_info
@@ -113,7 +115,9 @@ def __init__(self, hass, rest: IdracRest, device_info, unique_id, name, id):
             name=name,
             icon='mdi:fan',
             native_unit_of_measurement='RPM',
-            state_class=SensorStateClass.MEASUREMENT
+            state_class=SensorStateClass.MEASUREMENT,
+            device_info=device_info,
+            unique_id=unique_id,
         )
 
         self.custom_name = name
@@ -150,6 +154,8 @@ def __init__(self, hass, rest: IdracRest, device_info, unique_id, name, id):
             device_class=SensorDeviceClass.TEMPERATURE,
             state_class=SensorStateClass.MEASUREMENT,
             native_unit_of_measurement='°C',
+            device_info=device_info,
+            unique_id=unique_id,
         )
         self.custom_name = name