Fix exception when parts of Pollen.com can't be reached (#15296)

Fix exception when parts of Pollen.com can't be reached
2018-07-04 17:30:15 -06:00 · 2018-07-04 17:30:15 -06:00 · f65c3940ae
parent 91d6d0df84
commit f65c3940ae
1 changed files with 5 additions and 2 deletions
--- a/homeassistant/components/sensor/pollen.py
+++ b/homeassistant/components/sensor/pollen.py
@ -183,9 +183,12 @@ class PollencomSensor(Entity):
            return

        if self._category:
-            data = self.pollencom.data[self._category]['Location']
+            data = self.pollencom.data[self._category].get('Location')
        else:
-            data = self.pollencom.data[self._type]['Location']
+            data = self.pollencom.data[self._type].get('Location')
+
+        if not data:
+            return

        indices = [p['Index'] for p in data['periods']]
        average = round(mean(indices), 1)