Use host option when initializing hub

jason0x43 · May 11, 2020 · 1f755c5 · 1f755c5
1 parent d842e03
commit 1f755c5
Showing 1 changed file with 9 additions and 2 deletions.
diff --git a/custom_components/hubitat/device.py b/custom_components/hubitat/device.py
@@ -89,7 +89,12 @@ def entity_id(self) -> str:
     @property
     def host(self) -> str:
         """The IP address of the  associated Hubitat hub."""
-        return cast(str, self.config_entry.data.get(CONF_HOST))
+        return cast(
+            str,
+            self.config_entry.options.get(
+                CONF_HOST, self.config_entry.data.get(CONF_HOST)
+            ),
+        )
 
     @property
     def mac(self) -> Optional[str]:
@@ -221,7 +226,9 @@ async def async_update_options(hass: HomeAssistant, entry: ConfigEntry) -> None:
             _LOGGER.debug("Set temperature units to %s", temp_unit)
 
         hass.states.async_set(
-            hub.entity_id, "connected", {CONF_TEMPERATURE_UNIT: hub.temperature_unit}
+            hub.entity_id,
+            "connected",
+            {CONF_HOST: hub.host, CONF_TEMPERATURE_UNIT: hub.temperature_unit},
         )
 
     async def check_config(self) -> None: