fix(sensor): catch keyerror on update

alandtse · Sep 27, 2019 · 29e98c1 · 29e98c1
1 parent 5e1870d
commit 29e98c1
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/custom_components/alexa_media/sensor.py b/custom_components/alexa_media/sensor.py
@@ -248,7 +248,11 @@ async def async_update(self):
             pass
         account_dict = (self.hass.data[DATA_ALEXAMEDIA]['accounts']
                         [self._account])
-        self._n_dict = account_dict['notifications'][self._dev_id][self._type]
+        try:
+            self._n_dict = (account_dict['notifications'][self._dev_id]
+                            [self._type])
+        except KeyError:
+            self._n_dict = None
         self._all = (sorted(self._n_dict.items(),
                             key=lambda x: x[1][self._sensor_property])
                      if self._n_dict else [])