feat: allow missing sections by faking empty AmcData

vermut · Jun 3, 2024 · bc8ff8a · bc8ff8a
1 parent 03e32dc
commit bc8ff8a
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/custom_components/amc_alarm/amc_alarm_api/api.py b/custom_components/amc_alarm/amc_alarm_api/api.py
@@ -239,8 +239,11 @@ def raw_states(self) -> dict[str, AmcCentralResponse]:
 
     def _get_section(self, central_id, section_index) -> AmcData | AmcNotification:
         central = self._raw_states[central_id]
-        zones = next(x for x in central.data if x.index == section_index)
-        return zones
+        try:
+            section = next(x for x in central.data if x.index == section_index)
+            return section
+        except StopIteration:
+            return AmcData(index=0, list=[], name="_none")
 
     def groups(self, central_id: str) -> AmcData:
         return self._get_section(central_id, CentralDataSections.GROUPS)