Troplo/PlexKodiConnect
1
0
Fork 0
Code Issues Pull requests Projects Releases Wiki Activity

Move xArgsDeviceInfo to clientinfo

Browse source
This commit is contained in:
tomkat83 2016-04-05 18:23:00 +02:00
parent 9929ab2eb5
commit 12add3d369
3 changed files with 74 additions and 200 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

182
resources/lib/PlexAPI.py
View file

@ -67,21 +67,10 @@ class PlexAPI():
# CONSTANTS # CONSTANTS
# Timeout for POST/GET commands, I guess in seconds # Timeout for POST/GET commands, I guess in seconds
timeout = 10 timeout = 10
# VARIABLES
def __init__(self): def __init__(self):
self.__language__ = xbmcaddon.Addon().getLocalizedString self.__language__ = xbmcaddon.Addon().getLocalizedString
self.g_PMS = {} self.g_PMS = {}
client = clientinfo.ClientInfo()
self.addonId = client.getAddonId()
self.clientId = client.getDeviceId()
self.deviceName = client.getDeviceName()
self.plexversion = client.getVersion()
self.platform = client.getPlatform()
self.userId = utils.window('currUserId')
self.token = utils.window('pms_token')
self.server = utils.window('pms_server')
self.doUtils = downloadutils.DownloadUtils() self.doUtils = downloadutils.DownloadUtils()
def GetPlexLoginFromSettings(self): def GetPlexLoginFromSettings(self):
@ -145,7 +134,8 @@ class PlexAPI():
retrievedPlexLogin, authtoken = self.MyPlexSignIn( retrievedPlexLogin, authtoken = self.MyPlexSignIn(
plexLogin, plexLogin,
plexPassword, plexPassword,
{'X-Plex-Client-Identifier': self.clientId}) {'X-Plex-Client-Identifier':
clientinfo.ClientInfo().getDeviceId()})
self.logMsg("plex.tv username and token: %s, %s" self.logMsg("plex.tv username and token: %s, %s"
% (plexLogin, authtoken), 1) % (plexLogin, authtoken), 1)
if plexLogin == '': if plexLogin == '':
@ -345,7 +335,7 @@ class PlexAPI():
""" """
# Add '/clients' to URL because then an authentication is necessary # Add '/clients' to URL because then an authentication is necessary
# If a plex.tv URL was passed, this does not work. # If a plex.tv URL was passed, this does not work.
header = self.getXArgsDeviceInfo() header = clientinfo.ClientInfo().getXArgsDeviceInfo()
if token: if token:
header['X-Plex-Token'] = token header['X-Plex-Token'] = token
sslverify = utils.settings('sslverify') sslverify = utils.settings('sslverify')
@ -816,148 +806,6 @@ class PlexAPI():
XML = self.getXMLFromPMS(PMS['baseURL'],PMS['path'],PMS['options'],PMS['token']) XML = self.getXMLFromPMS(PMS['baseURL'],PMS['path'],PMS['options'],PMS['token'])
queue.put( (PMS['data'], XML) ) queue.put( (PMS['data'], XML) )
def getXArgsDeviceInfo(self, options={}):
"""
Returns a dictionary that can be used as headers for GET and POST
requests. An authentication option is NOT yet added.
Inputs:
options: dictionary of options that will override the
standard header options otherwise set.
Output:
header dictionary
"""
# Get addon infos
xargs = {
'Accept': '*/*',
'Connection': 'keep-alive',
"Content-Type": "application/x-www-form-urlencoded",
# "Access-Control-Allow-Origin": "*",
'X-Plex-Language': 'en',
'X-Plex-Device': self.addonName,
'X-Plex-Client-Platform': self.platform,
'X-Plex-Device-Name': self.deviceName,
'X-Plex-Platform': self.addonName,
'X-Plex-Platform-Version': 'unknown',
'X-Plex-Model': 'unknown',
'X-Plex-Product': self.addonName,
'X-Plex-Version': self.plexversion,
'X-Plex-Client-Identifier': self.clientId,
'X-Plex-Provides': 'player',
}
if self.token:
xargs['X-Plex-Token'] = self.token
if options:
xargs.update(options)
return xargs
def getXMLFromMultiplePMS(self, ATV_udid, path, type, options={}):
"""
provide combined XML representation of local servers' XMLs, eg. /library/section
parameters:
ATV_udid
path
type - owned <> shared (previously: local, myplex)
options
result:
XML
"""
queue = Queue.Queue()
threads = []
root = etree.Element("MediaConverter")
root.set('friendlyName', type+' Servers')
for uuid in g_PMS.get(ATV_udid, {}):
if (type=='all' and getPMSProperty(ATV_udid, uuid, 'name')!='plex.tv') or \
(type=='owned' and getPMSProperty(ATV_udid, uuid, 'owned')=='1') or \
(type=='shared' and getPMSProperty(ATV_udid, uuid, 'owned')=='0') or \
(type=='local' and getPMSProperty(ATV_udid, uuid, 'local')=='1') or \
(type=='remote' and getPMSProperty(ATV_udid, uuid, 'local')=='0'):
Server = etree.SubElement(root, 'Server') # create "Server" node
Server.set('name', getPMSProperty(ATV_udid, uuid, 'name'))
Server.set('address', getPMSProperty(ATV_udid, uuid, 'ip'))
Server.set('port', getPMSProperty(ATV_udid, uuid, 'port'))
Server.set('baseURL', getPMSProperty(ATV_udid, uuid, 'baseURL'))
Server.set('local', getPMSProperty(ATV_udid, uuid, 'local'))
Server.set('owned', getPMSProperty(ATV_udid, uuid, 'owned'))
baseURL = getPMSProperty(ATV_udid, uuid, 'baseURL')
token = getPMSProperty(ATV_udid, uuid, 'accesstoken')
PMS_mark = 'PMS(' + getPMSProperty(ATV_udid, uuid, 'address') + ')'
Server.set('searchKey', PMS_mark + getURL('', '', '/Search/Entry.xml'))
# request XMLs, one thread for each
PMS = { 'baseURL':baseURL, 'path':path, 'options':options, 'token':token, \
'data': {'uuid': uuid, 'Server': Server} }
t = Thread(target=getXMLFromPMSToQueue, args=(PMS, queue))
t.start()
threads.append(t)
# wait for requests being answered
for t in threads:
t.join()
# add new data to root XML, individual Server
while not queue.empty():
(data, XML) = queue.get()
uuid = data['uuid']
Server = data['Server']
baseURL = getPMSProperty(ATV_udid, uuid, 'baseURL')
token = getPMSProperty(ATV_udid, uuid, 'accesstoken')
PMS_mark = 'PMS(' + getPMSProperty(ATV_udid, uuid, 'address') + ')'
if XML==False:
Server.set('size', '0')
else:
Server.set('size', XML.getroot().get('size', '0'))
for Dir in XML.getiterator('Directory'): # copy "Directory" content, add PMS to links
key = Dir.get('key') # absolute path
Dir.set('key', PMS_mark + getURL('', path, key))
Dir.set('refreshKey', getURL(baseURL, path, key) + '/refresh')
if 'thumb' in Dir.attrib:
Dir.set('thumb', PMS_mark + getURL('', path, Dir.get('thumb')))
if 'art' in Dir.attrib:
Dir.set('art', PMS_mark + getURL('', path, Dir.get('art')))
Server.append(Dir)
for Playlist in XML.getiterator('Playlist'): # copy "Playlist" content, add PMS to links
key = Playlist.get('key') # absolute path
Playlist.set('key', PMS_mark + getURL('', path, key))
if 'composite' in Playlist.attrib:
Playlist.set('composite', PMS_mark + getURL('', path, Playlist.get('composite')))
Server.append(Playlist)
for Video in XML.getiterator('Video'): # copy "Video" content, add PMS to links
key = Video.get('key') # absolute path
Video.set('key', PMS_mark + getURL('', path, key))
if 'thumb' in Video.attrib:
Video.set('thumb', PMS_mark + getURL('', path, Video.get('thumb')))
if 'parentKey' in Video.attrib:
Video.set('parentKey', PMS_mark + getURL('', path, Video.get('parentKey')))
if 'parentThumb' in Video.attrib:
Video.set('parentThumb', PMS_mark + getURL('', path, Video.get('parentThumb')))
if 'grandparentKey' in Video.attrib:
Video.set('grandparentKey', PMS_mark + getURL('', path, Video.get('grandparentKey')))
if 'grandparentThumb' in Video.attrib:
Video.set('grandparentThumb', PMS_mark + getURL('', path, Video.get('grandparentThumb')))
Server.append(Video)
root.set('size', str(len(root.findall('Server'))))
XML = etree.ElementTree(root)
dprint(__name__, 1, "====== Local Server/Sections XML ======")
dprint(__name__, 1, XML.getroot())
dprint(__name__, 1, "====== Local Server/Sections XML finished ======")
return XML # XML representation - created "just in time". Do we need to cache it?
def getURL(self, baseURL, path, key): def getURL(self, baseURL, path, key):
if key.startswith('http://') or key.startswith('https://'): # external server if key.startswith('http://') or key.startswith('https://'): # external server
URL = key URL = key
@ -988,7 +836,7 @@ class PlexAPI():
MyPlexURL = 'https://' + MyPlexHost + MyPlexSignInPath MyPlexURL = 'https://' + MyPlexHost + MyPlexSignInPath
# create POST request # create POST request
xargs = self.getXArgsDeviceInfo(options) xargs = clientinfo.ClientInfo().getXArgsDeviceInfo(options)
self.logMsg("Header is: %s" % xargs, 1) self.logMsg("Header is: %s" % xargs, 1)
request = urllib2.Request(MyPlexURL, None, xargs) request = urllib2.Request(MyPlexURL, None, xargs)
request.get_method = lambda: 'POST' request.get_method = lambda: 'POST'
@ -1120,7 +968,7 @@ class PlexAPI():
self.logMsg("No user selected.", 0) self.logMsg("No user selected.", 0)
utils.settings('username', value='') utils.settings('username', value='')
xbmc.executebuiltin('Addon.OpenSettings(%s)' xbmc.executebuiltin('Addon.OpenSettings(%s)'
% self.addonId) % clientinfo.ClientInfo().getAddonId())
return False return False
# Only 1 user received, choose that one # Only 1 user received, choose that one
else: else:
@ -1164,7 +1012,8 @@ class PlexAPI():
if not username: if not username:
self.logMsg('Failed signing in a user to plex.tv', -1) self.logMsg('Failed signing in a user to plex.tv', -1)
xbmc.executebuiltin('Addon.OpenSettings(%s)' % self.addonId) xbmc.executebuiltin('Addon.OpenSettings(%s)'
% clientinfo.ClientInfo().getAddonId())
return False return False
return { return {
@ -1385,7 +1234,7 @@ class PlexAPI():
args['protocol'] = 'http' args['protocol'] = 'http'
args['maxAudioBitrate'] = maxAudioBitrate args['maxAudioBitrate'] = maxAudioBitrate
xargs = getXArgsDeviceInfo(options) xargs = clientinfo.ClientInfo().getXArgsDeviceInfo(options)
if not AuthToken=='': if not AuthToken=='':
xargs['X-Plex-Token'] = AuthToken xargs['X-Plex-Token'] = AuthToken
@ -1465,13 +1314,8 @@ class API():
self.item = item self.item = item
# which media part in the XML response shall we look at? # which media part in the XML response shall we look at?
self.part = 0 self.part = 0
self.clientinfo = clientinfo.ClientInfo()
self.clientId = self.clientinfo.getDeviceId()
self.__language__ = xbmcaddon.Addon().getLocalizedString self.__language__ = xbmcaddon.Addon().getLocalizedString
self.userId = utils.window('currUserId')
self.server = utils.window('pms_server') self.server = utils.window('pms_server')
self.token = utils.window('pms_token')
def setPartNumber(self, number=None): def setPartNumber(self, number=None):
""" """
@ -1888,7 +1732,7 @@ class API():
arguments overrule everything arguments overrule everything
""" """
xargs = PlexAPI().getXArgsDeviceInfo() xargs = clientinfo.ClientInfo().getXArgsDeviceInfo()
xargs.update(arguments) xargs.update(arguments)
if '?' not in url: if '?' not in url:
url = "%s?%s" % (url, urlencode(xargs)) url = "%s?%s" % (url, urlencode(xargs))
@ -1902,10 +1746,12 @@ class API():
url may or may not already contain a '?' url may or may not already contain a '?'
""" """
if utils.window('pms_token') == '':
return url
if '?' not in url: if '?' not in url:
url = "%s?X-Plex-Token=%s" % (url, self.token) url = "%s?X-Plex-Token=%s" % (url, utils.window('pms_token'))
else: else:
url = "%s&X-Plex-Token=%s" % (url, self.token) url = "%s&X-Plex-Token=%s" % (url, utils.window('pms_token'))
return url return url
def GetPlayQueueItemID(self): def GetPlayQueueItemID(self):
@ -2176,7 +2022,7 @@ class API():
TODO: mediaIndex TODO: mediaIndex
""" """
xargs = PlexAPI().getXArgsDeviceInfo() xargs = clientinfo.ClientInfo().getXArgsDeviceInfo()
# For DirectPlay, path/key of PART is needed # For DirectPlay, path/key of PART is needed
if action == "DirectPlay": if action == "DirectPlay":
path = self.item[0][self.part].attrib['key'] path = self.item[0][self.part].attrib['key']

70
resources/lib/clientinfo.py
View file

@ -7,40 +7,72 @@ from uuid import uuid4
import xbmc import xbmc
import xbmcaddon import xbmcaddon
import utils from utils import logging, window, settings
############################################################################### ###############################################################################
@utils.logging @logging
class ClientInfo(): class ClientInfo():
def __init__(self): def __init__(self):
self.addon = xbmcaddon.Addon() self.addon = xbmcaddon.Addon()
def getXArgsDeviceInfo(self, options=None):
"""
Returns a dictionary that can be used as headers for GET and POST
requests. An authentication option is NOT yet added.
Inputs:
options: dictionary of options that will override the
standard header options otherwise set.
Output:
header dictionary
"""
# Get addon infos
xargs = {
'Accept': '*/*',
'Connection': 'keep-alive',
"Content-Type": "application/x-www-form-urlencoded",
# "Access-Control-Allow-Origin": "*",
# 'X-Plex-Language': 'en',
'X-Plex-Device': self.getAddonName(),
'X-Plex-Client-Platform': self.getPlatform(),
'X-Plex-Device-Name': self.getDeviceName(),
'X-Plex-Platform': self.getPlatform(),
# 'X-Plex-Platform-Version': 'unknown',
# 'X-Plex-Model': 'unknown',
'X-Plex-Product': self.getAddonName(),
'X-Plex-Version': self.getVersion(),
'X-Plex-Client-Identifier': self.getDeviceId(),
'X-Plex-Provides': 'player',
}
if window('pms_token'):
xargs['X-Plex-Token'] = window('pms_token')
if options is not None:
xargs.update(options)
return xargs
def getAddonName(self): def getAddonName(self):
# Used for logging # Used for logging
return self.addon.getAddonInfo('name') return self.addon.getAddonInfo('name')
def getAddonId(self): def getAddonId(self):
return "plugin.video.plexkodiconnect" return "plugin.video.plexkodiconnect"
def getVersion(self): def getVersion(self):
return self.addon.getAddonInfo('version') return self.addon.getAddonInfo('version')
def getDeviceName(self): def getDeviceName(self):
if settings('deviceNameOpt') == "false":
if utils.settings('deviceNameOpt') == "false":
# Use Kodi's deviceName # Use Kodi's deviceName
deviceName = xbmc.getInfoLabel('System.FriendlyName').decode('utf-8') deviceName = xbmc.getInfoLabel(
'System.FriendlyName').decode('utf-8')
else: else:
deviceName = utils.settings('deviceName') deviceName = settings('deviceName')
deviceName = deviceName.replace("\"", "_") deviceName = deviceName.replace("\"", "_")
deviceName = deviceName.replace("/", "_") deviceName = deviceName.replace("/", "_")
return deviceName return deviceName
def getPlatform(self): def getPlatform(self):
@ -69,25 +101,25 @@ class ClientInfo():
If id does not exist, create one and save in Kodi settings file. If id does not exist, create one and save in Kodi settings file.
""" """
if reset is True: if reset is True:
utils.window('plex_client_Id', clear=True) window('plex_client_Id', clear=True)
utils.settings('plex_client_Id', value="") settings('plex_client_Id', value="")
clientId = utils.window('plex_client_Id') clientId = window('plex_client_Id')
if clientId: if clientId:
return clientId return clientId
clientId = utils.settings('plex_client_Id') clientId = settings('plex_client_Id')
# Because Kodi appears to cache file settings!! # Because Kodi appears to cache file settings!!
if clientId != "" and reset is False: if clientId != "" and reset is False:
utils.window('plex_client_Id', value=clientId) window('plex_client_Id', value=clientId)
self.logMsg("Unique device Id plex_client_Id loaded: %s" % clientId, 1) self.logMsg("Unique device Id plex_client_Id loaded: %s"
% clientId, 1)
return clientId return clientId
self.logMsg("Generating a new deviceid.", 0) self.logMsg("Generating a new deviceid.", 0)
clientId = str(uuid4()) clientId = str(uuid4())
utils.settings('plex_client_Id', value=clientId) settings('plex_client_Id', value=clientId)
utils.window('plex_client_Id', value=clientId) window('plex_client_Id', value=clientId)
self.logMsg("Unique device Id plex_client_Id loaded: %s" % clientId, 1) self.logMsg("Unique device Id plex_client_Id loaded: %s" % clientId, 1)
return clientId return clientId

20
resources/lib/downloadutils.py
View file

@ -13,8 +13,6 @@ import xbmcgui
import utils import utils
import clientinfo import clientinfo
import PlexAPI
############################################################################### ###############################################################################
# Disable requests logging # Disable requests logging
@ -30,16 +28,16 @@ class DownloadUtils():
# Borg - multiple instances, shared state # Borg - multiple instances, shared state
_shared_state = {} _shared_state = {}
clientInfo = clientinfo.ClientInfo()
# Requests session # Requests session
s = None s = None
timeout = 30 timeout = 30
def __init__(self): def __init__(self):
self.__dict__ = self._shared_state self.__dict__ = self._shared_state
self.clientInfo = clientinfo.ClientInfo()
def setUsername(self, username): def setUsername(self, username):
# Reserved for userclient only # Reserved for userclient only
self.username = username self.username = username
@ -175,17 +173,14 @@ class DownloadUtils():
self.logMsg("Requests session could not be terminated.", 1) self.logMsg("Requests session could not be terminated.", 1)
def getHeader(self, authenticate=True, options={}): def getHeader(self, authenticate=True, options={}):
plx = PlexAPI.PlexAPI()
if authenticate: if authenticate:
header = plx.getXArgsDeviceInfo(options=options) header = self.clientInfo.getXArgsDeviceInfo(options=options)
else: else:
header = plx.getXArgsDeviceInfo(options=options) header = self.clientInfo.getXArgsDeviceInfo(options=options)
return header return header
def downloadUrl(self, url, postBody=None, type="GET", parameters=None, authenticate=True, headerOptions={}): def downloadUrl(self, url, postBody=None, type="GET", parameters=None,
authenticate=True, headerOptions={}):
# self.logMsg("=== ENTER downloadUrl ===", 2)
timeout = self.timeout timeout = self.timeout
default_link = "" default_link = ""
@ -276,7 +271,8 @@ class DownloadUtils():
# If user is not authenticated # If user is not authenticated
elif not authenticate: elif not authenticate:
header = self.getHeader(authenticate=False, options=headerOptions) header = self.getHeader(authenticate=False,
options=headerOptions)
# If user enables ssl verification # If user enables ssl verification
try: try: