jellyfin-kodi/resources/lib/read_embyserver.py

577 lines
20 KiB
Python
Raw Normal View History

# -*- coding: utf-8 -*-
#################################################################################################
import logging
import hashlib
import xbmc
import downloadutils
import database
from utils import window, settings
from contextlib import closing
#################################################################################################
log = logging.getLogger("EMBY."+__name__)
#################################################################################################
class Read_EmbyServer():
2016-06-16 05:43:36 +00:00
limitIndex = int(settings('limitindex'))
def __init__(self):
self.doUtils = downloadutils.DownloadUtils().downloadUrl
self.userId = window('emby_currUser')
self.server = window('emby_server%s' % self.userId)
def get_emby_url(self, handler):
return "{server}/emby/%s" % handler
def split_list(self, itemlist, size):
# Split up list in pieces of size. Will generate a list of lists
return [itemlist[i:i+size] for i in range(0, len(itemlist), size)]
def getItem(self, itemid):
# This will return the full item
item = {}
2016-04-15 20:01:12 +00:00
result = self.doUtils("{server}/emby/Users/{UserId}/Items/%s?format=json" % itemid)
if result:
item = result
return item
def getItems(self, itemlist):
items = []
itemlists = self.split_list(itemlist, 50)
for itemlist in itemlists:
# Will return basic information
params = {
'Ids': ",".join(itemlist),
'Fields': "Etag"
}
2016-06-16 05:43:36 +00:00
url = "{server}/emby/Users/{UserId}/Items?&format=json"
result = self.doUtils(url, parameters=params)
if result:
items.extend(result['Items'])
return items
def getFullItems(self, itemlist):
items = []
itemlists = self.split_list(itemlist, 50)
for itemlist in itemlists:
params = {
"Ids": ",".join(itemlist),
"Fields": (
"Path,Genres,SortName,Studios,Writer,ProductionYear,Taglines,"
"CommunityRating,OfficialRating,CumulativeRunTimeTicks,"
"Metascore,AirTime,DateCreated,MediaStreams,People,Overview,"
"CriticRating,CriticRatingSummary,Etag,ShortOverview,ProductionLocations,"
"Tags,ProviderIds,ParentId,RemoteTrailers,SpecialEpisodeNumbers,"
2016-04-24 00:14:21 +00:00
"MediaSources,VoteCount"
)
}
2016-06-16 05:43:36 +00:00
url = "{server}/emby/Users/{UserId}/Items?format=json"
result = self.doUtils(url, parameters=params)
if result:
items.extend(result['Items'])
return items
2016-06-16 05:43:36 +00:00
def getFilteredSection(self, parentid, itemtype=None, sortby="SortName", recursive=True,
2016-08-18 06:41:18 +00:00
limit=None, sortorder="Ascending", filter_type=""):
params = {
'ParentId': parentid,
'IncludeItemTypes': itemtype,
'CollapseBoxSetItems': False,
'IsVirtualUnaired': False,
'IsMissing': False,
'Recursive': recursive,
'Limit': limit,
'SortBy': sortby,
'SortOrder': sortorder,
2016-09-05 01:19:51 +00:00
'Filters': filter_type,
2016-06-16 05:43:36 +00:00
'Fields': (
"Path,Genres,SortName,Studios,Writer,ProductionYear,Taglines,"
"CommunityRating,OfficialRating,CumulativeRunTimeTicks,"
"Metascore,AirTime,DateCreated,MediaStreams,People,Overview,"
"CriticRating,CriticRatingSummary,Etag,ShortOverview,ProductionLocations,"
"Tags,ProviderIds,ParentId,RemoteTrailers,SpecialEpisodeNumbers"
)
}
2016-03-31 18:18:49 +00:00
return self.doUtils("{server}/emby/Users/{UserId}/Items?format=json", parameters=params)
def getTvChannels(self):
2016-06-16 05:43:36 +00:00
params = {
'EnableImages': True,
2016-06-16 05:43:36 +00:00
'Fields': (
"Path,Genres,SortName,Studios,Writer,ProductionYear,Taglines,"
"CommunityRating,OfficialRating,CumulativeRunTimeTicks,"
"Metascore,AirTime,DateCreated,MediaStreams,People,Overview,"
"CriticRating,CriticRatingSummary,Etag,ShortOverview,ProductionLocations,"
"Tags,ProviderIds,ParentId,RemoteTrailers,SpecialEpisodeNumbers"
)
}
2016-06-16 05:43:36 +00:00
url = "{server}/emby/LiveTv/Channels/?userid={UserId}&format=json"
return self.doUtils(url, parameters=params)
def getTvRecordings(self, groupid):
2016-06-16 05:43:36 +00:00
if groupid == "root":
groupid = ""
params = {
'GroupId': groupid,
'EnableImages': True,
2016-06-16 05:43:36 +00:00
'Fields': (
"Path,Genres,SortName,Studios,Writer,ProductionYear,Taglines,"
"CommunityRating,OfficialRating,CumulativeRunTimeTicks,"
"Metascore,AirTime,DateCreated,MediaStreams,People,Overview,"
"CriticRating,CriticRatingSummary,Etag,ShortOverview,ProductionLocations,"
"Tags,ProviderIds,ParentId,RemoteTrailers,SpecialEpisodeNumbers"
)
}
2016-06-16 05:43:36 +00:00
url = "{server}/emby/LiveTv/Recordings/?userid={UserId}&format=json"
return self.doUtils(url, parameters=params)
2016-10-10 08:19:00 +00:00
def getSection(self, parentid, itemtype=None, sortby="SortName", artist_id=None, basic=False, dialog=None):
items = {
'Items': [],
'TotalRecordCount': 0
}
# Get total number of items
url = "{server}/emby/Users/{UserId}/Items?format=json"
params = {
'ParentId': parentid,
2016-10-10 08:19:00 +00:00
'ArtistIds': artist_id,
'IncludeItemTypes': itemtype,
2016-11-11 02:33:06 +00:00
'LocationTypes': "FileSystem,Remote,Offline",
'CollapseBoxSetItems': False,
'IsVirtualUnaired': False,
'IsMissing': False,
'Recursive': True,
'Limit': 1
}
result = self.doUtils(url, parameters=params)
try:
total = result['TotalRecordCount']
items['TotalRecordCount'] = total
except TypeError: # Failed to retrieve
log.debug("%s:%s Failed to retrieve the server response." % (url, params))
else:
index = 0
jump = self.limitIndex
throttled = False
highestjump = 0
while index < total:
# Get items by chunk to increase retrieval speed at scale
params = {
'ParentId': parentid,
2016-10-10 08:19:00 +00:00
'ArtistIds': artist_id,
'IncludeItemTypes': itemtype,
'CollapseBoxSetItems': False,
'IsVirtualUnaired': False,
2016-11-11 02:33:06 +00:00
'LocationTypes': "FileSystem,Remote,Offline",
'IsMissing': False,
'Recursive': True,
'StartIndex': index,
'Limit': jump,
'SortBy': sortby,
'SortOrder': "Ascending",
}
if basic:
params['Fields'] = "Etag"
else:
params['Fields'] = (
"Path,Genres,SortName,Studios,Writer,ProductionYear,Taglines,"
"CommunityRating,OfficialRating,CumulativeRunTimeTicks,"
"Metascore,AirTime,DateCreated,MediaStreams,People,Overview,"
"CriticRating,CriticRatingSummary,Etag,ShortOverview,ProductionLocations,"
"Tags,ProviderIds,ParentId,RemoteTrailers,SpecialEpisodeNumbers,"
2016-04-24 00:11:29 +00:00
"MediaSources,VoteCount"
)
try:
result = self.doUtils(url, parameters=params)
items['Items'].extend(result['Items'])
except Warning as error:
2016-10-05 00:42:01 +00:00
if "400" in error:
log.info("Something went wrong, aborting request.")
index += jump
except TypeError:
# Something happened to the connection
if not throttled:
throttled = True
log.info("Throttle activated.")
if jump == highestjump:
# We already tried with the highestjump, but it failed. Reset value.
log.info("Reset highest value.")
highestjump = 0
# Lower the number by half
if highestjump:
throttled = False
jump = highestjump
log.info("Throttle deactivated.")
else:
jump = int(jump/4)
log.debug("Set jump limit to recover: %s" % jump)
retry = 0
2016-06-16 05:43:36 +00:00
while window('emby_online') != "true":
# Wait server to come back online
if retry == 5:
log.info("Unable to reconnect to server. Abort process.")
return items
retry += 1
if xbmc.Monitor().waitForAbort(1):
# Abort was requested while waiting.
return items
else:
# Request succeeded
index += jump
if dialog:
percentage = int((float(index) / float(total))*100)
dialog.update(percentage)
if jump > highestjump:
# Adjust with the latest number, if it's greater
highestjump = jump
if throttled:
# We needed to adjust the number of item requested.
# keep increasing until the connection times out again
# to find the highest value
increment = int(jump*0.33)
if not increment: # Incase the increment is 0
increment = 10
jump += increment
log.info("Increase jump limit to: %s" % jump)
return items
2016-11-05 07:16:59 +00:00
def get_views(self, root=False):
if not root:
url = "{server}/emby/Users/{UserId}/Views?format=json"
else: # Views ungrouped
url = "{server}/emby/Users/{UserId}/Items?Sortby=SortName&format=json"
2016-11-05 07:16:59 +00:00
return self.doUtils(url)
def getViews(self, mediatype="", root=False, sortedlist=False):
# Build a list of user views
views = []
mediatype = mediatype.lower()
try:
2016-11-05 07:16:59 +00:00
items = self.get_views(root)['Items']
except TypeError:
log.debug("Error retrieving views for type: %s" % mediatype)
else:
for item in items:
if item['Type'] in ("Channel", "PlaylistsFolder"):
# Filter view types
continue
# 3/4/2016 OriginalCollectionType is added
itemtype = item.get('OriginalCollectionType', item.get('CollectionType', "mixed"))
if item['Name'] not in ('Collections', 'Trailers', 'Playlists'):
if sortedlist:
views.append({
2016-03-31 18:35:23 +00:00
'name': item['Name'],
'type': itemtype,
2016-03-31 18:35:23 +00:00
'id': item['Id']
})
elif (itemtype == mediatype or
(itemtype == "mixed" and mediatype in ("movies", "tvshows"))):
views.append({
2016-03-31 18:35:23 +00:00
'name': item['Name'],
'type': itemtype,
2016-03-31 18:35:23 +00:00
'id': item['Id']
})
return views
def verifyView(self, parentid, itemid):
belongs = False
params = {
'ParentId': parentid,
'CollapseBoxSetItems': False,
'IsVirtualUnaired': False,
2016-11-11 02:33:06 +00:00
'LocationTypes': "FileSystem,Remote,Offline",
'IsMissing': False,
'Recursive': True,
'Ids': itemid
}
2016-03-31 18:35:23 +00:00
result = self.doUtils("{server}/emby/Users/{UserId}/Items?format=json", parameters=params)
try:
total = result['TotalRecordCount']
except TypeError:
# Something happened to the connection
pass
else:
if total:
belongs = True
return belongs
def getMovies(self, parentId, basic=False, dialog=None):
2016-06-16 05:43:36 +00:00
2016-03-31 18:35:23 +00:00
return self.getSection(parentId, "Movie", basic=basic, dialog=dialog)
def getBoxset(self, dialog=None):
2016-06-16 05:43:36 +00:00
2016-03-31 18:35:23 +00:00
return self.getSection(None, "BoxSet", dialog=dialog)
def getMovies_byBoxset(self, boxsetid):
2016-03-31 18:35:23 +00:00
return self.getSection(boxsetid, "Movie")
def getMusicVideos(self, parentId, basic=False, dialog=None):
2016-06-16 05:43:36 +00:00
2016-03-31 18:35:23 +00:00
return self.getSection(parentId, "MusicVideo", basic=basic, dialog=dialog)
def getHomeVideos(self, parentId):
2016-03-31 18:35:23 +00:00
return self.getSection(parentId, "Video")
def getShows(self, parentId, basic=False, dialog=None):
2016-06-16 05:43:36 +00:00
2016-03-31 18:35:23 +00:00
return self.getSection(parentId, "Series", basic=basic, dialog=dialog)
def getSeasons(self, showId):
items = {
'Items': [],
'TotalRecordCount': 0
}
params = {
'IsVirtualUnaired': False,
'Fields': "Etag"
}
2016-06-16 05:43:36 +00:00
url = "{server}/emby/Shows/%s/Seasons?UserId={UserId}&format=json" % showId
result = self.doUtils(url, parameters=params)
if result:
items = result
return items
def getEpisodes(self, parentId, basic=False, dialog=None):
2016-03-31 18:35:23 +00:00
return self.getSection(parentId, "Episode", basic=basic, dialog=dialog)
def getEpisodesbyShow(self, showId):
2016-03-31 18:35:23 +00:00
return self.getSection(showId, "Episode")
def getEpisodesbySeason(self, seasonId):
2016-03-31 18:35:23 +00:00
return self.getSection(seasonId, "Episode")
2016-10-10 08:19:00 +00:00
def getArtists(self, parent_id=None, dialog=None):
items = {
'Items': [],
'TotalRecordCount': 0
}
# Get total number of items
url = "{server}/emby/Artists?UserId={UserId}&format=json"
params = {
2016-10-18 11:43:40 +00:00
'ParentId': parent_id,
'Recursive': True,
'Limit': 1
}
result = self.doUtils(url, parameters=params)
try:
total = result['TotalRecordCount']
items['TotalRecordCount'] = total
except TypeError: # Failed to retrieve
log.debug("%s:%s Failed to retrieve the server response." % (url, params))
else:
2016-10-10 08:19:00 +00:00
index = 0
jump = self.limitIndex
while index < total:
# Get items by chunk to increase retrieval speed at scale
params = {
2016-10-10 08:19:00 +00:00
'ParentId': parent_id,
'Recursive': True,
'IsVirtualUnaired': False,
2016-11-11 02:33:06 +00:00
'LocationTypes': "FileSystem,Remote,Offline",
'IsMissing': False,
'StartIndex': index,
'Limit': jump,
'SortBy': "SortName",
'SortOrder': "Ascending",
'Fields': (
"Etag,Genres,SortName,Studios,Writer,ProductionYear,"
"CommunityRating,OfficialRating,CumulativeRunTimeTicks,Metascore,"
"AirTime,DateCreated,MediaStreams,People,ProviderIds,Overview"
)
}
result = self.doUtils(url, parameters=params)
items['Items'].extend(result['Items'])
index += jump
if dialog:
percentage = int((float(index) / float(total))*100)
dialog.update(percentage)
return items
def getAlbums(self, basic=False, dialog=None):
2016-06-16 05:43:36 +00:00
2016-03-31 18:35:23 +00:00
return self.getSection(None, "MusicAlbum", sortby="DateCreated", basic=basic, dialog=dialog)
def getAlbumsbyArtist(self, artistId):
2016-06-16 05:43:36 +00:00
2016-10-10 08:19:00 +00:00
return self.getSection(None, "MusicAlbum", sortby="DateCreated", artist_id=artistId)
def getSongs(self, basic=False, dialog=None):
2016-06-16 05:43:36 +00:00
2016-03-31 18:35:23 +00:00
return self.getSection(None, "Audio", basic=basic, dialog=dialog)
def getSongsbyAlbum(self, albumId):
2016-06-16 05:43:36 +00:00
return self.getSection(albumId, "Audio")
def getAdditionalParts(self, itemId):
items = {
'Items': [],
'TotalRecordCount': 0
}
2016-06-16 05:43:36 +00:00
url = "{server}/emby/Videos/%s/AdditionalParts?UserId={UserId}&format=json" % itemId
result = self.doUtils(url)
if result:
items = result
return items
def sortby_mediatype(self, itemids):
sorted_items = {}
# Sort items
items = self.getFullItems(itemids)
for item in items:
mediatype = item.get('Type')
if mediatype:
sorted_items.setdefault(mediatype, []).append(item)
return sorted_items
def updateUserRating(self, itemid, favourite=None):
# Updates the user rating to Emby
2016-06-16 05:43:36 +00:00
doUtils = self.doUtils
if favourite:
2016-06-16 05:43:36 +00:00
url = "{server}/emby/Users/{UserId}/FavoriteItems/%s?format=json" % itemid
doUtils(url, action_type="POST")
elif not favourite:
2016-06-16 05:43:36 +00:00
url = "{server}/emby/Users/{UserId}/FavoriteItems/%s?format=json" % itemid
doUtils(url, action_type="DELETE")
2016-04-24 00:11:29 +00:00
else:
log.info("Error processing user rating.")
log.info("Update user rating to emby for itemid: %s | favourite: %s" % (itemid, favourite))
2016-06-20 00:24:42 +00:00
def refreshItem(self, itemid):
url = "{server}/emby/Items/%s/Refresh?format=json" % itemid
params = {
'Recursive': True,
'ImageRefreshMode': "FullRefresh",
'MetadataRefreshMode': "FullRefresh",
'ReplaceAllImages': False,
'ReplaceAllMetadata': True
}
self.doUtils(url, postBody=params, action_type="POST")
def deleteItem(self, itemid):
2016-06-21 20:26:42 +00:00
url = "{server}/emby/Items/%s?format=json" % itemid
self.doUtils(url, action_type="DELETE")
def getUsers(self, server):
url = "%s/emby/Users/Public?format=json" % server
users = self.doUtils(url, authenticate=False)
return users or []
def loginUser(self, server, username, password=None):
password = password or ""
url = "%s/emby/Users/AuthenticateByName?format=json" % server
data = {'username': username, 'password': hashlib.sha1(password).hexdigest()}
user = self.doUtils(url, postBody=data, action_type="POST", authenticate=False)
return user
def get_single_item(self, media_type, parent_id):
params = {
'ParentId': parent_id,
'Recursive': True,
'Limit': 1,
'IncludeItemTypes': media_type
}
url = self.get_emby_url('Users/{UserId}/Items?format=json')
return self.doUtils(url, parameters=params)