Add support to entity registry for overriding device_class (#59985)

2021-11-22 17:38:06 +01:00 · 2021-11-22 17:38:06 +01:00 · 49a27e12ad
commit 49a27e12ad
parent 67e13b35db
31 changed files with 136 additions and 69 deletions
--- a/homeassistant/components/hue/migration.py
+++ b/homeassistant/components/hue/migration.py
@ -126,7 +126,7 @@ async def handle_v2_migration(hass: core.HomeAssistant, entry: ConfigEntry) -> N
                    continue
                # migrate sensors
                matched_dev_class = sensor_class_mapping.get(
-                    ent.device_class or "unknown"
+                    ent.original_device_class or "unknown"
                )
                if matched_dev_class is None:
                    # this may happen if we're looking at orphaned or unsupported entity