Handle missing device config

andrew-codechimp · Dec 13, 2023 · d04b92e · d04b92e
1 parent ee7a95a
commit d04b92e
Showing 1 changed file with 11 additions and 4 deletions.
diff --git a/custom_components/battery_notes/library.py b/custom_components/battery_notes/library.py
@@ -27,10 +27,17 @@ class Library:  # pylint: disable=too-few-public-methods
 
     def __init__(self, hass: HomeAssistant) -> None:
         """Init."""
-        json_path = os.path.join(
-            BUILT_IN_DATA_DIRECTORY,
-            hass.data[DOMAIN][DOMAIN_CONFIG].get(CONF_LIBRARY, "library.json"),
-        )
+
+        if DOMAIN_CONFIG not in hass.data[DOMAIN]:
+            json_path = os.path.join(
+                BUILT_IN_DATA_DIRECTORY,
+                "library.json",
+            )
+        else:
+            json_path = os.path.join(
+                BUILT_IN_DATA_DIRECTORY,
+                hass.data[DOMAIN][DOMAIN_CONFIG].get(CONF_LIBRARY, "library.json"),
+            )
 
         _LOGGER.debug("Using library file at %s", json_path)