518 lines
15 KiB
Python
518 lines
15 KiB
Python
# -*- coding: utf-8 -*-
|
|
from urllib import urlencode
|
|
from ast import literal_eval
|
|
from urlparse import urlparse, parse_qs
|
|
import re
|
|
from copy import deepcopy
|
|
|
|
import downloadutils
|
|
from utils import logMsg, settings
|
|
|
|
|
|
addonName = 'PlexKodiConnect'
|
|
title = "%s %s" % (addonName, __name__)
|
|
|
|
|
|
def PlexToKodiTimefactor():
|
|
"""
|
|
Kodi measures time in seconds, but Plex in milliseconds
|
|
"""
|
|
return 1.0 / 1000.0
|
|
|
|
|
|
def ConvertPlexToKodiTime(plexTime):
|
|
"""
|
|
Converts Plextime to Koditime. Returns an int (in seconds).
|
|
"""
|
|
if plexTime is None:
|
|
return None
|
|
return int(float(plexTime) * PlexToKodiTimefactor())
|
|
|
|
|
|
def GetItemClassFromType(itemType):
|
|
classes = {
|
|
'movie': 'Movies',
|
|
'season': 'TVShows',
|
|
'episode': 'TVShows',
|
|
'show': 'TVShows',
|
|
'artist': 'Music',
|
|
'album': 'Music',
|
|
'track': 'Music',
|
|
'song': 'Music'
|
|
}
|
|
return classes[itemType]
|
|
|
|
|
|
def GetItemClassFromNumber(itemType):
|
|
classes = {
|
|
1: 'Movies',
|
|
4: 'TVShows',
|
|
}
|
|
return classes[itemType]
|
|
|
|
|
|
def GetKodiTypeFromPlex(plexItemType):
|
|
"""
|
|
As used in playlist.item here: http://kodi.wiki/view/JSON-RPC_API
|
|
"""
|
|
classes = {
|
|
'movie': 'movie',
|
|
'episode': 'episode',
|
|
'track': 'song',
|
|
'artist': 'artist',
|
|
'album': 'album',
|
|
'XXXXXX': 'musicvideo',
|
|
'XXXXXXX': 'genre'
|
|
}
|
|
return classes[plexItemType.lower()]
|
|
|
|
|
|
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 (query as a special dict)
|
|
"""
|
|
result = urlparse(containerKey)
|
|
library, key = GetPlexKeyNumber(result.path)
|
|
query = parse_qs(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), 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 if something went wrong
|
|
"""
|
|
key = str(key)
|
|
if '/library/metadata/' in key:
|
|
url = "{server}" + key
|
|
else:
|
|
url = "{server}/library/metadata/" + key
|
|
arguments = {
|
|
'checkFiles': 1, # No idea
|
|
'includeExtras': 1, # Trailers and Extras => Extras
|
|
# 'includeRelated': 1, # Similar movies => Video -> Related
|
|
# 'includeRelatedCount': 5,
|
|
# 'includeOnDeck': 1,
|
|
'includeChapters': 1,
|
|
'includePopularLeaves': 1,
|
|
'includeConcerts': 1
|
|
}
|
|
url = url + '?' + urlencode(arguments)
|
|
xml = downloadutils.DownloadUtils().downloadUrl(url)
|
|
# Did we receive a valid XML?
|
|
try:
|
|
xml.attrib
|
|
# Nope we did not receive a valid XML
|
|
except AttributeError:
|
|
logMsg(title, "Error retrieving metadata for %s" % url, -1)
|
|
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])
|
|
try:
|
|
xml.attrib
|
|
except AttributeError:
|
|
# Nope, not an XML, abort
|
|
logMsg(title, "Error getting url %s" % url[:-1], -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:
|
|
logMsg(title, 'Error while downloading chunks: %s'
|
|
% (url + urlencode(args)), -1)
|
|
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:
|
|
logMsg(title, 'Fatal error while downloading chunks for %s' % url, -1)
|
|
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:
|
|
logMsg(title, 'Could not download PMS sections for %s' % url, -1)
|
|
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), type="POST")
|
|
try:
|
|
xml[0].tag
|
|
except (IndexError, TypeError, AttributeError):
|
|
logMsg(title, "Error retrieving metadata for %s" % url, -1)
|
|
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:
|
|
# Might have SSL deactivated. Try with http
|
|
res = doUtils('http://%s/identity' % url,
|
|
authenticate=False,
|
|
verifySSL=False)
|
|
try:
|
|
res.attrib
|
|
except:
|
|
logMsg(title, "Could not contact PMS %s" % url, -1)
|
|
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(
|
|
url + '/identity', type="GET")
|
|
try:
|
|
xml.attrib
|
|
except:
|
|
logMsg(title, 'Could not get the PMS machineIdentifier for %s'
|
|
% url, -1)
|
|
return None
|
|
machineIdentifier = xml.attrib.get('machineIdentifier')
|
|
logMsg(title, 'Found machineIdentifier %s for %s'
|
|
% (machineIdentifier, url), 1)
|
|
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',
|
|
type="GET",
|
|
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, type="GET")
|
|
logMsg(title, "Toggled watched state for Plex item %s" % ratingKey, 1)
|