From 5c3e087500f2ae4b10c4149b216615c9d615e096 Mon Sep 17 00:00:00 2001 From: Necroneco Date: Sun, 5 Jan 2025 23:27:37 +0800 Subject: [PATCH] feat: avoid setting icon when device_class is defined --- custom_components/xiaomi_home/number.py | 2 +- custom_components/xiaomi_home/sensor.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/custom_components/xiaomi_home/number.py b/custom_components/xiaomi_home/number.py index 29bd6b7..1a38592 100644 --- a/custom_components/xiaomi_home/number.py +++ b/custom_components/xiaomi_home/number.py @@ -88,7 +88,7 @@ class Number(MIoTPropertyEntity, NumberEntity): if self.spec.external_unit: self._attr_native_unit_of_measurement = self.spec.external_unit # Set icon - if self.spec.icon: + if self.spec.icon and not self.device_class: self._attr_icon = self.spec.icon # Set value range if self._value_range: diff --git a/custom_components/xiaomi_home/sensor.py b/custom_components/xiaomi_home/sensor.py index 68e2d2d..fb9f30b 100644 --- a/custom_components/xiaomi_home/sensor.py +++ b/custom_components/xiaomi_home/sensor.py @@ -116,7 +116,7 @@ class Sensor(MIoTPropertyEntity, SensorEntity): if spec.state_class: self._attr_state_class = spec.state_class # Set icon - if spec.icon: + if spec.icon and not self.device_class: self._attr_icon = spec.icon @property