Fix error when Series missing 'episodeFileCount' or 'episodeCount' (#…

…15824) * Fix error when Series missing 'episodeFileCount' or 'episodeCount' * Update sonarr.py * Update sonarr.py
home-assistant · Aug 6, 2018 · 6031801 · 6031801
1 parent 9cfe0db
commit 6031801
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/homeassistant/components/sensor/sonarr.py b/homeassistant/components/sensor/sonarr.py
@@ -158,8 +158,12 @@ def device_state_attributes(self):
                 )
         elif self.type == 'series':
             for show in self.data:
-                attributes[show['title']] = '{}/{} Episodes'.format(
-                    show['episodeFileCount'], show['episodeCount'])
+                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':
             attributes = self.data
         return attributes