# -- coding: utf-8 -- # API.py # This class helps translate more complex cases from the MediaBrowser API to the XBMC API from datetime import datetime from random import randrange import xbmc import xbmcgui import Utils as utils class API(): def getPeople(self, item): # Process People director = [] writer = [] cast = [] try: people = item['People'] except: pass else: for person in people: type = person['Type'] Name = person['Name'] if "Director" in type: director.append(Name) elif "Writing" in type: writer.append(Name) elif "Writer" in type: writer.append(Name) elif "Actor" in type: cast.append(Name) return { 'Director': director, 'Writer': writer, 'Cast': cast } def getTimeInfo(self, item): # Runtime and Resume point tempRuntime = 0 runtime = 0 resume = 0 try: # Get resume point userdata = item['UserData'] playbackPosition = userdata['PlaybackPositionTicks'] resume = playbackPosition / 10000000.0 except: pass try: # Get total runtime tempRuntime = item['RunTimeTicks'] except: try: tempRuntime = item['CumulativeRunTimeTicks'] except: pass finally: runtime = tempRuntime / 10000000.0 return { 'ResumeTime': resume, 'TotalTime': runtime } def getStudios(self, item): # Process Studio studios = [] try: studio = item['SeriesStudio'] studios.append(self.getStudio(studio)) except: try: studioArray = item['Studios'] for studio in studioArray: studios.append(self.getStudio(studio['Name'])) except: pass return studios def getStudio(self, studioName): # Convert studio for Kodi to properly detect them studios = { 'abc (us)': "ABC", 'fox (us)': "FOX", 'mtv (us)': "MTV", 'showcase (ca)': "Showcase", 'wgn america': "WGN" } return studios.get(studioName.lower(), studioName) 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 getMediaStreams(self, item, mediaSources = False): videotracks = [] # Height, Width, Codec, AspectRatio, AspectFloat, 3D audiotracks = [] # Codec, Channels, language subtitlelanguages = [] # Language if mediaSources: try: MediaStreams = item['MediaSources'][0]['MediaStreams'] except: MediaStreams = None else: MediaStreams = item.get('MediaStreams') if MediaStreams: # Sort through the Video, Audio, Subtitle tracks for mediaStream in MediaStreams: type = mediaStream.get('Type', "") profile = mediaStream.get('Profile', "").lower() codec = mediaStream.get('Codec', "").lower() if "Video" in type: videotrack = {} videotrack['videocodec'] = codec container = item['MediaSources'][0].get('Container', "").lower() if "msmpeg4" in videotrack['videocodec']: videotrack['videocodec'] = "divx" elif "mpeg4" in videotrack['videocodec']: if "simple profile" in profile or profile == "": videotrack['videocodec'] = "xvid" elif "h264" in videotrack['videocodec']: if container in ("mp4", "mov", "m4v"): videotrack['videocodec'] = "avc1" videotrack['height'] = mediaStream.get('Height') videotrack['width'] = mediaStream.get('Width') videotrack['Video3DFormat'] = item.get('Video3DFormat') if item.get('AspectRatio'): # Metadata aspect ratio videotrack['aspectratio'] = item['AspectRatio'] else: # File aspect ratio videotrack['aspectratio'] = mediaStream.get('AspectRatio', "0") if len(videotrack['aspectratio']) >= 3: try: aspectwidth, aspectheight = videotrack['aspectratio'].split(':') videotrack['aspectratio'] = round(float(aspectwidth) / float(aspectheight), 6) except: videotrack['aspectratio'] = 1.85 else: try: videotrack['aspectratio'] = round(float(videotrack['width'] / videotrack['height']), 6) except: # In the event the aspect ratio is missing and the width and height are missing as well. videotrack['aspectratio'] = 1.85 videotracks.append(videotrack) elif "Audio" in type: audiotrack = {} audiotrack['audiocodec'] = codec if "dca" in audiotrack['audiocodec'] and "dts-hd ma" in profile: audiotrack['audiocodec'] = "dtshd_ma" audiotrack['channels'] = mediaStream.get('Channels') audiotrack['audiolanguage'] = mediaStream.get('Language') audiotracks.append(audiotrack) elif "Subtitle" in type: try: subtitlelanguages.append(mediaStream['Language']) except: subtitlelanguages.append("Unknown") return { 'videocodec' : videotracks, 'audiocodec' : audiotracks, 'subtitlelanguage' : subtitlelanguages } def getChecksum(self, item): # use the etags checksum for this if available # AND the userdata checksum = "" if item.get("Etag") != None: checksum = item.get("Etag") userData = item.get("UserData") if(userData != None): checksum += str(userData.get("Played")) checksum += str(userData.get("IsFavorite")) if userData.get('UnplayedItemCount') != None: checksum += str(userData.get("UnplayedItemCount")) if userData.get('LastPlayedDate') != None: checksum += str(userData.get("LastPlayedDate")) if userData.get('PlaybackPositionTicks') != None: checksum += str(userData.get("PlaybackPositionTicks")) return checksum def getUserData(self, item): # Default favorite = False playcount = None lastPlayedDate = None userKey = "" try: userdata = item['UserData'] except: # No userdata found. pass else: favorite = userdata['IsFavorite'] userKey = userdata.get('Key', "") watched = userdata['Played'] if watched: # Playcount is tied to the watch status playcount = userdata['PlayCount'] if playcount == 0: playcount = 1 else: playcount = None lastPlayedDate = userdata.get('LastPlayedDate', None) if lastPlayedDate: lastPlayedDate = lastPlayedDate.split('.')[0].replace('T', " ") return { 'Favorite': favorite, 'PlayCount': playcount, 'LastPlayedDate': lastPlayedDate, 'Key': userKey } def getRecursiveItemCount(self, item): if item.get("RecursiveItemCount") != None: return str(item.get("RecursiveItemCount")) else: return "0" def getOverview(self, item): overview = "" try: overview = item['Overview'] overview = overview.replace("\"", "\'") overview = overview.replace("\n", " ") overview = overview.replace("\r", " ") except: pass return overview 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): try: dateadded = item['DateCreated'] dateadded = dateadded.split('.')[0].replace('T', " ") except: dateadded = None return dateadded def getPremiereDate(self, item): try: premiere = item['PremiereDate'] premiere = premiere.split('.')[0].replace('T', " ") except: premiere = None return premiere def getTagline(self, item): try: tagline = item['Taglines'][0] except: tagline = None return tagline def getProvider(self, item, providername): # Provider Name: imdb or tvdb try: if "imdb" in providername: provider = item['ProviderIds']['Imdb'] elif "tvdb" in providername: provider = item['ProviderIds']['Tvdb'] elif "musicBrainzArtist" in providername: provider = item['ProviderIds']['MusicBrainzArtist'] elif "musicBrainzAlbum" in providername: provider = item['ProviderIds']['MusicBrainzAlbum'] elif "musicBrainzTrackId" in providername: provider = item['ProviderIds']['MusicBrainzTrackId'] except: provider = None return provider def getCountry(self, item): try: country = item['ProductionLocations'][0] except: country = None return country def getMpaa(self, item): # Convert more complex cases mpaa = item.get('OfficialRating', "") if mpaa in ("NR", "UR"): # Kodi seems to not like NR, but will accept Rated Not Rated mpaa = "Rated Not Rated" return mpaa def getAllArtwork(self, item, parentInfo = False): """ Get all artwork, it will return an empty string for the artwork type not found. Artwork type: Primary, Art, Banner, Logo, Thumb, Disc, Backdrop """ username = utils.window('currUser') server = utils.window('server%s' % username) id = item['Id'] artworks = item['ImageTags'] backdrops = item['BackdropImageTags'] maxHeight = 10000 maxWidth = 10000 quality = "" if utils.settings('compressArt') == "true": quality = "&Quality=90" allartworks = { 'Primary': "", 'Art': "", 'Banner': "", 'Logo': "", 'Thumb': "", 'Disc': "", 'Backdrop': [] } # Process backdrops backdropIndex = 0 for backdroptag in backdrops: artwork = "%s/mediabrowser/Items/%s/Images/Backdrop/%s?MaxWidth=%s&MaxHeight=%s&Format=original&Tag=%s%s" % (server, id, backdropIndex, maxWidth, maxHeight, backdroptag, quality) allartworks['Backdrop'].append(artwork) backdropIndex += 1 # Process the rest of the artwork for art in artworks: tag = artworks[art] artwork = "%s/mediabrowser/Items/%s/Images/%s/0?MaxWidth=%s&MaxHeight=%s&Format=original&Tag=%s%s" % (server, id, art, maxWidth, maxHeight, tag, quality) allartworks[art] = artwork # Process parent items if the main item is missing artwork if parentInfo: # Process backdrops if not allartworks['Backdrop']: parentId = item.get('ParentBackdropItemId') if parentId: # If there is a parentId, go through the parent backdrop list parentbackdrops = item['ParentBackdropImageTags'] backdropIndex = 0 for parentbackdroptag in parentbackdrops: artwork = "%s/mediabrowser/Items/%s/Images/Backdrop/%s?MaxWidth=%s&MaxHeight=%s&Format=original&Tag=%s%s" % (server, parentId, backdropIndex, maxWidth, maxHeight, parentbackdroptag, quality) allartworks['Backdrop'].append(artwork) backdropIndex += 1 # Process the rest of the artwork parentartwork = ['Logo', 'Art', 'Thumb'] for parentart in parentartwork: if not allartworks[parentart]: parentId = item.get('Parent%sItemId' % parentart) if parentId: parentTag = item['Parent%sItemId' % parentart] artwork = "%s/mediabrowser/Items/%s/Images/%s/0?MaxWidth=%s&MaxHeight=%s&Format=original&Tag=%s%s" % (server, parentId, parentart, maxWidth, maxHeight, parentTag, quality) allartworks[parentart] = artwork return allartworks def getArtwork(self, data, type, mediaType = "", 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 utils.settings('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" and type!="Backdrop": 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" or data.get("Type") == "Season") and type=="Backdrop": if data.get("BackdropImageTags"): imageTag = data['BackdropImageTags'][0] 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") # for music we return the parent art if no image exists if (data.get("Type") == "MusicAlbum" or data.get("Type") == "Audio") and type=="Backdrop" and not data.get("BackdropImageTags"): data["BackdropImageTags"] = data.get("ParentBackdropImageTags") id = data.get("ParentBackdropItemId") if (data.get("Type") == "MusicAlbum" or data.get("Type") == "Audio") and type=="Logo" and (not imageTag or imageTag == "e3ab56fe27d389446754d0fb04910a34"): imageTag = data.get("ParentLogoImageTag") id = data.get("ParentLogoItemId") if (data.get("Type") == "MusicAlbum" or data.get("Type") == "Audio") and type=="Art" and (not imageTag or imageTag == "e3ab56fe27d389446754d0fb04910a34"): imageTag = data.get("ParentArtImageTag") id = data.get("ParentArtItemId") query = "" maxHeight = "10000" maxWidth = "10000" height = "" width = "" played = "0" totalbackdrops = 0 if utils.settings('coverArtratio') == "true": if mediaType in ("movie","boxset","tvshow"): if "Primary" in type: # Only force ratio for cover art for main covers aspectratio = data.get("PrimaryImageAspectRatio") width = "&Width=1000" height = "&Height=1480" 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 # Load user information set by UserClient WINDOW = xbmcgui.Window(10000) username = WINDOW.getProperty('currUser') server = WINDOW.getProperty('server%s' % username) if utils.settings('compressArt')=='true': query = query + "&Quality=90" if imageTag == None: imageTag = "e3ab56fe27d389446754d0fb04910a34" artwork = "%s/mediabrowser/Items/%s/Images/%s/%s?MaxWidth=%s&MaxHeight=%s%s%s&Format=original&Tag=%s%s" % (server, id, type, index, maxWidth, maxHeight, height, width, imageTag, query) #artwork = "%s/mediabrowser/Items/%s/Images/%s/%s/%s/original/%s/%s/%s?%s" % (server, id, type, index, imageTag, width, height, played, query) <- broken if utils.settings('disableCoverArt')=='true': artwork = artwork + "&EnableImageEnhancers=false" # 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 imageUrl(self, id, type, index, width, height): WINDOW = xbmcgui.Window(10000) username = WINDOW.getProperty('currUser') server = WINDOW.getProperty('server%s' % username) # For people image - actors, directors, writers return "%s/mediabrowser/Items/%s/Images/%s?MaxWidth=%s&MaxHeight=%s&Index=%s" % (server, id, type, width, height, index) def getUserArtwork(self, data, type, index = "0"): # Load user information set by UserClient WINDOW = xbmcgui.Window(10000) username = WINDOW.getProperty('currUser') server = WINDOW.getProperty('server%s' % username) id = data.get("Id") artwork = "%s/mediabrowser/Users/%s/Images/%s?Format=original" % (server, id, type) return artwork