KeyError fix

dext0r · Oct 7, 2023 · 0277eae · 0277eae
1 parent 392fc65
commit 0277eae
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 2 deletions.
diff --git a/custom_components/yandex_smart_home/capability_color.py b/custom_components/yandex_smart_home/capability_color.py
@@ -395,7 +395,10 @@ def scenes_map(self) -> dict[ColorScene, list[str]]:
 
         if CONF_ENTITY_MODE_MAP in self._entity_config:
             scenes_map.update(
-                {ColorScene(k): v for k, v in self._entity_config[CONF_ENTITY_MODE_MAP].get(self.instance).items()}
+                {
+                    ColorScene(k): v
+                    for k, v in self._entity_config.get(CONF_ENTITY_MODE_MAP, {}).get(self.instance).items()
+                }
             )
 
         for scene, effects in scenes_map.items():

diff --git a/custom_components/yandex_smart_home/capability_mode.py b/custom_components/yandex_smart_home/capability_mode.py
@@ -76,7 +76,7 @@ def modes_map_config(self) -> dict[ModeCapabilityMode, list[str]] | None:
         if CONF_ENTITY_MODE_MAP in self._entity_config:
             return {
                 ModeCapabilityMode(k): v
-                for k, v in self._entity_config[CONF_ENTITY_MODE_MAP].get(self.instance).items()
+                for k, v in self._entity_config.get(CONF_ENTITY_MODE_MAP, {}).get(self.instance).items()
             }
 
         return None