fix: Missing extra state attributes

davidrapan · Dec 16, 2024 · 70a3f76 · 70a3f76
1 parent 118e8f8
commit 70a3f76
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/custom_components/solarman/entity.py b/custom_components/solarman/entity.py
@@ -79,7 +79,7 @@ def update(self):
             if "inverse" in self.attributes and self._attr_native_value:
                 self._attr_extra_state_attributes["−x"] = -self._attr_native_value
             for attr in filter(lambda a: a in self.coordinator.data, self.attributes):
-                self._attr_extra_state_attributes[attr.replace(f"{self._attr_name} ", "")] = get_tuple(self.coordinator.data.get(attr))
+                self._attr_extra_state_attributes[self.attributes[attr].replace(f"{self._attr_name} ", "")] = get_tuple(self.coordinator.data.get(attr))
 
 class SolarmanEntity(SolarmanCoordinatorEntity):
     def __init__(self, coordinator, sensor):
@@ -111,7 +111,7 @@ def __init__(self, coordinator, sensor):
         if description := sensor.get("description"):
             self._attr_extra_state_attributes = self._attr_extra_state_attributes | { "description": description }
 
-        self.attributes = sensor.get("attributes")
+        self.attributes = {slugify('_'.join(filter(None, (x, "sensor")))): x for x in attrs} if (attrs := sensor.get("attributes")) is not None else None
         self.registers = sensor.get("registers")
 
     def _friendly_name_internal(self) -> str | None: