Rewire the set-up of audio and subtitle streams, esp. before starting a transcoding session

resources/lib/playback.py

@@ -12,16 +12,10 @@ import xbmc
 
 from .plex_api import API
 from .plex_db import PlexDB
-from . import plex_functions as PF
-from . import utils
 from .kodi_db import KodiVideoDB
-from . import playlist_func as PL
+from . import plex_functions as PF, playlist_func as PL, playqueue as PQ
-from . import playqueue as PQ
+from . import json_rpc as js, variables as v, utils, transfer
-from . import json_rpc as js
+from . import playback_decision, app
-from . import transfer
-from .playback_decision import set_playurl, audio_subtitle_prefs
-from . import variables as v
-from . import app
 
 ###############################################################################
 LOG = getLogger('PLEX.playback')
@@ -457,21 +451,20 @@ def _conclude_playback(playqueue, pos):
         _ensure_resolve()
         return
     api.part = item.part or 0
-    listitem = api.listitem(listitem=transfer.PKCListItem, resume=False)
+    playback_decision.set_pkc_playmethod(api, item)
-    set_playurl(api, item)
+    if not playback_decision.audio_subtitle_prefs(api, item):
-    if not item.file:
+        LOG.info('Did not set audio subtitle prefs, aborting silently')
-        LOG.debug('Did not get a playurl, aborting playback silently')
         _ensure_resolve()
         return
+    playback_decision.set_playurl(api, item)
+    if not item.file:
+        LOG.info('Did not get a playurl, aborting playback silently')
+        _ensure_resolve()
+        return
+    listitem = api.listitem(listitem=transfer.PKCListItem, resume=False)
     listitem.setPath(item.file.encode('utf-8'))
-    if item.playmethod == v.PLAYBACK_METHOD_DIRECT_PLAY:
+    if item.playmethod != v.PLAYBACK_METHOD_DIRECT_PATH:
         listitem.setSubtitles(api.cache_external_subs())
-    elif item.playmethod in (v.PLAYBACK_METHOD_DIRECT_STREAM,
-                             v.PLAYBACK_METHOD_TRANSCODE):
-        audio_subtitle_prefs(api, listitem)
-        # Need to hit the PMS api again in order to get the selected
-        # burn-in subtitles set-up correctly
-        set_playurl(api, item)
     transfer.send(listitem)
     LOG.debug('Done concluding playback')
 

resources/lib/playback_decision.py

@@ -18,18 +18,20 @@ DIRECT_PLAY_OK = 1000
 CONVERSION_OK = 1001  # PMS can either direct stream or transcode
 
 
-def set_playurl(api, item):
+def set_pkc_playmethod(api, item):
     item.playmethod = int(utils.settings('playType'))
     LOG.info('User chose playback method %s in PKC settings',
              v.EXPLICIT_PLAYBACK_METHOD[item.playmethod])
     _initial_best_playback_method(api, item)
     LOG.info('PKC decided on playback method %s',
              v.EXPLICIT_PLAYBACK_METHOD[item.playmethod])
+
+
+def set_playurl(api, item):
+    try:
         if item.playmethod == v.PLAYBACK_METHOD_DIRECT_PATH:
             # No need to ask the PMS whether we can play - we circumvent
             # the PMS entirely
-        LOG.info('The playurl for %s is: %s',
-                 v.EXPLICIT_PLAYBACK_METHOD[item.playmethod], item.file)
             return
         LOG.info('Lets ask the PMS next')
         try:
@@ -45,6 +47,7 @@ def set_playurl(api, item):
         else:
             item.file = api.transcode_video_path(item.playmethod,
                                                  quality=item.quality)
+    finally:
         LOG.info('The playurl for %s is: %s',
                  v.EXPLICIT_PLAYBACK_METHOD[item.playmethod], item.file)
 
@@ -316,12 +319,16 @@ def _getH265():
     return H265[utils.settings('transcodeH265')]
 
 
-def audio_subtitle_prefs(api, listitem):
+def audio_subtitle_prefs(api, item):
     """
-    For transcoding only
+    Sets the stage for transcoding, letting the user potentially choose both
+    audio and subtitle streams; subtitle streams to burn-into the video file.
 
-    Called at the very beginning of play; used to change audio and subtitle
+    Uses a PUT request to the PMS, simulating e.g. the user using Plex Web,
-    stream by a PUT request to the PMS
+    choosing a different stream in the video's metadata and THEN initiating
+    playback.
+
+    Returns None if user cancelled or we need to abort, True otherwise
     """
     # Set media and part where we're at
     if (api.mediastream is None and
@@ -334,13 +341,21 @@ def audio_subtitle_prefs(api, listitem):
                   api.mediastream, api.part)
         return
     part_id = mediastreams.attrib['id']
+    if item.playmethod != v.PLAYBACK_METHOD_TRANSCODE:
+        LOG.debug('Telling PMS we are not burning in any subtitles')
+        args = {
+            'subtitleStreamID': 0,
+            'allParts': 1
+        }
+        DU().downloadUrl('{server}/library/parts/%s' % part_id,
+                         action_type='PUT',
+                         parameters=args)
+        return True
     audio_streams_list = []
     audio_streams = []
     subtitle_streams_list = []
     # "Don't burn-in any subtitle"
     subtitle_streams = ['1 %s' % utils.lang(39706)]
-    downloadable_streams = []
-    download_subs = []
     # selectAudioIndex = ""
     select_subs_index = ""
     audio_numb = 0
@@ -372,19 +387,11 @@ def audio_subtitle_prefs(api, listitem):
 
         # Subtitles
         elif typus == "3":
-            downloadable = stream.get('key')
+            if stream.get('key'):
-            if downloadable:
+                # Subtitle can and will be downloaded - don't let user choose
-                # Download the subtitle to Kodi - the user will need to
+                # this subtitle to burn-in
-                # manually select the subtitle on the Kodi side
+                continue
-                # Hence do NOT show dialog for this sub
+            # Subtitle is available within the video file
-                path = api.download_external_subtitles(
-                    '{{server}}{}'.format(stream.get('key')),
-                    stream.get('displayTitle'),
-                    stream.get('codec'))
-                if path:
-                    downloadable_streams.append(index)
-                    download_subs.append(path.encode('utf-8'))
-            else:
             # Burn in the subtitle, if user chooses to do so
             default = stream.get('default')
             forced = stream.get('forced')
@@ -406,8 +413,9 @@ def audio_subtitle_prefs(api, listitem):
 
     if audio_numb > 1:
         resp = utils.dialog('select', utils.lang(33013), audio_streams)
-        if resp > -1:
+        if resp == -1:
-            # User selected some audio track
+            LOG.info('User aborted dialog to select audio stream')
+            return
         args = {
             'audioStreamID': audio_streams_list[resp],
             'allParts': 1
@@ -416,22 +424,21 @@ def audio_subtitle_prefs(api, listitem):
                          action_type='PUT',
                          parameters=args)
 
-    LOG.debug('Adding downloadable subtitles: %s', download_subs)
-    # Enable Kodi to switch autonomously to downloadable subtitles
-    if download_subs:
-        listitem.setSubtitles(download_subs)
     select_subs_index = ''
     if sub_num == 1:
         # Note: we DO need to tell the PMS that we DONT want any sub
         # Otherwise, the PMS might pick-up the last one
-        LOG.debug('No subtitles to burn-in')
+        LOG.info('No subtitles to burn-in')
     else:
         resp = utils.dialog('select', utils.lang(33014), subtitle_streams)
-        if resp < 1:
+        if resp == -1:
+            LOG.info('User aborted dialog to select subtitle stream')
+            return
+        elif resp == 0:
             # User did not select a subtitle or backed out of the dialog
-            LOG.debug('User chose to not burn-in any subtitles')
+            LOG.info('User chose to not burn-in any subtitles')
         else:
-            LOG.debug('User chose to burn-in subtitle %s: %s',
+            LOG.info('User chose to burn-in subtitle %s: %s',
                       select_subs_index,
                       subtitle_streams[resp].decode('utf-8'))
             select_subs_index = subtitle_streams_list[resp - 1]
@@ -443,3 +450,4 @@ def audio_subtitle_prefs(api, listitem):
     DU().downloadUrl('{server}/library/parts/%s' % part_id,
                      action_type='PUT',
                      parameters=args)
+    return True

resources/lib/plex_functions.py

@@ -1085,13 +1085,15 @@ def transcoding_arguments(path, media, part, playmethod, args=None):
         'protocol': 'hls',   # seen in the wild: 'http', 'dash', 'http', 'hls'
         'session': v.PKC_MACHINE_IDENTIFIER,  # TODO: create new unique id
         'fastSeek': 1,
-        # none, embedded, sidecar
-        # Essentially indicating what you want to do with subtitles and state
-        # you aren’t want it to burn them into the video (requires transcoding)
-        # 'subtitles': 'none',
-        'subtitleSize': utils.settings('subtitleSize'),
         'copyts': 1
     }
+    if playmethod != v.PLAYBACK_METHOD_TRANSCODE:
+        # Essentially indicating what you want to do with subtitles and state
+        # you aren’t want it to burn them into the video (requires transcoding)
+        # none, embedded, sidecar
+        args['subtitles'] = 'none'
+    else:
+        args['subtitleSize'] = utils.settings('subtitleSize')
     if args:
         arguments.update(args)
     return arguments