196 lines
11 KiB
Python
196 lines
11 KiB
Python
#################################################################################################
|
|
# ReadKodiDB
|
|
#################################################################################################
|
|
|
|
|
|
import xbmc
|
|
import xbmcgui
|
|
import xbmcaddon
|
|
import json
|
|
import os
|
|
|
|
addon = xbmcaddon.Addon(id='plugin.video.mb3sync')
|
|
addondir = xbmc.translatePath(addon.getAddonInfo('profile'))
|
|
dataPath = os.path.join(addondir,"library")
|
|
movieLibrary = os.path.join(dataPath,'movies')
|
|
tvLibrary = os.path.join(dataPath,'tvshows')
|
|
|
|
#sleepval is used to throttle the calls to the xbmc json API
|
|
sleepVal = 15
|
|
|
|
class ReadKodiDB():
|
|
def getKodiMovie(self, id):
|
|
#returns a single movie from Kodi db selected on MB item ID
|
|
xbmc.sleep(sleepVal)
|
|
json_response = xbmc.executeJSONRPC('{"jsonrpc": "2.0", "method": "VideoLibrary.GetMovies", "params": { "filter": {"operator": "contains", "field": "path", "value": "' + id + '"}, "properties" : ["art", "rating", "thumbnail", "resume", "runtime", "year", "genre", "cast", "trailer", "country", "studio", "set", "imdbnumber", "mpaa", "tagline", "plotoutline","plot", "sorttitle", "director", "writer", "playcount", "tag", "file"], "sort": { "order": "ascending", "method": "label", "ignorearticle": true } }, "id": "libMovies"}')
|
|
jsonobject = json.loads(json_response.decode('utf-8','replace'))
|
|
movie = None
|
|
|
|
if(jsonobject.has_key('result')):
|
|
result = jsonobject['result']
|
|
if(result.has_key('movies')):
|
|
movies = result['movies']
|
|
movie = movies[0]
|
|
|
|
return movie
|
|
|
|
def getKodiMovies(self,fullInfo = False):
|
|
#returns all movies in Kodi db inserted by MB
|
|
xbmc.sleep(sleepVal)
|
|
if fullInfo:
|
|
json_response = xbmc.executeJSONRPC('{"jsonrpc": "2.0", "method": "VideoLibrary.GetMovies", "params": { "filter": {"operator": "contains", "field": "path", "value": "plugin.video.mb3sync"}, "properties" : ["art", "rating", "thumbnail", "resume", "runtime", "year", "genre", "cast", "trailer", "country", "studio", "set", "imdbnumber", "mpaa", "tagline", "plotoutline","plot", "sorttitle", "director", "writer", "playcount", "tag", "file"] }, "id": "libMovies"}')
|
|
else:
|
|
json_response = xbmc.executeJSONRPC('{"jsonrpc": "2.0", "method": "VideoLibrary.GetMovies", "params": { "filter": {"operator": "contains", "field": "path", "value": "plugin.video.mb3sync"}, "properties" : ["resume", "playcount", "file"] }, "id": "libMovies"}')
|
|
jsonobject = json.loads(json_response.decode('utf-8','replace'))
|
|
movies = None
|
|
|
|
if(jsonobject.has_key('result')):
|
|
result = jsonobject['result']
|
|
if(result.has_key('movies')):
|
|
movies = result['movies']
|
|
|
|
kodiMovieMap = None
|
|
if(movies != None and len(movies) > 0):
|
|
kodiMovieMap = {}
|
|
for kodimovie in movies:
|
|
key = kodimovie["file"][-37:-5] #extract the id from the file name
|
|
kodiMovieMap[key] = kodimovie
|
|
|
|
return kodiMovieMap
|
|
|
|
def getKodiMoviesIds(self,returnMB3Ids = False):
|
|
# returns a list of movieIds or MB3 Id's from all movies currently in the Kodi library
|
|
allKodiMovies = self.getKodiMovies(False)
|
|
|
|
if(allKodiMovies == None):
|
|
return list()
|
|
|
|
if(returnMB3Ids):
|
|
allKodiMovieIds = list(allKodiMovies.keys())
|
|
return allKodiMovieIds
|
|
else:
|
|
allKodiMovieIds = list()
|
|
for kodimovie in allKodiMovies.values():
|
|
id = str(kodimovie["movieid"])
|
|
allKodiMovieIds.append(id)
|
|
|
|
return allKodiMovieIds
|
|
|
|
def getKodiTvShowsIds(self,returnMB3Ids = False):
|
|
# returns a list of tvshowIds or MB3 Id's from all tvshows currently in the Kodi library
|
|
allKodiTvShows = self.getKodiTvShows(False)
|
|
|
|
if allKodiTvShows == None:
|
|
return list()
|
|
|
|
if(returnMB3Ids):
|
|
allKodiTvShowsIds = list(allKodiTvShows.keys())
|
|
return allKodiTvShowsIds
|
|
else:
|
|
allKodiTvShowsIds = list()
|
|
for kodishow in allKodiTvShows.values():
|
|
id = str(kodishow["tvshowid"])
|
|
allKodiTvShowsIds.append(id)
|
|
|
|
return allKodiTvShowsIds
|
|
|
|
def getKodiTvShows(self,fullInfo = False):
|
|
#returns all tvshows in Kodi db inserted by MB
|
|
xbmc.sleep(sleepVal)
|
|
if fullInfo:
|
|
json_response = xbmc.executeJSONRPC('{"jsonrpc": "2.0", "method": "VideoLibrary.GetTVShows", "params": { "filter": {"operator": "contains", "field": "path", "value": "plugin.video.mb3sync"}, "properties": ["art", "genre", "plot", "mpaa", "cast", "studio", "sorttitle", "title", "originaltitle", "imdbnumber", "year", "premiered", "rating", "thumbnail", "playcount", "file", "fanart"], "sort": { "order": "ascending", "method": "label", "ignorearticle": true } }, "id": "libTvShows"}')
|
|
else:
|
|
json_response = xbmc.executeJSONRPC('{"jsonrpc": "2.0", "method": "VideoLibrary.GetTVShows", "params": { "filter": {"operator": "contains", "field": "path", "value": "plugin.video.mb3sync"}, "properties": ["sorttitle", "title", "playcount", "file"], "sort": { "order": "ascending", "method": "label", "ignorearticle": true } }, "id": "libTvShows"}')
|
|
jsonobject = json.loads(json_response.decode('utf-8','replace'))
|
|
tvshows = None
|
|
|
|
if(jsonobject.has_key('result')):
|
|
result = jsonobject['result']
|
|
if(result.has_key('tvshows')):
|
|
tvshows = result['tvshows']
|
|
|
|
kodiShowMap = None
|
|
if(tvshows != None and len(tvshows) > 0):
|
|
kodiShowMap = {}
|
|
for kodishow in tvshows:
|
|
filepath = kodishow["file"]
|
|
filepath = filepath.replace(tvLibrary + os.sep, "")
|
|
filepath = filepath.replace(".strm", "")
|
|
filepath = filepath.split(os.sep)[0]
|
|
key = filepath #extract the id from the file path
|
|
kodiShowMap[key] = kodishow
|
|
|
|
return kodiShowMap
|
|
|
|
def getKodiTVShow(self, id):
|
|
xbmc.sleep(sleepVal)
|
|
json_response = xbmc.executeJSONRPC('{"jsonrpc": "2.0", "method": "VideoLibrary.GetTVShows", "params": { "filter": {"operator": "contains", "field": "path", "value": "' + id + '"}, "properties": ["art", "genre", "plot", "mpaa", "cast", "studio", "sorttitle", "title", "originaltitle", "imdbnumber", "year", "premiered", "rating", "thumbnail", "playcount", "file", "fanart"], "sort": { "order": "ascending", "method": "label", "ignorearticle": true } }, "id": "libTvShows"}')
|
|
jsonobject = json.loads(json_response.decode('utf-8','replace'))
|
|
tvshow = None
|
|
if(jsonobject.has_key('result')):
|
|
result = jsonobject['result']
|
|
if(result.has_key('tvshows')):
|
|
tvshows = result['tvshows']
|
|
tvshow = tvshows[0]
|
|
return tvshow
|
|
|
|
def getKodiEpisodes(self, id, fullInfo = True, returnmap = True):
|
|
xbmc.sleep(sleepVal)
|
|
episodes = None
|
|
json_response = xbmc.executeJSONRPC('{"jsonrpc": "2.0", "method": "VideoLibrary.GetTVShows", "params": { "filter": {"operator": "contains", "field": "path", "value": "' + id + '"}, "properties": ["title", "file"], "sort": { "order": "ascending", "method": "label", "ignorearticle": true } }, "id": "libTvShows"}')
|
|
jsonobject = json.loads(json_response.decode('utf-8','replace'))
|
|
tvshow = None
|
|
if(jsonobject.has_key('result')):
|
|
result = jsonobject['result']
|
|
if(result.has_key('tvshows')):
|
|
tvshows = result['tvshows']
|
|
tvshow = tvshows[0]
|
|
if fullInfo:
|
|
json_response = xbmc.executeJSONRPC('{"jsonrpc": "2.0", "method": "VideoLibrary.GetEpisodes", "params": {"tvshowid": %d, "properties": ["title", "playcount", "plot", "season", "episode", "showtitle", "file", "lastplayed", "rating", "resume", "art", "streamdetails", "firstaired", "runtime", "writer", "cast", "dateadded","uniqueid"], "sort": {"method": "episode"}}, "id": 1}' %tvshow['tvshowid'])
|
|
else:
|
|
json_response = xbmc.executeJSONRPC('{"jsonrpc": "2.0", "method": "VideoLibrary.GetEpisodes", "params": {"tvshowid": %d, "properties": ["title", "playcount", "season", "episode", "lastplayed", "resume","file","uniqueid"], "sort": {"method": "episode"}}, "id": 1}' %tvshow['tvshowid'])
|
|
jsonobject = json.loads(json_response.decode('utf-8','replace'))
|
|
episodes = None
|
|
if(jsonobject.has_key('result')):
|
|
result = jsonobject['result']
|
|
if(result.has_key('episodes')):
|
|
episodes = result['episodes']
|
|
if returnmap:
|
|
episodeMap = None
|
|
if(episodes != None):
|
|
episodeMap = {}
|
|
for KodiItem in episodes:
|
|
key = str(KodiItem["season"]) + "-" + str(KodiItem["episode"])
|
|
episodeMap[key] = KodiItem
|
|
return episodeMap
|
|
else:
|
|
return episodes
|
|
|
|
def getKodiEpisodeByMbItem(self, MBitem):
|
|
xbmc.sleep(sleepVal)
|
|
json_response = xbmc.executeJSONRPC('{"jsonrpc": "2.0", "method": "VideoLibrary.GetTVShows", "params": { "filter": {"operator": "is", "field": "title", "value": "' + MBitem.get("SeriesName").encode('utf-8') + '"} }, "id": "libTvShows"}')
|
|
jsonobject = json.loads(json_response.decode('utf-8','replace'))
|
|
episode = None
|
|
if(jsonobject.has_key('result')):
|
|
result = jsonobject['result']
|
|
if(result.has_key('tvshows')):
|
|
tvshows = result['tvshows']
|
|
tvshow = tvshows[0]
|
|
|
|
# find the episode by combination of season and episode
|
|
json_response = xbmc.executeJSONRPC('{"jsonrpc": "2.0", "method": "VideoLibrary.GetEpisodes", "params": {"tvshowid": %d, "properties": ["playcount","season", "resume", "episode"], "sort": {"method": "episode"}}, "id": 1}' %tvshow['tvshowid'])
|
|
jsonobject = json.loads(json_response.decode('utf-8','replace'))
|
|
episodes = None
|
|
if(jsonobject.has_key('result')):
|
|
result = jsonobject['result']
|
|
if(result.has_key('episodes')):
|
|
episodes = result['episodes']
|
|
|
|
comparestring1 = str(MBitem.get("ParentIndexNumber")) + "-" + str(MBitem.get("IndexNumber"))
|
|
for item in episodes:
|
|
comparestring2 = str(item["season"]) + "-" + str(item["episode"])
|
|
if comparestring1 == comparestring2:
|
|
episode = item
|
|
|
|
return episode
|
|
|