# -*- coding: utf-8 -*- import logging from urllib import urlencode from ast import literal_eval from urlparse import urlparse, parse_qsl import re from copy import deepcopy import downloadutils from utils import settings ############################################################################### log = logging.getLogger("PLEX."+__name__) addonName = 'PlexKodiConnect' ############################################################################### # Multiply Plex time by this factor to receive Kodi time PLEX_TO_KODI_TIMEFACTOR = 1.0 / 1000.0 # Possible output of Kodi's ListItem.DBTYPE for all video items KODI_VIDEOTYPES = ( 'video', 'movie', 'set', 'tvshow', 'season', 'episode', 'musicvideo' ) # Possible output of Kodi's ListItem.DBTYPE for all audio items KODI_AUDIOTYPES = ( 'music', 'song', 'album', 'artist' ) ITEMTYPE_FROM_PLEXTYPE = { 'movie': 'Movies', 'season': 'TVShows', 'episode': 'TVShows', 'show': 'TVShows', 'artist': 'Music', 'album': 'Music', 'track': 'Music', 'song': 'Music' } KODITYPE_FROM_PLEXTYPE = { 'movie': 'movie', 'episode': 'episode', 'track': 'song', 'artist': 'artist', 'album': 'album', 'XXXXXX': 'musicvideo', 'XXXXXXX': 'genre' } REMAP_TYPE_FROM_PLEXTYPE = { 'movie': 'movie', 'show': 'tv', 'season': 'tv', 'episode': 'tv', 'artist': 'music', 'album': 'music', 'song': 'music', 'track': 'music', 'clip': 'clip', 'photo': 'photo' } def ConvertPlexToKodiTime(plexTime): """ Converts Plextime to Koditime. Returns an int (in seconds). """ if plexTime is None: return None return int(float(plexTime) * PLEX_TO_KODI_TIMEFACTOR) def GetPlexKeyNumber(plexKey): """ Deconstructs e.g. '/library/metadata/xxxx' to the tuple ('library/metadata', 'xxxx') Returns ('','') if nothing is found """ regex = re.compile(r'''/(.+)/(\d+)$''') try: result = regex.findall(plexKey)[0] except IndexError: result = ('', '') return result def ParseContainerKey(containerKey): """ Parses e.g. /playQueues/3045?own=1&repeat=0&window=200 to: 'playQueues', '3045', {'window': '200', 'own': '1', 'repeat': '0'} Output hence: library, key, query (str, str, dict) """ result = urlparse(containerKey) library, key = GetPlexKeyNumber(result.path) query = dict(parse_qsl(result.query)) return library, key, query def LiteralEval(string): """ Turns a string e.g. in a dict, safely :-) """ return literal_eval(string) def GetMethodFromPlexType(plexType): methods = { 'movie': 'add_update', 'episode': 'add_updateEpisode', 'show': 'add_update', 'season': 'add_updateSeason', 'track': 'add_updateSong', 'album': 'add_updateAlbum', 'artist': 'add_updateArtist' } return methods[plexType] def XbmcItemtypes(): return ['photo', 'video', 'audio'] def PlexItemtypes(): return ['photo', 'video', 'audio'] def PlexLibraryItemtypes(): return ['movie', 'show'] # later add: 'artist', 'photo' def EmbyItemtypes(): return ['Movie', 'Series', 'Season', 'Episode'] def SelectStreams(url, args): """ Does a PUT request to tell the PMS what audio and subtitle streams we have chosen. """ downloadutils.DownloadUtils().downloadUrl( url + '?' + urlencode(args), action_type='PUT') def GetPlayQueue(playQueueID): """ Fetches the PMS playqueue with the playQueueID as an XML Returns None if something went wrong """ url = "{server}/playQueues/%s" % playQueueID args = {'Accept': 'application/xml'} xml = downloadutils.DownloadUtils().downloadUrl(url, headerOptions=args) try: xml.attrib['playQueueID'] except (AttributeError, KeyError): return None return xml def GetPlexMetadata(key): """ Returns raw API metadata for key as an etree XML. Can be called with either Plex key '/library/metadata/xxxx'metadata OR with the digits 'xxxx' only. Returns None or 401 if something went wrong """ key = str(key) if '/library/metadata/' in key: url = "{server}" + key else: url = "{server}/library/metadata/" + key arguments = { 'checkFiles': 0, 'includeExtras': 1, # Trailers and Extras => Extras 'includeReviews': 1, 'includeRelated': 0, # Similar movies => Video -> Related # 'includeRelatedCount': 0, # 'includeOnDeck': 1, # 'includeChapters': 1, # 'includePopularLeaves': 1, # 'includeConcerts': 1 } url = url + '?' + urlencode(arguments) xml = downloadutils.DownloadUtils().downloadUrl(url) if xml == 401: # Either unauthorized (taken care of by doUtils) or PMS under strain return 401 # Did we receive a valid XML? try: xml.attrib # Nope we did not receive a valid XML except AttributeError: log.error("Error retrieving metadata for %s" % url) xml = None return xml def GetAllPlexChildren(key, containerSize=None): """ Returns a list (raw xml API dump) of all Plex children for the key. (e.g. /library/metadata/194853/children pointing to a season) Input: key Key to a Plex item, e.g. 12345 """ url = "{server}/library/metadata/%s/children?" % key return DownloadChunks(url, containerSize) def GetPlexSectionResults(viewId, args=None, containerSize=None): """ Returns a list (XML API dump) of all Plex items in the Plex section with key = viewId. Input: args: optional dict to be urlencoded Returns None if something went wrong """ url = "{server}/library/sections/%s/all?" % viewId if args: url += urlencode(args) + '&' return DownloadChunks(url, containerSize) def DownloadChunks(url, containerSize): """ Downloads PMS url in chunks of containerSize (int). If containerSize is None: ONE xml is fetched directly url MUST end with '?' (if no other url encoded args are present) or '&' Returns a stitched-together xml or None. """ if containerSize is None: # Get rid of '?' or '&' at the end of url xml = downloadutils.DownloadUtils().downloadUrl(url[:-1]) if xml == 401: return 401 try: xml.attrib except AttributeError: # Nope, not an XML, abort log.error("Error getting url %s" % url[:-1]) return None else: return xml xml = None pos = 0 errorCounter = 0 while errorCounter < 10: args = { 'X-Plex-Container-Size': containerSize, 'X-Plex-Container-Start': pos } xmlpart = downloadutils.DownloadUtils().downloadUrl( url + urlencode(args)) # If something went wrong - skip in the hope that it works next time try: xmlpart.attrib except AttributeError: log.error('Error while downloading chunks: %s' % (url + urlencode(args))) pos += containerSize errorCounter += 1 continue # Very first run: starting xml (to retain data in xml's root!) if xml is None: xml = deepcopy(xmlpart) if len(xmlpart) < containerSize: break else: pos += containerSize continue # Build answer xml - containing the entire library for child in xmlpart: xml.append(child) # Done as soon as we don't receive a full complement of items if len(xmlpart) < containerSize: break pos += containerSize if errorCounter == 10: log.error('Fatal error while downloading chunks for %s' % url) return None return xml def GetAllPlexLeaves(viewId, lastViewedAt=None, updatedAt=None, containerSize=None): """ Returns a list (raw XML API dump) of all Plex subitems for the key. (e.g. /library/sections/2/allLeaves pointing to all TV shows) Input: viewId Id of Plex library, e.g. '2' lastViewedAt Unix timestamp; only retrieves PMS items viewed since that point of time until now. updatedAt Unix timestamp; only retrieves PMS items updated by the PMS since that point of time until now. containerSize Number of items simultaneously fetched from PMS If lastViewedAt and updatedAt=None, ALL PMS items are returned. Warning: lastViewedAt and updatedAt are combined with AND by the PMS! Relevant "master time": PMS server. I guess this COULD lead to problems, e.g. when server and client are in different time zones. """ args = [] url = "{server}/library/sections/%s/allLeaves" % viewId if lastViewedAt: args.append('lastViewedAt>=%s' % lastViewedAt) if updatedAt: args.append('updatedAt>=%s' % updatedAt) if args: url += '?' + '&'.join(args) + '&' else: url += '?' return DownloadChunks(url, containerSize) def GetPlexOnDeck(viewId, containerSize=None): """ """ url = "{server}/library/sections/%s/onDeck?" % viewId return DownloadChunks(url, containerSize) def GetPlexCollections(mediatype): """ Input: mediatype String or list of strings with possible values 'movie', 'show', 'artist', 'photo' Output: List with an entry of the form: { 'name': xxx Plex title for the media section 'type': xxx Plex type: 'movie', 'show', 'artist', 'photo' 'id': xxx Plex unique key for the section (1, 2, 3...) 'uuid': xxx Other unique Plex key, e.g. 74aec9f2-a312-4723-9436-de2ea43843c1 } Returns an empty list if nothing is found. """ collections = [] url = "{server}/library/sections" xml = downloadutils.DownloadUtils().downloadUrl(url) try: xml.attrib except AttributeError: log.error('Could not download PMS sections for %s' % url) return {} for item in xml: contentType = item['type'] if contentType in mediatype: name = item['title'] contentId = item['key'] uuid = item['uuid'] collections.append({ 'name': name, 'type': contentType, 'id': str(contentId), 'uuid': uuid }) return collections def GetPlexPlaylist(itemid, librarySectionUUID, mediatype='movie'): """ Returns raw API metadata XML dump for a playlist with e.g. trailers. """ trailerNumber = settings('trailerNumber') if not trailerNumber: trailerNumber = '3' url = "{server}/playQueues" args = { 'type': mediatype, 'uri': ('library://' + librarySectionUUID + '/item/%2Flibrary%2Fmetadata%2F' + itemid), 'includeChapters': '1', 'extrasPrefixCount': trailerNumber, 'shuffle': '0', 'repeat': '0' } xml = downloadutils.DownloadUtils().downloadUrl( url + '?' + urlencode(args), action_type="POST") try: xml[0].tag except (IndexError, TypeError, AttributeError): log.error("Error retrieving metadata for %s" % url) return None return xml def getPlexRepeat(kodiRepeat): plexRepeat = { 'off': '0', 'one': '1', 'all': '2' # does this work?!? } return plexRepeat.get(kodiRepeat) def PMSHttpsEnabled(url): """ Returns True if the PMS can talk https, False otherwise. None if error occured, e.g. the connection timed out Call with e.g. url='192.168.0.1:32400' (NO http/https) This is done by GET /identity (returns an error if https is enabled and we are trying to use http) Prefers HTTPS over HTTP """ doUtils = downloadutils.DownloadUtils().downloadUrl res = doUtils('https://%s/identity' % url, authenticate=False, verifySSL=False) try: res.attrib except AttributeError: # Might have SSL deactivated. Try with http res = doUtils('http://%s/identity' % url, authenticate=False, verifySSL=False) try: res.attrib except AttributeError: log.error("Could not contact PMS %s" % url) return None else: # Received a valid XML. Server wants to talk HTTP return False else: # Received a valid XML. Server wants to talk HTTPS return True def GetMachineIdentifier(url): """ Returns the unique PMS machine identifier of url Returns None if something went wrong """ xml = downloadutils.DownloadUtils().downloadUrl('%s/identity' % url, authenticate=False, verifySSL=False, timeout=4) try: machineIdentifier = xml.attrib['machineIdentifier'] except (AttributeError, KeyError): log.error('Could not get the PMS machineIdentifier for %s' % url) return None log.debug('Found machineIdentifier %s for the PMS %s' % (machineIdentifier, url)) return machineIdentifier def GetPMSStatus(token): """ token: Needs to be authorized with a master Plex token (not a managed user token)! Calls /status/sessions on currently active PMS. Returns a dict with: 'sessionKey': { 'userId': Plex ID of the user (if applicable, otherwise '') 'username': Plex name (if applicable, otherwise '') 'ratingKey': Unique Plex id of item being played } or an empty dict. """ answer = {} xml = downloadutils.DownloadUtils().downloadUrl( '{server}/status/sessions', headerOptions={'X-Plex-Token': token}) try: xml.attrib except AttributeError: return answer for item in xml: ratingKey = item.attrib.get('ratingKey') sessionKey = item.attrib.get('sessionKey') userId = item.find('User') username = '' if userId is not None: username = userId.attrib.get('title', '') userId = userId.attrib.get('id', '') else: userId = '' answer[sessionKey] = { 'userId': userId, 'username': username, 'ratingKey': ratingKey } return answer def scrobble(ratingKey, state): """ Tells the PMS to set an item's watched state to state="watched" or state="unwatched" """ args = { 'key': ratingKey, 'identifier': 'com.plexapp.plugins.library' } if state == "watched": url = "{server}/:/scrobble?" + urlencode(args) elif state == "unwatched": url = "{server}/:/unscrobble?" + urlencode(args) else: return downloadutils.DownloadUtils().downloadUrl(url) log.info("Toggled watched state for Plex item %s" % ratingKey) def delete_item_from_pms(plexid): """ Deletes the item plexid from the Plex Media Server (and the harddrive!). Do make sure that the currently logged in user has the credentials Returns True if successful, False otherwise """ if downloadutils.DownloadUtils().downloadUrl( '{server}/library/metadata/%s' % plexid, action_type="DELETE") is True: log.info('Successfully deleted Plex id %s from the PMS' % plexid) return True else: log.error('Could not delete Plex id %s from the PMS' % plexid) return False def get_PMS_settings(url, token): """ Retrieve the PMS' settings via /:/ Call with url: scheme://ip:port """ return downloadutils.DownloadUtils().downloadUrl( '%s/:/prefs' % url, authenticate=False, verifySSL=False, headerOptions={'X-Plex-Token': token} if token else None)