diff --git a/pyControl4/alarm.py b/pyControl4/alarm.py index aecb476..9d3bb64 100644 --- a/pyControl4/alarm.py +++ b/pyControl4/alarm.py @@ -141,13 +141,16 @@ async def getEmergencyTypes(self): data = await self.director.getItemInfo(self.item_id) jsonDictionary = json.loads(data) - if jsonDictionary[0]["capabilities"]["has_fire"]: + capabilities = ( + jsonDictionary[0].get("capabilities", {}) if jsonDictionary else {} + ) + if capabilities.get("has_fire"): types_list.append("Fire") - if jsonDictionary[0]["capabilities"]["has_medical"]: + if capabilities.get("has_medical"): types_list.append("Medical") - if jsonDictionary[0]["capabilities"]["has_panic"]: + if capabilities.get("has_panic"): types_list.append("Panic") - if jsonDictionary[0]["capabilities"]["has_police"]: + if capabilities.get("has_police"): types_list.append("Police") return types_list