Merge pull request #11 from fredrike/patch-1

Add support to configure sections.

Author: MKR

README.md

@@ -28,6 +28,7 @@ This component does not require, nor conflict with, the default Plex components.
 | download_images | true | no | Setting this to false will turn off downloading of images, but will require certain Plex settings to work. See below.
 | img_dir | '/custom-lovelace/upcoming-media-card/images/plex/' | no | This option allows you to choose a custom directory to store images in if you enable download_images.
 | ssl_cert | false | no | If you provide your own SSL certificate in Plex's network settings set this to true.
+| section_types | movie, show | no | Allows you to specify which section types to consider [movie, show].
 
 #### By default this addon automatically downloads images from Plex to your /www/custom-lovelace/upcoming-media-card/ directory. The directory is automatically created & only images reported in the upcoming list are downloaded. Images are small in size and are removed automatically when no longer needed. Currently & unfortunately, this may not work on all systems.
 
@@ -45,6 +46,9 @@ This component does not require, nor conflict with, the default Plex components.
       ssl_cert: false
       download_images: false
       max: 5
+      section_types:
+      - movie
+      - show
 
 ### Sample for ui-lovelace.yaml:
 

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,