Fix KeyError from Rain Delay binary sensor if no key exists on the de…

…vice status Fixes #7
sebr · Mar 12, 2020 · d26fd2c · d26fd2c
1 parent d9a47df
commit d26fd2c
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/custom_components/bhyve/binary_sensor.py b/custom_components/bhyve/binary_sensor.py
@@ -47,8 +47,8 @@ def _setup(self, device):
         self._attrs = {}
         self._available = device.get("is_connected", False)
 
-        device_status = device["status"]
-        rain_delay = device_status["rain_delay"]
+        device_status = device.get("status", {})
+        rain_delay = device_status.get("rain_delay")
 
         self._update_device_cb = None
         self._extract_rain_delay(rain_delay, device_status)