From b26e65d7e5e35a669dc84e089b2ba0c22db4b69b Mon Sep 17 00:00:00 2001 From: Milan Meulemans Date: Tue, 27 Jul 2021 00:47:06 +0200 Subject: [PATCH] Fix Rituals Perfume Genie sensors icons (#53517) --- .../components/rituals_perfume_genie/sensor.py | 18 ++++++++++++------ 1 file changed, 12 insertions(+), 6 deletions(-) diff --git a/homeassistant/components/rituals_perfume_genie/sensor.py b/homeassistant/components/rituals_perfume_genie/sensor.py index d4e10ba141b..7c957722384 100644 --- a/homeassistant/components/rituals_perfume_genie/sensor.py +++ b/homeassistant/components/rituals_perfume_genie/sensor.py @@ -51,9 +51,12 @@ class DiffuserPerfumeSensor(DiffuserEntity): """Initialize the perfume sensor.""" super().__init__(diffuser, coordinator, PERFUME_SUFFIX) - self._attr_icon = "mdi:tag-remove" - if diffuser.has_cartridge: - self._attr_icon = "mdi:tag-text" + @property + def icon(self) -> str: + """Return the perfume sensor icon.""" + if self._diffuser.has_cartridge: + return "mdi:tag-text" + return "mdi:tag-remove" @property def state(self) -> str: @@ -70,9 +73,12 @@ class DiffuserFillSensor(DiffuserEntity): """Initialize the fill sensor.""" super().__init__(diffuser, coordinator, FILL_SUFFIX) - self._attr_icon = "mdi:beaker-question" - if diffuser.has_cartridge: - self.attr_icon = "mdi:beaker" + @property + def icon(self) -> str: + """Return the fill sensor icon.""" + if self._diffuser.has_cartridge: + return "mdi:beaker" + return "mdi:beaker-question" @property def state(self) -> str: