Add/Remove Watchlist context menu commands

lint

lint

lint

Author: Spacetech

addon.xml

@@ -14,6 +14,16 @@
   </extension>
   <extension point="kodi.context.item">
     <menu id="kodi.core.main">
+      <item library="context_watchlist_add.py">
+        <!-- Add to Plex Watchlist --> 
+        <label>30402</label>
+        <visible>[[[String.IsEqual(ListItem.DBTYPE,movie) | String.IsEqual(ListItem.DBTYPE,tvshow) | String.IsEqual(ListItem.DBTYPE,season) | String.IsEqual(ListItem.DBTYPE,episode)] + !String.IsEmpty(ListItem.DBID)] + !String.IsEqual(ListItem.Property(LISTINGKEY),watchlist)]</visible>
+      </item>
+      <item library="context_watchlist_remove.py">
+        <!-- Remove from Plex Watchlist --> 
+        <label>30403</label>
+        <visible>[[[String.IsEqual(ListItem.DBTYPE,movie) | String.IsEqual(ListItem.DBTYPE,tvshow)] + !String.IsEmpty(ListItem.DBID)] + String.IsEqual(ListItem.Property(LISTINGKEY),watchlist)]</visible>
+      </item>
       <item library="context_extras.py">
         <!-- Extras -->
         <label>30235</label>

context_watchlist_add.py

@@ -0,0 +1,28 @@
+# -*- coding: utf-8 -*-
+"""
+Grabs kodi_id and kodi_type for the Kodi item the context menu was called for
+and sends a request to our main Python instance
+"""
+from urllib.parse import urlencode
+
+from xbmc import sleep
+from xbmcgui import Window
+
+from resources.lib.contextmenu.common import kodi_item_from_listitem
+
+
+def main():
+    kodi_id, kodi_type = kodi_item_from_listitem()
+    args = {
+        'kodi_id': kodi_id,
+        'kodi_type': kodi_type
+    }
+    window = Window(10000)
+    while window.getProperty('plexkodiconnect.command'):
+        sleep(20)
+    window.setProperty('plexkodiconnect.command',
+                       'WATCHLIST_ADD?%s' % urlencode(args))
+
+
+if __name__ == "__main__":
+    main()

context_watchlist_remove.py

@@ -0,0 +1,28 @@
+# -*- coding: utf-8 -*-
+"""
+Grabs kodi_id and kodi_type for the Kodi item the context menu was called for
+and sends a request to our main Python instance
+"""
+from urllib.parse import urlencode
+
+from xbmc import sleep
+from xbmcgui import Window
+
+from resources.lib.contextmenu.common import kodi_item_from_listitem
+
+
+def main():
+    kodi_id, kodi_type = kodi_item_from_listitem()
+    args = {
+        'kodi_id': kodi_id,
+        'kodi_type': kodi_type
+    }
+    window = Window(10000)
+    while window.getProperty('plexkodiconnect.command'):
+        sleep(20)
+    window.setProperty('plexkodiconnect.command',
+                       'WATCHLIST_REMOVE?%s' % urlencode(args))
+
+
+if __name__ == "__main__":
+    main()

resources/language/resource.language.en_gb/strings.po

@@ -409,6 +409,16 @@ msgctxt "#30401"
 msgid "Plex options"
 msgstr ""
 
+# contextmenu entry
+msgctxt "#30402"
+msgid "Add to Plex Watchlist"
+msgstr ""
+
+# contextmenu entry
+msgctxt "#30403"
+msgid "Remove from Plex Watchlist"
+msgstr ""
+
 # contextmenu entry
 msgctxt "#30405"
 msgid "Add to Plex favorites"

resources/lib/entrypoint.py

@@ -262,7 +262,7 @@ def show_listing(xml, plex_type=None, section_id=None, synched=True, key=None):
             all_items = [item for item in all_items if item.section_id == utils.cast(int, section_id)]
 
     all_items = [widgets.generate_item(api) for api in all_items]
-    all_items = [widgets.prepare_listitem(item) for item in all_items]
+    all_items = [widgets.prepare_listitem(item, key) for item in all_items]
     # fill that listing...
     all_items = [widgets.create_listitem(item) for item in all_items]
     xbmcplugin.addDirectoryItems(int(sys.argv[1]), all_items, len(all_items))

resources/lib/service_entry.py

@@ -11,6 +11,7 @@
 from . import kodimonitor
 from . import sync, library_sync
 from . import websocket_client
+from . import plex_db
 from . import plex_companion
 from . import plex_functions as PF
 from . import playback_starter
@@ -19,6 +20,7 @@
 from . import loghandler
 from . import backgroundthread
 from . import skip_plex_markers
+from . import downloadutils
 from .windows import userselect
 
 ###############################################################################
@@ -240,6 +242,53 @@ def toggle_plex_tv(self):
                 # Enable the main loop to continue
                 app.APP.suspend = False
 
+    def watchlist_add(self, raw_params):
+        return self.watchlist_modify('addToWatchlist', raw_params)
+
+    def watchlist_remove(self, raw_params):
+        return self.watchlist_modify('removeFromWatchlist', raw_params)
+
+    def watchlist_modify(self, api_type, raw_params):
+        params = dict(utils.parse_qsl(raw_params))
+        kodi_id = params.get('kodi_id')
+        kodi_type = params.get('kodi_type')
+
+        LOG.info('watchlist_modify %s %s %s', api_type, kodi_id, kodi_type)
+
+        watchlist_plex_guid = None
+
+        with plex_db.PlexDB(lock=False) as plexdb:
+            plex_item = plexdb.item_by_kodi_id(kodi_id, kodi_type)
+            if not plex_item:
+                return False
+
+            plex_guid = plex_item['plex_guid']
+            if not plex_guid:
+                return False
+
+            plex_type = plex_item['plex_type']
+
+            if plex_type == v.PLEX_TYPE_MOVIE or plex_type == v.PLEX_TYPE_SHOW:
+                watchlist_plex_guid = plex_guid
+
+            elif plex_type == v.PLEX_TYPE_SEASON or plex_type == v.PLEX_TYPE_EPISODE:
+                plex_show_item = plexdb.item_by_id(plex_item['show_id'], v.PLEX_TYPE_SHOW)
+                if plex_show_item:
+                    watchlist_plex_guid = plex_show_item['plex_guid']
+
+        if watchlist_plex_guid is None:
+            return False
+
+        # ratingKey query param accepts the last section in the plex_guid
+        watchlist_rating_key = watchlist_plex_guid.split('/')[-1]
+
+        downloadutils.DownloadUtils().downloadUrl('https://discover.provider.plex.tv/actions/%s?ratingKey=%s' % (api_type, watchlist_rating_key),
+            action_type = 'PUT',
+            authenticate=False,
+            headerOptions={'X-Plex-Token': utils.window('plex_token')})
+
+        xbmc.executebuiltin('UpdateLibrary(video)')
+
     def authenticate(self):
         """
         Authenticate the current user or prompt to log-in
@@ -473,6 +522,12 @@ def ServiceEntryPoint(self):
                     task = playback_starter.PlaybackTask(
                         'dummy?mode=context_menu&%s'
                         % plex_command.replace('CONTEXT_menu?', ''))
+                elif plex_command.startswith('WATCHLIST_ADD?'):
+                    task = backgroundthread.FunctionAsTask(
+                        self.watchlist_add, None, plex_command.replace('WATCHLIST_ADD?', ''))
+                elif plex_command.startswith('WATCHLIST_REMOVE?'):
+                    task = backgroundthread.FunctionAsTask(
+                        self.watchlist_remove, None, plex_command.replace('WATCHLIST_REMOVE?', ''))
                 elif plex_command == 'choose_pms_server':
                     task = backgroundthread.FunctionAsTask(
                         self.choose_pms_server, None)

resources/lib/widgets.py

@@ -249,7 +249,7 @@ def _generate_content(api):
     return item
 
 
-def prepare_listitem(item):
+def prepare_listitem(item, listing_key = None):
     """helper to convert kodi output from json api to compatible format for
     listitems"""
     try:
@@ -316,6 +316,9 @@ def prepare_listitem(item):
         properties["type"] = item["type"]
         properties["path"] = item.get("file")
 
+        if listing_key is not None:
+            properties["LISTINGKEY"] = listing_key
+
         # cast
         list_cast = []
         list_castandrole = []