Add support to configure sections.

Author: fredrike

custom_components/sensor/plex_recently_added.py

@@ -28,6 +28,7 @@
 CONF_TOKEN = 'token'
 CONF_MAX = 'max'
 CONF_IMG_CACHE = 'img_dir'
+CONF_SECTION_TYPES = 'section_types'
 
 PLATFORM_SCHEMA = PLATFORM_SCHEMA.extend({
     vol.Optional(CONF_SSL, default=False): cv.boolean,
@@ -38,6 +39,8 @@
     vol.Optional(CONF_DL_IMAGES, default=True): cv.boolean,
     vol.Optional(CONF_HOST, default='localhost'): cv.string,
     vol.Optional(CONF_PORT, default=32400): cv.port,
+    vol.Optional(CONF_SECTION_TYPES, 
+                default=['movie', 'show']): vol.All(cv.ensure_list, [cv.string]),
     vol.Optional(CONF_IMG_CACHE, 
                 default='/custom-lovelace/upcoming-media-card/images/plex/'): cv.string
 })
@@ -63,6 +66,7 @@ def __init__(self, hass, conf):
         self.server_name = conf.get(CONF_SERVER)
         self.max_items = int(conf.get(CONF_MAX))
         self.dl_images = conf.get(CONF_DL_IMAGES)
+        self.sections = conf.get(CONF_SECTION_TYPES)
         if self.server_name:
             self.server_ip, self.local_ip, self.port = get_server_ip(
                 self.server_name, self.token)
@@ -197,7 +201,8 @@ def update(self):
         try:
             libraries = plex.get(all_libraries, headers=headers, timeout=10)
             for lib_section in libraries.json()['MediaContainer']['Directory']:
-                sections.append(lib_section['key'])
+                if lib_section['type'] in self.sections:
+                    sections.append(lib_section['key'])
         except OSError:
             _LOGGER.warning("Host %s is not available", self.server_ip)
             self._state = '%s cannot be reached' % self.server_ip
@@ -254,6 +259,9 @@ def update(self):
                         elif media['type'] == 'episode':
                             poster = media.get('grandparentThumb', '')
                             fanart = media.get('grandparentArt', '')
+                        else:
+                            _LOGGER.error("Media type: %s", media['type'])
+                            continue
                         poster_jpg = '{}p{}.jpg'.format(directory,
                                                         media['ratingKey'])
                         fanart_jpg = '{}f{}.jpg'.format(directory,