From be7ecb95dbe076ab61ae169a4ef786b68e025046 Mon Sep 17 00:00:00 2001 From: tomkat83 Date: Wed, 8 Mar 2017 16:21:00 +0100 Subject: [PATCH] Specify Kodi sort methods --- default.py | 8 +- resources/lib/entrypoint.py | 171 +++++++++++++++++++----------------- resources/lib/variables.py | 92 ++++++++++++++++++- 3 files changed, 181 insertions(+), 90 deletions(-) diff --git a/default.py b/default.py index 321514b7..1887faea 100644 --- a/default.py +++ b/default.py @@ -86,13 +86,9 @@ class Main(): elif mode == 'Plex_Node': entrypoint.Plex_Node(itemid, params.get('viewOffset')) - elif mode == 'browse_plex_folder': - entrypoint.browse_plex_folder(itemid) - elif mode == 'browseplex': - entrypoint.BrowsePlexContent(itemid, - params.get('type'), - params.get('folderid')) + entrypoint.browse_plex(key=params.get('key'), + plex_section_id=params.get('id')) elif mode == 'getsubfolders': entrypoint.GetSubFolders(itemid) diff --git a/resources/lib/entrypoint.py b/resources/lib/entrypoint.py index e3a97abc..f693e49e 100644 --- a/resources/lib/entrypoint.py +++ b/resources/lib/entrypoint.py @@ -644,83 +644,6 @@ def RunLibScan(mode): window('plex_runLibScan', value='full') -def BrowsePlexContent(viewid, mediatype="", folderid=""): - """ - Browse Plex Photos: - viewid: PMS name of the library - mediatype: mediatype, 'photos' - nodetype: e.g. 'ondeck' (TBD!!) - """ - log.debug("BrowsePlexContent called with viewid: %s, mediatype: " - "%s, folderid: %s" % (viewid, mediatype, folderid)) - - if not folderid: - # Top-level navigation, so get the content of this section - # Get all sections - xml = GetPlexSectionResults( - viewid, - containerSize=int(settings('limitindex'))) - try: - xml.attrib - except AttributeError: - log.error("Error download section %s" % viewid) - return xbmcplugin.endOfDirectory(HANDLE, False) - else: - # folderid was passed so we can directly access the folder - xml = downloadutils.DownloadUtils().downloadUrl( - "{server}%s" % folderid) - try: - xml.attrib - except AttributeError: - log.error("Error downloading %s" % folderid) - return xbmcplugin.endOfDirectory(HANDLE, False) - - # Set the folder's name - xbmcplugin.setPluginCategory(HANDLE, - xml.attrib.get('librarySectionTitle')) - - # set the correct params for the content type - if mediatype == "photos": - xbmcplugin.setContent(HANDLE, 'photos') - - # process the listing - for item in xml: - api = API(item) - if item.tag == 'Directory': - li = ListItem(item.attrib.get('title', 'Missing title')) - # for folders we add an additional browse request, passing the - # folderId - li.setProperty('IsFolder', 'true') - li.setProperty('IsPlayable', 'false') - path = "%s?id=%s&mode=browseplex&type=%s&folderid=%s" \ - % (ARGV_0, viewid, mediatype, api.getKey()) - api.set_listitem_artwork(li) - xbmcplugin.addDirectoryItem(handle=HANDLE, - url=path, - listitem=li, - isFolder=True) - else: - li = api.CreateListItemFromPlexItem() - api.set_listitem_artwork(li) - xbmcplugin.addDirectoryItem( - handle=HANDLE, - url=li.getProperty("path"), - listitem=li) - - xbmcplugin.addSortMethod(HANDLE, - xbmcplugin.SORT_METHOD_VIDEO_TITLE) - xbmcplugin.addSortMethod(HANDLE, - xbmcplugin.SORT_METHOD_DATE) - xbmcplugin.addSortMethod(HANDLE, - xbmcplugin.SORT_METHOD_VIDEO_RATING) - xbmcplugin.addSortMethod(HANDLE, - xbmcplugin.SORT_METHOD_VIDEO_RUNTIME) - - xbmcplugin.endOfDirectory( - handle=HANDLE, - cacheToDisc=settings('enableTextureCache') == 'true') - - def getOnDeck(viewid, mediatype, tagname, limit): """ Retrieves Plex On Deck items, currently only for TV shows @@ -913,6 +836,8 @@ def channels(): log.info('Displaying Plex Channels') xbmcplugin.setContent(HANDLE, 'files') + for method in v.SORT_METHODS_DIRECTORY: + xbmcplugin.addSortMethod(HANDLE, getattr(xbmcplugin, method)) for item in xml: __build_folder(item) xbmcplugin.endOfDirectory( @@ -920,22 +845,102 @@ def channels(): cacheToDisc=settings('enableTextureCache') == 'true') -def browse_plex_folder(key): +def browse_plex(key=None, plex_section_id=None): """ - Lists the content of a Plex folder, e.g. channels + Lists the content of a Plex folder, e.g. channels. Either pass in key (to + be used directly for PMS url {server}) or the plex_section_id """ - xml = downloadutils.DownloadUtils().downloadUrl('{server}%s' % key) + if key: + xml = downloadutils.DownloadUtils().downloadUrl('{server}%s' % key) + else: + xml = GetPlexSectionResults( + plex_section_id, + containerSize=int(settings('limitindex'))) try: xml[0].attrib except (ValueError, AttributeError, IndexError): log.error('Could not browse to %s' % key) return xbmcplugin.endOfDirectory(HANDLE, False) - xbmcplugin.setContent(HANDLE, 'files') + + directory = False + photos = False + movies = False + clips = False + tvshows = False + episodes = False + songs = False + artists = False + albums = False + musicvideos = False for item in xml: + typus = item.attrib.get('type') if item.tag == 'Directory': + directory = True __build_folder(item) else: __build_item(item) + if typus == v.PLEX_TYPE_PHOTO: + photos = True + elif typus == v.PLEX_TYPE_MOVIE: + movies = True + elif typus == v.PLEX_TYPE_CLIP: + clips = True + elif typus in (v.PLEX_TYPE_SHOW, v.PLEX_TYPE_SEASON): + tvshows = True + elif typus == v.PLEX_TYPE_EPISODE: + episodes = True + elif typus == v.PLEX_TYPE_SONG: + songs = True + elif typus == v.PLEX_TYPE_ARTIST: + artists = True + elif typus == v.PLEX_TYPE_ALBUM: + albums = True + elif typus == v.PLEX_TYPE_MUSICVIDEO: + musicvideos = True + + # Set the correct content type + if directory is True: + xbmcplugin.setContent(HANDLE, 'files') + sort_methods = v.SORT_METHODS_DIRECTORY + elif photos is True: + # E.g. Plex foto sections with videos (both!!) + xbmcplugin.setContent(HANDLE, 'files') + sort_methods = v.SORT_METHODS_PHOTOS + elif clips is True: + xbmcplugin.setContent(HANDLE, 'movies') + sort_methods = v.SORT_METHODS_CLIPS + elif movies is True: + xbmcplugin.setContent(HANDLE, 'movies') + sort_methods = v.SORT_METHODS_MOVIES + elif tvshows is True: + xbmcplugin.setContent(HANDLE, 'tvshows') + sort_methods = v.SORT_METHOD_TVSHOWS + elif episodes is True: + xbmcplugin.setContent(HANDLE, 'episodes') + sort_methods = v.SORT_METHODS_EPISODES + elif songs is True: + xbmcplugin.setContent(HANDLE, 'songs') + sort_methods = v.SORT_METHODS_SONGS + elif artists is True: + xbmcplugin.setContent(HANDLE, 'artists') + sort_methods = v.SORT_METHODS_ARTISTS + elif albums is True: + xbmcplugin.setContent(HANDLE, 'albums') + sort_methods = v.SORT_METHODS_ALBUMS + elif musicvideos is True: + xbmcplugin.setContent(HANDLE, 'musicvideos') + sort_methods = v.SORT_METHODS_MOVIES + else: + xbmcplugin.setContent(HANDLE, 'files') + sort_methods = v.SORT_METHODS_DIRECTORY + + for method in sort_methods: + xbmcplugin.addSortMethod(HANDLE, getattr(xbmcplugin, method)) + + # Set the Kodi title for this view + title = xml.attrib.get('librarySectionTitle', xml.attrib.get('title1')) + xbmcplugin.setPluginCategory(HANDLE, title) + xbmcplugin.endOfDirectory( handle=HANDLE, cacheToDisc=settings('enableTextureCache') == 'true') @@ -944,8 +949,8 @@ def browse_plex_folder(key): def __build_folder(xml_element): url = "plugin://%s/" % v.ADDON_ID params = { - 'mode': "browse_plex_folder", - 'id': xml_element.attrib.get('key') + 'mode': "browseplex", + 'key': xml_element.attrib.get('key') } listitem = ListItem(xml_element.attrib.get('title')) listitem.setArt({'thumb': xml_element.attrib.get('thumb'), diff --git a/resources/lib/variables.py b/resources/lib/variables.py index 1ed3fa9c..f28af541 100644 --- a/resources/lib/variables.py +++ b/resources/lib/variables.py @@ -113,6 +113,7 @@ PLEX_TYPE_AUDIO = 'music' PLEX_TYPE_SONG = 'track' PLEX_TYPE_ALBUM = 'album' PLEX_TYPE_ARTIST = 'artist' +PLEX_TYPE_MUSICVIDEO = 'musicvideo' PLEX_TYPE_PHOTO = 'photo' @@ -131,6 +132,7 @@ KODI_TYPE_AUDIO = 'audio' KODI_TYPE_SONG = 'song' KODI_TYPE_ALBUM = 'album' KODI_TYPE_ARTIST = 'artist' +KODI_TYPE_MUSICVIDEO = 'musicvideo' KODI_TYPE_PHOTO = 'photo' @@ -195,7 +197,8 @@ KODI_PLAYLIST_TYPE_FROM_PLEX_TYPE = { PLEX_TYPE_ARTIST: KODI_TYPE_AUDIO, PLEX_TYPE_ALBUM: KODI_TYPE_AUDIO, PLEX_TYPE_SONG: KODI_TYPE_AUDIO, - PLEX_TYPE_AUDIO: KODI_TYPE_AUDIO + PLEX_TYPE_AUDIO: KODI_TYPE_AUDIO, + PLEX_TYPE_PHOTO: 'picture' } @@ -266,3 +269,90 @@ ALEXA_TO_COMPANION = { 'queryContainerKey': 'containerKey', 'queryToken': 'token', } + +# Kodi sort methods for xbmcplugin.addSortMethod() +SORT_METHODS_DIRECTORY = ( + 'SORT_METHOD_UNSORTED', # sorted as returned from Plex + 'SORT_METHOD_LABEL', +) + +SORT_METHODS_PHOTOS = ( + 'SORT_METHOD_UNSORTED', + 'SORT_METHOD_LABEL', + 'SORT_METHOD_DATE', + 'SORT_METHOD_DATEADDED', +) + +SORT_METHODS_CLIPS = ( + 'SORT_METHOD_UNSORTED', + 'SORT_METHOD_TITLE', + 'SORT_METHOD_DURATION', +) + +SORT_METHODS_MOVIES = ( + 'SORT_METHOD_UNSORTED', + 'SORT_METHOD_TITLE', + 'SORT_METHOD_DURATION', + 'SORT_METHOD_VIDEO_RATING', + 'SORT_METHOD_VIDEO_USER_RATING', + 'SORT_METHOD_MPAA_RATING', + 'SORT_METHOD_COUNTRY', + 'SORT_METHOD_STUDIO', + 'SORT_METHOD_GENRE', +) + +SORT_METHOD_TVSHOWS = ( + 'SORT_METHOD_UNSORTED', + 'SORT_METHOD_TITLE', + 'SORT_METHOD_VIDEO_RATING', + 'SORT_METHOD_VIDEO_USER_RATING', + 'SORT_METHOD_MPAA_RATING', + 'SORT_METHOD_COUNTRY', + 'SORT_METHOD_GENRE', +) + +SORT_METHODS_EPISODES = ( + 'SORT_METHOD_UNSORTED', + 'SORT_METHOD_TITLE', + 'SORT_METHOD_DURATION', + 'SORT_METHOD_VIDEO_RATING', + 'SORT_METHOD_VIDEO_USER_RATING', + 'SORT_METHOD_EPISODE', + 'SORT_METHOD_VIDEO_SORT_TITLE', + 'SORT_METHOD_MPAA_RATING', + 'SORT_METHOD_COUNTRY', + 'SORT_METHOD_STUDIO', + 'SORT_METHOD_GENRE', +) + +SORT_METHODS_SONGS = ( + 'SORT_METHOD_UNSORTED', + 'SORT_METHOD_TITLE', + 'SORT_METHOD_TRACKNUM', + 'SORT_METHOD_DURATION', + 'SORT_METHOD_ARTIST', + 'SORT_METHOD_ARTIST_AND_YEAR', + 'SORT_METHOD_ALBUM', + 'SORT_METHOD_SONG_RATING', + 'SORT_METHOD_SONG_USER_RATING' +) + +SORT_METHODS_ARTISTS = ( + 'SORT_METHOD_UNSORTED', + 'SORT_METHOD_TITLE', + 'SORT_METHOD_TRACKNUM', + 'SORT_METHOD_DURATION', + 'SORT_METHOD_ARTIST', + 'SORT_METHOD_ARTIST_AND_YEAR', + 'SORT_METHOD_ALBUM', +) + +SORT_METHODS_ALBUMS = ( + 'SORT_METHOD_UNSORTED', + 'SORT_METHOD_TITLE', + 'SORT_METHOD_TRACKNUM', + 'SORT_METHOD_DURATION', + 'SORT_METHOD_ARTIST', + 'SORT_METHOD_ARTIST_AND_YEAR', + 'SORT_METHOD_ALBUM', +)