fix: #283 fix error on empty cross unit groups

fwmarcel · Feb 10, 2025 · 7e6fe13 · 7e6fe13
1 parent 4721369
commit 7e6fe13
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 5 deletions.
diff --git a/custom_components/divera/divera.py b/custom_components/divera/divera.py
@@ -320,8 +320,8 @@ def get_last_alarm_attributes(self) -> dict:
         alarm = self.__data["data"]["alarm"]["items"].get(str(last_alarm_id), {})
 
         cross_unit_meta = alarm.get("cross_unit_meta", {})
-        cross_unit_groups = cross_unit_meta.get("groups", [])
-        cross_unit_clusters = cross_unit_meta.get("clusters", [])
+        cross_unit_groups = cross_unit_meta.get("groups", {})
+        cross_unit_clusters = cross_unit_meta.get("clusters", {})
 
         groups = []
 
@@ -330,7 +330,7 @@ def get_last_alarm_attributes(self) -> dict:
             if group_name is not None:
                 groups.append(group_name)
             else:
-                cug = cross_unit_groups.get(str(group_id))
+                cug = cross_unit_groups.get(str(group_id), {})
                 if cug:
                     cluster_id = cug.get("cluster_id")
                     cluster_name = cross_unit_clusters.get(str(cluster_id), {}).get(

diff --git a/custom_components/divera/sensor.py b/custom_components/divera/sensor.py
@@ -89,10 +89,11 @@ async def async_setup_entry(
                     },
                     icon="mdi:fire-truck",
                     value_fn=lambda divera, vid=vehicle_id: divera.get_vehicle_state(
-                        vid  # noqa: B023
-                    ),
+                        vid
+                    ),  # noqa: B023
                     attribute_fn=lambda divera,
                     vid=vehicle_id: divera.get_vehicle_attributes(vid),  # noqa: B023
+                    entity_registry_enabled_default=False,
                 ),
             )
             entities.append(vehicle_entity)