From 66ec37d60ae83c3d7223f72b8beb76860634c93a Mon Sep 17 00:00:00 2001 From: Maciej Bieniek Date: Sun, 26 Dec 2021 23:45:22 +0100 Subject: [PATCH 1/3] Remove led brightnes entity --- homeassistant/components/xiaomi_miio/number.py | 10 ++++++++++ 1 file changed, 10 insertions(+) diff --git a/homeassistant/components/xiaomi_miio/number.py b/homeassistant/components/xiaomi_miio/number.py index b2a3f4e9f5e39..f23ff82cbabef 100644 --- a/homeassistant/components/xiaomi_miio/number.py +++ b/homeassistant/components/xiaomi_miio/number.py @@ -5,6 +5,7 @@ from homeassistant.components.number import NumberEntity, NumberEntityDescription from homeassistant.config_entries import ConfigEntry +from homeassistant.components.number.const import DOMAIN as PLATFORM_DOMAIN from homeassistant.const import DEGREE, TIME_MINUTES from homeassistant.core import HomeAssistant, callback from homeassistant.helpers.entity import EntityCategory @@ -252,6 +253,15 @@ async def async_setup_entry( return for feature, description in NUMBER_TYPES.items(): + if feature == FEATURE_SET_LED_BRIGHTNESS and model != MODEL_FAN_ZA5: + # Delete LED bightness entity created by mistake if it exists + entity_reg = await hass.helpers.entity_registry.async_get_registry() + entity_id = entity_reg.async_get_entity_id( + PLATFORM_DOMAIN, DOMAIN, f"{description.key}_{config_entry.unique_id}" + ) + if entity_id: + entity_reg.async_remove(entity_id) + continue if feature & features: if ( description.key == ATTR_OSCILLATION_ANGLE From 61a252e59ed5c3660bef48b4bdbc83b7cfd25150 Mon Sep 17 00:00:00 2001 From: Maciej Bieniek Date: Sat, 1 Jan 2022 23:05:36 +0100 Subject: [PATCH 2/3] Isort --- homeassistant/components/xiaomi_miio/number.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/homeassistant/components/xiaomi_miio/number.py b/homeassistant/components/xiaomi_miio/number.py index f23ff82cbabef..e50e36d79e641 100644 --- a/homeassistant/components/xiaomi_miio/number.py +++ b/homeassistant/components/xiaomi_miio/number.py @@ -4,8 +4,8 @@ from dataclasses import dataclass from homeassistant.components.number import NumberEntity, NumberEntityDescription -from homeassistant.config_entries import ConfigEntry from homeassistant.components.number.const import DOMAIN as PLATFORM_DOMAIN +from homeassistant.config_entries import ConfigEntry from homeassistant.const import DEGREE, TIME_MINUTES from homeassistant.core import HomeAssistant, callback from homeassistant.helpers.entity import EntityCategory From 53bd5b82ddf6803d1e063b0bc506e8b7f34e84cd Mon Sep 17 00:00:00 2001 From: Paulus Schoutsen Date: Tue, 4 Jan 2022 13:36:46 -0800 Subject: [PATCH 3/3] Update homeassistant/components/xiaomi_miio/number.py --- homeassistant/components/xiaomi_miio/number.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/homeassistant/components/xiaomi_miio/number.py b/homeassistant/components/xiaomi_miio/number.py index e50e36d79e641..22e6b8da55fc0 100644 --- a/homeassistant/components/xiaomi_miio/number.py +++ b/homeassistant/components/xiaomi_miio/number.py @@ -255,7 +255,7 @@ async def async_setup_entry( for feature, description in NUMBER_TYPES.items(): if feature == FEATURE_SET_LED_BRIGHTNESS and model != MODEL_FAN_ZA5: # Delete LED bightness entity created by mistake if it exists - entity_reg = await hass.helpers.entity_registry.async_get_registry() + entity_reg = hass.helpers.entity_registry.async_get() entity_id = entity_reg.async_get_entity_id( PLATFORM_DOMAIN, DOMAIN, f"{description.key}_{config_entry.unique_id}" )