Merge branch 'main' of github.com:hugobloem/stateful_scenes

hugobloem · Apr 27, 2024 · 5a26a5f · 5a26a5f
2 parents 231141d + ee09db1
commit 5a26a5f
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/custom_components/stateful_scenes/StatefulScenes.py b/custom_components/stateful_scenes/StatefulScenes.py
@@ -25,14 +25,14 @@ def get_entity_id_from_id(hass: HomeAssistant, id: str) -> str:
     entity_ids = hass.states.async_entity_ids("scene")
     for entity_id in entity_ids:
         state = hass.states.get(entity_id)
-        if state.attributes["id"] == id:
+        if state.attributes.get("id", None) == id:
             return entity_id
     return None
 
 def get_id_from_entity_id(hass: HomeAssistant, entity_id: str) -> str:
     """Get scene id from entity_id."""
     state = hass.states.get(entity_id)
-    return state.attributes["id"]
+    return state.attributes.get("id", entity_id)
 
 
 def get_name_from_entity_id(hass: HomeAssistant, entity_id: str) -> str: