Fix error when Series missing 'episodeFileCount' or 'episodeCount' (#15824)

* Fix error when Series missing 'episodeFileCount' or 'episodeCount' * Update sonarr.py * Update sonarr.py
2018-08-06 13:18:36 +03:00 · 2018-08-06 13:18:36 +03:00 · 6031801206
parent 9cfe0db3c8
commit 6031801206
1 changed files with 6 additions and 2 deletions
--- a/homeassistant/components/sensor/sonarr.py
+++ b/homeassistant/components/sensor/sonarr.py
@ -158,6 +158,10 @@ class SonarrSensor(Entity):
                )
        elif self.type == 'series':
            for show in self.data:
+                if 'episodeFileCount' not in show \
+                        or 'episodeCount' not in show:
+                    attributes[show['title']] = 'N/A'
+                else:
                    attributes[show['title']] = '{}/{} Episodes'.format(
                        show['episodeFileCount'], show['episodeCount'])
        elif self.type == 'status':