From 3439a599e9d50e2802179613115f7a221d934190 Mon Sep 17 00:00:00 2001 From: im85288 Date: Sat, 14 Mar 2015 22:43:36 +0000 Subject: [PATCH] start of getting tv shows - needs further work --- resources/lib/LibrarySync.py | 154 +++++++++++++++++++++++++++++++++++ 1 file changed, 154 insertions(+) diff --git a/resources/lib/LibrarySync.py b/resources/lib/LibrarySync.py index 507a6a79..87b71d38 100644 --- a/resources/lib/LibrarySync.py +++ b/resources/lib/LibrarySync.py @@ -52,6 +52,21 @@ class LibrarySync(): updateNeeded = True else: self.updateMovieToKodiLibrary(item, kodiItem) + + allTVShows = list() + tvShowData = self.getTVShows(True) + + if (tvShowData == None): + return + for item in tvShowData: + if item.get('IsFolder'): + kodiItem = self.getKodiTVShow(item["Id"]) + allTVShows.append(item["Id"]) + if kodiItem == None: + self.addTVShowToKodiLibrary(item) + updateNeeded = True + else: + self.updateTVShowToKodiLibrary(item, kodiItem) cleanNeeded = False # process deletes @@ -62,6 +77,13 @@ class LibrarySync(): self.deleteMovieFromKodiLibrary(dir) cleanneeded = True + allLocaldirs, filesTVShows = xbmcvfs.listdir(tvLibrary) + allMB3TVShows = set(allTVShows) + for dir in allLocaldirs: + if not dir in allMB3TVShows: + self.deleteTVShowFromKodiLibrary(dir) + cleanneeded = True + if cleanNeeded: xbmc.executebuiltin("CleanLibrary(video)") @@ -125,6 +147,30 @@ class LibrarySync(): return result + def getTVShows(self, fullinfo = False): + result = None + + addon = xbmcaddon.Addon(id='plugin.video.mb3sync') + port = addon.getSetting('port') + host = addon.getSetting('ipaddress') + server = host + ":" + port + + downloadUtils = DownloadUtils() + userid = downloadUtils.getUserId() + + if fullinfo: + url = server + '/mediabrowser/Users/' + userid + '/Items?&SortBy=SortName&Fields=Path,Genres,SortName,Studios,Writer,ProductionYear,Taglines,CommunityRating,OfficialRating,CumulativeRunTimeTicks,Metascore,AirTime,DateCreated,MediaStreams,People,Overview&Recursive=true&SortOrder=Ascending&IncludeItemTypes=Series&format=json&ImageTypeLimit=1' + else: + url = server + '/mediabrowser/Users/' + userid + '/Items?&SortBy=SortName&Fields=CumulativeRunTimeTicks&Recursive=true&SortOrder=Ascending&IncludeItemTypes=Series&format=json&ImageTypeLimit=1' + + jsonData = downloadUtils.downloadUrl(url, suppress=True, popup=0) + if jsonData != None and jsonData != "": + result = json.loads(jsonData) + if(result.has_key('Items')): + result = result['Items'] + + return result + def updatePlayCountFromKodi(self, id, playcount=0): #when user marks item watched from kodi interface update this to MB3 @@ -235,6 +281,76 @@ class LibrarySync(): if KodiItem['playcount'] != int(userData.get("PlayCount")): xbmc.executeJSONRPC('{"jsonrpc": "2.0", "method": "VideoLibrary.SetMovieDetails", "params": { "movieid": %i, "playcount": %i}, "id": 1 }' %(KodiItem['movieid'], int(userData.get("PlayCount")))) + def updateTVShowToKodiLibrary( self, MBitem, KodiItem ): + + addon = xbmcaddon.Addon(id='plugin.video.mb3sync') + port = addon.getSetting('port') + host = addon.getSetting('ipaddress') + server = host + ":" + port + downloadUtils = DownloadUtils() + + timeInfo = API().getTimeInfo(MBitem) + userData=API().getUserData(MBitem) + people = API().getPeople(MBitem) + genre = API().getGenre(MBitem) + studios = API().getStudios(MBitem) + mediaStreams=API().getMediaStreams(MBitem) + + thumbPath = API().getArtwork(MBitem, "Primary") + + utils.logMsg("Updating item to Kodi Library", MBitem["Id"] + " - " + MBitem["Name"]) + + #update artwork + self.updateArtWork(KodiItem,"poster", API().getArtwork(MBitem, "poster"),"tvshow") + self.updateArtWork(KodiItem,"clearlogo", API().getArtwork(MBitem, "Logo"),"tvshow") + self.updateArtWork(KodiItem,"clearart", API().getArtwork(MBitem, "Art"),"tvshow") + self.updateArtWork(KodiItem,"banner", API().getArtwork(MBitem, "Banner"),"tvshow") + self.updateArtWork(KodiItem,"landscape", API().getArtwork(MBitem, "Thumb"),"tvshow") + self.updateArtWork(KodiItem,"discart", API().getArtwork(MBitem, "Disc"),"tvshow") + self.updateArtWork(KodiItem,"fanart", API().getArtwork(MBitem, "Backdrop"),"tvshow") + + #update common properties + duration = (int(timeInfo.get('Duration'))*60) + self.updateProperty(KodiItem,"runtime",duration,"tvshow") + self.updateProperty(KodiItem,"year",MBitem.get("ProductionYear"),"tvshow") + self.updateProperty(KodiItem,"mpaa",MBitem.get("OfficialRating"),"tvshow") + + if MBitem.get("CriticRating") != None: + self.updateProperty(KodiItem,"rating",int(MBitem.get("CriticRating"))/10,"tvshow") + + self.updateProperty(KodiItem,"plotoutline",MBitem.get("ShortOverview"),"tvshow") + self.updateProperty(KodiItem,"sorttitle",MBitem.get("SortName"),"tvshow") + + if MBitem.get("ProviderIds") != None: + if MBitem.get("ProviderIds").get("Imdb") != None: + self.updateProperty(KodiItem,"imdbnumber",MBitem.get("ProviderIds").get("Imdb"),"tvshow") + + # FIXME --> Taglines not returned by MB3 server !? + if MBitem.get("TagLines") != None: + self.updateProperty(KodiItem,"tagline",MBitem.get("TagLines")[0],"tvshow") + + self.updatePropertyArray(KodiItem,"writer",people.get("Writer"),"tvshow") + self.updatePropertyArray(KodiItem,"director",people.get("Director"),"tvshow") + self.updatePropertyArray(KodiItem,"genre",MBitem.get("Genres"),"tvshow") + self.updatePropertyArray(KodiItem,"studio",studios,"tvshow") + + # FIXME --> ProductionLocations not returned by MB3 server !? + self.updatePropertyArray(KodiItem,"country",MBitem.get("ProductionLocations"),"tvshow") + + #update strm file - TODO: only update strm when path has changed + self.createSTRM(MBitem["Id"]) + + #add actors + self.AddActorsToMedia(KodiItem,MBitem.get("People"),"tvshow") + + #create nfo file if not exists + nfoFile = os.path.join(tvLibrary,MBitem["Id"],MBitem["Id"] + ".nfo") + if not xbmcvfs.exists(nfoFile): + self.createTVNFO(MBitem) + + #update playcounts + if KodiItem['playcount'] != int(userData.get("PlayCount")): + xbmc.executeJSONRPC('{"jsonrpc": "2.0", "method": "VideoLibrary.SetTVShowDetails", "params": { "tvshowid": %i, "playcount": %i}, "id": 1 }' %(KodiItem['tvshowid'], int(userData.get("PlayCount")))) # adds or updates artwork to the given Kodi file in database def updateArtWork(self,KodiItem,artWorkName,artworkValue, fileType="movie"): @@ -378,6 +494,31 @@ class LibrarySync(): path = os.path.join(movieLibrary,id) xbmcvfs.rmdir(path) + + def addTVShowToKodiLibrary( self, item ): + itemPath = os.path.join(tvLibrary,item["Id"]) + strmFile = os.path.join(itemPath,item["Id"] + ".strm") + + utils.logMsg("Adding item to Kodi Library",item["Id"] + " - " + item["Name"]) + + #create path if not exists + if not xbmcvfs.exists(itemPath): + xbmcvfs.mkdir(itemPath) + + #create nfo file + self.createTVNFO(item) + + # create strm file + self.createSTRM(item["Id"]) + + def deleteTVShowFromKodiLibrary(self, id ): + kodiItem = self.getKodiTVShow(id) + utils.logMsg("deleting tvshow from Kodi library",id) + if kodiItem != None: + xbmc.executeJSONRPC('{"jsonrpc": "2.0", "method": "VideoLibrary.RemoveTVShow", "params": { "tvshowid": %i}, "id": 1 }' %(kodiItem["tvshowid"])) + + path = os.path.join(tvLibrary,id) + xbmcvfs.rmdir(path) def setKodiResumePoint(self, id, resume_seconds, total_seconds): #use sqlite to set the resume point while json api doesn't support this yet @@ -459,3 +600,16 @@ class LibrarySync(): movie = movies[0] return movie + + def getKodiTVShow(self, id): + json_response = xbmc.executeJSONRPC('{"jsonrpc": "2.0", "method": "VideoLibrary.GetTVShows", "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", "file"], "sort": { "order": "ascending", "method": "label", "ignorearticle": true } }, "id": "libMovies"}') + 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