PlexUpdate plugin updated for Plex Home

beetsplug/plexupdate.py

@@ -1,9 +1,11 @@
 """Updates an Plex library whenever the beets library is changed.
 
+Plex Home users enter the Plex Token to enable updating.
 Put something like the following in your config.yaml to configure:
     plex:
         host: localhost
         port: 32400
+        token: token
 """
 from __future__ import (division, absolute_import, print_function,
                         unicode_literals)
@@ -15,10 +17,10 @@
 from beets.plugins import BeetsPlugin
 
 
-def get_music_section(host, port):
+def get_music_section(host, port, token):
     """Getting the section key for the music library in Plex.
     """
-    api_endpoint = 'library/sections'
+    api_endpoint = append_token('library/sections', token)
     url = urljoin('http://{0}:{1}'.format(host, port), api_endpoint)
 
     # Sends request.
@@ -31,27 +33,37 @@ def get_music_section(host, port):
             return child.get('key')
 
 
-def update_plex(host, port):
+def update_plex(host, port, token):
     """Sends request to the Plex api to start a library refresh.
     """
     # Getting section key and build url.
-    section_key = get_music_section(host, port)
+    section_key = get_music_section(host, port, token)
     api_endpoint = 'library/sections/{0}/refresh'.format(section_key)
+    api_endpoint = append_token(api_endpoint, token)
     url = urljoin('http://{0}:{1}'.format(host, port), api_endpoint)
 
     # Sends request and returns requests object.
     r = requests.get(url)
     return r
 
 
+def append_token(url, token):
+    """Appends the Plex Home token to the api call if required.
+    """
+    if token:
+        url += '?X-Plex-Token={0}'.format(token)
+    return url
+
+
 class PlexUpdate(BeetsPlugin):
     def __init__(self):
         super(PlexUpdate, self).__init__()
 
         # Adding defaults.
         config['plex'].add({
             u'host': u'localhost',
-            u'port': 32400})
+            u'port': 32400,
+            u'token': u''})
 
         self.register_listener('database_change', self.listen_for_db_change)
 
@@ -68,7 +80,8 @@ def update(self, lib):
         try:
             update_plex(
                 config['plex']['host'].get(),
-                config['plex']['port'].get())
+                config['plex']['port'].get(),
+                config['plex']['token'].get())
             self._log.info('... started.')
 
         except requests.exceptions.RequestException:

test/test_plexupdate.py

@@ -87,7 +87,8 @@ def test_get_music_section(self):
         # Test if section key is "2" out of the mocking data.
         self.assertEqual(get_music_section(
             self.config['plex']['host'],
-            self.config['plex']['port']), '2')
+            self.config['plex']['port'],
+            self.config['plex']['token']), '2')
 
     @responses.activate
     def test_update_plex(self):
@@ -98,7 +99,8 @@ def test_update_plex(self):
         # Testing status code of the mocking request.
         self.assertEqual(update_plex(
             self.config['plex']['host'],
-            self.config['plex']['port']).status_code, 200)
+            self.config['plex']['port'],
+            self.config['plex']['token']).status_code, 200)
 
 
 def suite():