377 lines
No EOL
16 KiB
Python
377 lines
No EOL
16 KiB
Python
# API.py
|
|
# This class helps translate more complex cases from the MediaBrowser API to the XBMC API
|
|
|
|
from datetime import datetime
|
|
import xbmc
|
|
import xbmcgui
|
|
import xbmcaddon
|
|
|
|
class API():
|
|
logLevel = 0
|
|
addonSettings = None
|
|
getString = None
|
|
LogCalls = False
|
|
TrackLog = ""
|
|
TotalUrlCalls = 0
|
|
|
|
def __init__(self, *args):
|
|
self.addonSettings = xbmcaddon.Addon(id='plugin.video.mb3sync')
|
|
self.getString = self.addonSettings.getLocalizedString
|
|
level = self.addonSettings.getSetting('logLevel')
|
|
self.logLevel = 0
|
|
if(level != None and level != ""):
|
|
self.logLevel = int(level)
|
|
if(self.logLevel == 2):
|
|
self.LogCalls = True
|
|
|
|
def logMsg(self, msg, level = 1):
|
|
if(self.logLevel >= level):
|
|
try:
|
|
xbmc.log("mb3sync DownloadUtils -> " + str(msg))
|
|
except UnicodeEncodeError:
|
|
try:
|
|
xbmc.log("mb3sync DownloadUtils -> " + str(msg.encode('utf-8')))
|
|
except: pass
|
|
|
|
def getPeople(self, item):
|
|
# Process People
|
|
director=''
|
|
writer=''
|
|
cast=[]
|
|
people = item.get("People")
|
|
if(people != None):
|
|
for person in people:
|
|
if(person.get("Type") == "Director"):
|
|
director = director + person.get("Name") + ' '
|
|
if(person.get("Type") == "Writing"):
|
|
writer = person.get("Name")
|
|
if(person.get("Type") == "Writer"):
|
|
writer = person.get("Name")
|
|
if(person.get("Type") == "Actor"):
|
|
Name = person.get("Name")
|
|
Role = person.get("Role")
|
|
if Role == None:
|
|
Role = ''
|
|
cast.append(Name)
|
|
return {'Director' : director,
|
|
'Writer' : writer,
|
|
'Cast' : cast
|
|
}
|
|
|
|
def getTimeInfo(self, item):
|
|
resumeTime = ''
|
|
userData = item.get("UserData")
|
|
PlaybackPositionTicks = '100'
|
|
if userData.get("PlaybackPositionTicks") != None:
|
|
PlaybackPositionTicks = str(userData.get("PlaybackPositionTicks"))
|
|
reasonableTicks = int(userData.get("PlaybackPositionTicks")) / 1000
|
|
resumeTime = reasonableTicks / 10000
|
|
|
|
try:
|
|
tempDuration = str(int(item.get("RunTimeTicks", "0"))/(10000000*60))
|
|
except TypeError:
|
|
try:
|
|
tempDuration = str(int(item.get("CumulativeRunTimeTicks"))/(10000000*60))
|
|
except TypeError:
|
|
tempDuration = "0"
|
|
cappedPercentage = None
|
|
resume=0
|
|
percentage=0
|
|
if (resumeTime != "" and int(resumeTime) > 0):
|
|
duration = float(tempDuration)
|
|
if(duration > 0):
|
|
resume = float(resumeTime) / 60
|
|
percentage = int((resume / duration) * 100.0)
|
|
return {'Duration' : tempDuration,
|
|
'TotalTime' : tempDuration,
|
|
'Percent' : str(percentage),
|
|
'ResumeTime' : str(resume)
|
|
}
|
|
|
|
def getStudio(self, item):
|
|
# Process Studio
|
|
studio = ""
|
|
if item.get("SeriesStudio") != None and item.get("SeriesStudio") != '':
|
|
studio = item.get("SeriesStudio")
|
|
if studio == "":
|
|
studios = item.get("Studios")
|
|
if(studios != None):
|
|
for studio_string in studios:
|
|
if studio=="": #Just take the first one
|
|
temp=studio_string.get("Name")
|
|
studio=temp.encode('utf-8')
|
|
return studio
|
|
|
|
def getMediaStreams(self, item, mediaSources=False):
|
|
# Process MediaStreams
|
|
channels = ''
|
|
videocodec = ''
|
|
audiocodec = ''
|
|
height = ''
|
|
width = ''
|
|
aspectratio = '1:1'
|
|
aspectfloat = 1.85
|
|
|
|
if mediaSources == True:
|
|
mediaSources = item.get("MediaSources")
|
|
if(mediaSources != None):
|
|
MediaStreams = mediaSources[0].get("MediaStreams")
|
|
else:
|
|
MediaStreams = None
|
|
else:
|
|
MediaStreams = item.get("MediaStreams")
|
|
if(MediaStreams != None):
|
|
#mediaStreams = MediaStreams[0].get("MediaStreams")
|
|
if(MediaStreams != None):
|
|
for mediaStream in MediaStreams:
|
|
if(mediaStream.get("Type") == "Video"):
|
|
videocodec = mediaStream.get("Codec")
|
|
height = str(mediaStream.get("Height"))
|
|
width = str(mediaStream.get("Width"))
|
|
aspectratio = mediaStream.get("AspectRatio")
|
|
if aspectratio != None and len(aspectratio) >= 3:
|
|
try:
|
|
aspectwidth,aspectheight = aspectratio.split(':')
|
|
aspectfloat = float(aspectwidth) / float(aspectheight)
|
|
except:
|
|
aspectfloat = 1.85
|
|
if(mediaStream.get("Type") == "Audio"):
|
|
audiocodec = mediaStream.get("Codec")
|
|
channels = mediaStream.get("Channels")
|
|
return {'channels' : str(channels),
|
|
'videocodec' : videocodec,
|
|
'audiocodec' : audiocodec,
|
|
'height' : height,
|
|
'width' : width,
|
|
'aspectratio' : str(aspectfloat)
|
|
}
|
|
|
|
def getUserData(self, item):
|
|
userData = item.get("UserData")
|
|
resumeTime = 0
|
|
if(userData != None):
|
|
if userData.get("Played") != True:
|
|
watched="True"
|
|
else:
|
|
watched="False"
|
|
if userData.get("IsFavorite") == True:
|
|
favorite="True"
|
|
else:
|
|
favorite="False"
|
|
if(userData.get("Played") == True):
|
|
playcount="1"
|
|
else:
|
|
playcount="0"
|
|
if userData.get('UnplayedItemCount') != None:
|
|
UnplayedItemCount = userData.get('UnplayedItemCount')
|
|
else:
|
|
UnplayedItemCount = "0"
|
|
if userData.get('PlaybackPositionTicks') != None:
|
|
PlaybackPositionTicks = userData.get('PlaybackPositionTicks')
|
|
else:
|
|
PlaybackPositionTicks = ''
|
|
return {'Watched' : watched,
|
|
'Favorite' : favorite,
|
|
'PlayCount': playcount,
|
|
'UnplayedItemCount' : UnplayedItemCount,
|
|
'PlaybackPositionTicks' : str(PlaybackPositionTicks)
|
|
}
|
|
|
|
def getGenre(self,item):
|
|
genre = ""
|
|
genres = item.get("Genres")
|
|
if genres != None and genres != []:
|
|
for genre_string in genres:
|
|
if genre == "": #Just take the first genre
|
|
genre = genre_string
|
|
else:
|
|
genre = genre + " / " + genre_string
|
|
elif item.get("SeriesGenres") != None and item.get("SeriesGenres") != '':
|
|
genres = item.get("SeriesGenres")
|
|
if genres != None and genres != []:
|
|
for genre_string in genres:
|
|
if genre == "": #Just take the first genre
|
|
genre = genre_string
|
|
else:
|
|
genre = genre + " / " + genre_string
|
|
return genre
|
|
|
|
def getName(self, item):
|
|
Temp = item.get("Name")
|
|
if Temp == None:
|
|
Temp = ""
|
|
Name=Temp.encode('utf-8')
|
|
return Name
|
|
|
|
def getRecursiveItemCount(self, item):
|
|
if item.get("RecursiveItemCount") != None:
|
|
return str(item.get("RecursiveItemCount"))
|
|
else:
|
|
return "0"
|
|
|
|
def getSeriesName(self, item):
|
|
Temp = item.get("SeriesName")
|
|
if Temp == None:
|
|
Temp = ""
|
|
Name=Temp.encode('utf-8')
|
|
return Name
|
|
|
|
def getOverview(self, item):
|
|
Temp = item.get("Overview")
|
|
if Temp == None:
|
|
Temp=''
|
|
Overview1=Temp.encode('utf-8')
|
|
Overview=str(Overview1)
|
|
return Overview
|
|
|
|
def getPremiereDate(self, item):
|
|
if(item.get("PremiereDate") != None):
|
|
premieredatelist = (item.get("PremiereDate")).split("T")
|
|
premieredate = premieredatelist[0]
|
|
else:
|
|
premieredate = ""
|
|
Temp = premieredate
|
|
premieredate = Temp.encode('utf-8')
|
|
return premieredate
|
|
|
|
def getTVInfo(self, item, userData):
|
|
TotalSeasons = 0 if item.get("ChildCount")==None else item.get("ChildCount")
|
|
TotalEpisodes = 0 if item.get("RecursiveItemCount")==None else item.get("RecursiveItemCount")
|
|
WatchedEpisodes = 0 if userData.get("UnplayedItemCount")==None else TotalEpisodes-int(userData.get("UnplayedItemCount"))
|
|
UnWatchedEpisodes = 0 if userData.get("UnplayedItemCount")==None else int(userData.get("UnplayedItemCount"))
|
|
NumEpisodes = TotalEpisodes
|
|
tempEpisode = ""
|
|
if (item.get("IndexNumber") != None):
|
|
episodeNum = item.get("IndexNumber")
|
|
if episodeNum < 10:
|
|
tempEpisode = "0" + str(episodeNum)
|
|
else:
|
|
tempEpisode = str(episodeNum)
|
|
|
|
tempSeason = ""
|
|
if (str(item.get("ParentIndexNumber")) != None):
|
|
tempSeason = str(item.get("ParentIndexNumber"))
|
|
if item.get("ParentIndexNumber") < 10:
|
|
tempSeason = "0" + tempSeason
|
|
if item.get("SeriesName") != None:
|
|
temp=item.get("SeriesName")
|
|
SeriesName=temp.encode('utf-8')
|
|
else:
|
|
SeriesName=''
|
|
return {'TotalSeasons' : str(TotalSeasons),
|
|
'TotalEpisodes' : str(TotalEpisodes),
|
|
'WatchedEpisodes' : str(WatchedEpisodes),
|
|
'UnWatchedEpisodes': str(UnWatchedEpisodes),
|
|
'NumEpisodes' : str(NumEpisodes),
|
|
'Season' : tempSeason,
|
|
'Episode' : tempEpisode,
|
|
'SeriesName' : SeriesName
|
|
}
|
|
def getDateCreated(self, item):
|
|
tempDate = item.get("DateCreated")
|
|
if tempDate != None:
|
|
tempDate = tempDate.split("T")[0]
|
|
date = tempDate.split("-")
|
|
tempDate = date[2] + "." + date[1] + "." +date[0]
|
|
else:
|
|
tempDate = "01.01.2000"
|
|
return tempDate
|
|
|
|
def getArtwork(self, data, type, index = "0", userParentInfo = False):
|
|
|
|
id = data.get("Id")
|
|
getSeriesData = False
|
|
userData = data.get("UserData")
|
|
|
|
if type == "tvshow.poster": # Change the Id to the series to get the overall series poster
|
|
if data.get("Type") == "Season" or data.get("Type")== "Episode":
|
|
id = data.get("SeriesId")
|
|
getSeriesData = True
|
|
elif type == "poster" and data.get("Type") == "Episode" and self.addonSettings.getSetting('useSeasonPoster')=='true': # Change the Id to the Season to get the season poster
|
|
id = data.get("SeasonId")
|
|
if type == "poster" or type == "tvshow.poster": # Now that the Ids are right, change type to MB3 name
|
|
type="Primary"
|
|
if data.get("Type") == "Season": # For seasons: primary (poster), thumb and banner get season art, rest series art
|
|
if type != "Primary" and type != "Primary2" and type != "Primary3" and type != "Primary4" and type != "Thumb" and type != "Banner" and type!="Thumb3":
|
|
id = data.get("SeriesId")
|
|
getSeriesData = True
|
|
if data.get("Type") == "Episode": # For episodes: primary (episode thumb) gets episode art, rest series art.
|
|
if type != "Primary" and type != "Primary2" and type != "Primary3" and type != "Primary4":
|
|
id = data.get("SeriesId")
|
|
getSeriesData = True
|
|
if type =="Primary2" or type=="Primary3" or type=="Primary4":
|
|
id = data.get("SeasonId")
|
|
getSeriesData = True
|
|
if data.get("SeasonUserData") != None:
|
|
userData = data.get("SeasonUserData")
|
|
if id == None:
|
|
id=data.get("Id")
|
|
|
|
imageTag = "e3ab56fe27d389446754d0fb04910a34" # a place holder tag, needs to be in this format
|
|
originalType = type
|
|
if type == "Primary2" or type == "Primary3" or type == "Primary4" or type=="SeriesPrimary":
|
|
type = "Primary"
|
|
if type == "Backdrop2" or type=="Backdrop3" or type=="BackdropNoIndicators":
|
|
type = "Backdrop"
|
|
if type == "Thumb2" or type=="Thumb3":
|
|
type = "Thumb"
|
|
if(data.get("ImageTags") != None and data.get("ImageTags").get(type) != None):
|
|
imageTag = data.get("ImageTags").get(type)
|
|
|
|
if (data.get("Type") == "Episode" or data.get("Type") == "Season") and type=="Logo":
|
|
imageTag = data.get("ParentLogoImageTag")
|
|
if (data.get("Type") == "Episode" or data.get("Type") == "Season") and type=="Art":
|
|
imageTag = data.get("ParentArtImageTag")
|
|
if (data.get("Type") == "Episode") and originalType=="Thumb3":
|
|
imageTag = data.get("SeriesThumbImageTag")
|
|
if (data.get("Type") == "Season") and originalType=="Thumb3" and imageTag=="e3ab56fe27d389446754d0fb04910a34" :
|
|
imageTag = data.get("ParentThumbImageTag")
|
|
id = data.get("SeriesId")
|
|
|
|
query = ""
|
|
height = "10000"
|
|
width = "10000"
|
|
played = "0"
|
|
totalbackdrops = 0
|
|
|
|
if originalType =="BackdropNoIndicators" and index == "0" and data.get("BackdropImageTags") != None:
|
|
totalbackdrops = len(data.get("BackdropImageTags"))
|
|
if totalbackdrops != 0:
|
|
index = str(randrange(0,totalbackdrops))
|
|
# use the local image proxy server that is made available by this addons service
|
|
|
|
port = self.addonSettings.getSetting('port')
|
|
host = self.addonSettings.getSetting('ipaddress')
|
|
server = host + ":" + port
|
|
|
|
if self.addonSettings.getSetting('compressArt')=='true':
|
|
query = query + "&Quality=90"
|
|
|
|
if imageTag == None:
|
|
imageTag = "e3ab56fe27d389446754d0fb04910a34"
|
|
artwork = "http://" + server + "/mediabrowser/Items/" + str(id) + "/Images/" + type + "/" + index + "/" + imageTag + "/original/" + width + "/" + height + "/" + played + "?" + query
|
|
if self.addonSettings.getSetting('disableCoverArt')=='true':
|
|
artwork = artwork + "&EnableImageEnhancers=false"
|
|
|
|
self.logMsg("getArtwork : " + artwork, level=2)
|
|
|
|
# do not return non-existing images
|
|
if ( (type!="Backdrop" and imageTag=="e3ab56fe27d389446754d0fb04910a34") | #Remember, this is the placeholder tag, meaning we didn't find a valid tag
|
|
(type=="Backdrop" and data.get("BackdropImageTags") != None and len(data.get("BackdropImageTags")) == 0) |
|
|
(type=="Backdrop" and data.get("BackdropImageTag") != None and len(data.get("BackdropImageTag")) == 0)
|
|
):
|
|
if type != "Backdrop" or (type=="Backdrop" and getSeriesData==True and data.get("ParentBackdropImageTags") == None) or (type=="Backdrop" and getSeriesData!=True):
|
|
artwork=''
|
|
|
|
return artwork
|
|
|
|
def getUserArtwork(self, data, type, index = "0"):
|
|
|
|
id = data.get("Id")
|
|
port = self.addonSettings.getSetting('port')
|
|
host = self.addonSettings.getSetting('ipaddress')
|
|
server = host + ":" + port
|
|
artwork = "http://" + server + "/mediabrowser/Users/" + str(id) + "/Images/" + type + "?Format=original"
|
|
|
|
return artwork
|
|
|