################################################################################################# # VideoNodes - utils to create video nodes listings in kodi for the emby addon ################################################################################################# import xbmc import xbmcgui import xbmcaddon import xbmcvfs import json import os import shutil import xml.etree.ElementTree as etree import Utils as utils from ReadEmbyDB import ReadEmbyDB WINDOW = xbmcgui.Window(10000) addonSettings = xbmcaddon.Addon(id='plugin.video.emby') language = addonSettings.getLocalizedString class VideoNodes(): def buildVideoNodeForView(self, tagname, type, windowPropId): #this method will build a video node for a particular Emby view (= tag in kodi) #we set some window props here to for easy future reference and to be used in skins (for easy access only) libraryPath = xbmc.translatePath("special://profile/library/video/Emby - %s/" %tagname) #create tag node - index xbmcvfs.mkdir(libraryPath) nodefile = os.path.join(libraryPath, "index.xml") root = etree.Element("node", {"order":"0"}) etree.SubElement(root, "label").text = "Emby - " + tagname etree.SubElement(root, "icon").text = "special://home/addons/plugin.video.emby/icon.png" path = "library://video/Emby - %s/"%tagname WINDOW.setProperty("Emby.nodes.%s.index" %str(windowPropId),path) etree.ElementTree(root).write(nodefile) #create tag node - all items nodefile = os.path.join(libraryPath, tagname + "_all.xml") root = etree.Element("node", {"order":"1", "type":"filter"}) etree.SubElement(root, "label").text = tagname etree.SubElement(root, "match").text = "all" etree.SubElement(root, "content").text = type etree.SubElement(root, "icon").text = "special://home/addons/plugin.video.emby/icon.png" etree.SubElement(root, "order", {"direction":"ascending"}).text = "sorttitle" Rule = etree.SubElement(root, "rule", {"field":"tag","operator":"is"}) WINDOW.setProperty("Emby.nodes.%s.title" %str(windowPropId),tagname) path = "library://video/Emby - %s/%s_all.xml"%(tagname,tagname) WINDOW.setProperty("Emby.nodes.%s.path" %str(windowPropId),"ActivateWindow(Video,%s,return)"%path) WINDOW.setProperty("Emby.nodes.%s.content" %str(windowPropId),path) WINDOW.setProperty("Emby.nodes.%s.type" %str(windowPropId),type) etree.SubElement(Rule, "value").text = tagname try: etree.ElementTree(root).write(nodefile, xml_declaration=True) except: etree.ElementTree(root).write(nodefile) #create tag node - recent items nodefile = os.path.join(libraryPath, tagname + "_recent.xml") root = etree.Element("node", {"order":"2", "type":"filter"}) if type == "tvshows": label = tagname + " - " + language(30170) else: label = tagname + " - " + language(30174) etree.SubElement(root, "label").text = label etree.SubElement(root, "match").text = "all" etree.SubElement(root, "content").text = type etree.SubElement(root, "icon").text = "special://home/addons/plugin.video.emby/icon.png" Rule = etree.SubElement(root, "rule", {"field":"tag","operator":"is"}) etree.SubElement(Rule, "value").text = tagname etree.SubElement(root, "order", {"direction":"descending"}).text = "dateadded" #set limit to 25 --> currently hardcoded --> TODO: add a setting for this ? etree.SubElement(root, "limit").text = "25" #exclude watched items --> currently hardcoded --> TODO: add a setting for this ? Rule2 = etree.SubElement(root, "rule", {"field":"playcount","operator":"is"}) etree.SubElement(Rule2, "value").text = "0" WINDOW.setProperty("Emby.nodes.%s.recent.title" %str(windowPropId),label) path = "library://video/Emby - %s/%s_recent.xml"%(tagname,tagname) WINDOW.setProperty("Emby.nodes.%s.recent.path" %str(windowPropId),"ActivateWindow(Video,%s,return)"%path) WINDOW.setProperty("Emby.nodes.%s.recent.content" %str(windowPropId),path) try: etree.ElementTree(root).write(nodefile, xml_declaration=True) except: etree.ElementTree(root).write(nodefile) #create tag node - inprogress items nodefile = os.path.join(libraryPath, tagname + "_progress.xml") root = etree.Element("node", {"order":"3", "type":"filter"}) if type == "tvshows": label = tagname + " - " + language(30171) else: label = tagname + " - " + language(30177) etree.SubElement(root, "label").text = label etree.SubElement(root, "match").text = "all" etree.SubElement(root, "content").text = type etree.SubElement(root, "icon").text = "special://home/addons/plugin.video.emby/icon.png" Rule = etree.SubElement(root, "rule", {"field":"tag","operator":"is"}) etree.SubElement(Rule, "value").text = tagname #set limit to 25 --> currently hardcoded --> TODO: add a setting for this ? etree.SubElement(root, "limit").text = "25" Rule2 = etree.SubElement(root, "rule", {"field":"inprogress","operator":"true"}) WINDOW.setProperty("Emby.nodes.%s.inprogress.title" %str(windowPropId),label) path = "library://video/Emby - %s/%s_progress.xml"%(tagname,tagname) WINDOW.setProperty("Emby.nodes.%s.inprogress.path" %str(windowPropId),"ActivateWindow(Video,%s,return)"%path) WINDOW.setProperty("Emby.nodes.%s.inprogress.content" %str(windowPropId),path) try: etree.ElementTree(root).write(nodefile, xml_declaration=True) except: etree.ElementTree(root).write(nodefile) #create tag node - add unwatched movies node for movies if type == "movies": nodefile = os.path.join(libraryPath, tagname + "_unwatched.xml") root = etree.Element("node", {"order":"4", "type":"filter"}) label = tagname + " - " + language(30189) etree.SubElement(root, "label").text = label etree.SubElement(root, "match").text = "all" etree.SubElement(root, "content").text = "movies" etree.SubElement(root, "icon").text = "special://home/addons/plugin.video.emby/icon.png" Rule = etree.SubElement(root, "rule", {"field":"tag","operator":"is"}) etree.SubElement(Rule, "value").text = tagname Rule = etree.SubElement(root, "rule", {"field":"playcount","operator":"is"}) etree.SubElement(Rule, "value").text = "0" etree.SubElement(root, "order", {"direction":"ascending"}).text = "sorttitle" #set limit to 25 --> currently hardcoded --> TODO: add a setting for this ? etree.SubElement(root, "limit").text = "25" #exclude watched items --> currently hardcoded --> TODO: add a setting for this ? Rule2 = etree.SubElement(root, "rule", {"field":"playcount","operator":"is"}) etree.SubElement(Rule2, "value").text = "0" WINDOW.setProperty("Emby.nodes.%s.unwatched.title" %str(windowPropId),label) path = "library://video/Emby - %s/%s_unwatched.xml"%(tagname,tagname) WINDOW.setProperty("Emby.nodes.%s.unwatched.path" %str(windowPropId),"ActivateWindow(Video,%s,return)"%path) WINDOW.setProperty("Emby.nodes.%s.unwatched.content" %str(windowPropId),path) try: etree.ElementTree(root).write(nodefile, xml_declaration=True) except: etree.ElementTree(root).write(nodefile) #add some additional nodes for episodes if type == "tvshows": #create tag node - recent episodes nodefile = os.path.join(libraryPath, tagname + "_recent_episodes.xml") root = etree.Element("node", {"order":"3", "type":"filter"}) label = tagname + " - " + language(30175) etree.SubElement(root, "label").text = label etree.SubElement(root, "match").text = "all" etree.SubElement(root, "content").text = "episodes" etree.SubElement(root, "icon").text = "special://home/addons/plugin.video.emby/icon.png" Rule = etree.SubElement(root, "rule", {"field":"tag","operator":"is"}) etree.SubElement(Rule, "value").text = tagname etree.SubElement(root, "order", {"direction":"descending"}).text = "dateadded" #set limit to 25 --> currently hardcoded --> TODO: add a setting for this ? etree.SubElement(root, "limit").text = "25" #exclude watched items --> currently hardcoded --> TODO: add a setting for this ? Rule2 = etree.SubElement(root, "rule", {"field":"playcount","operator":"is"}) etree.SubElement(Rule2, "value").text = "0" WINDOW.setProperty("Emby.nodes.%s.recentepisodes.title" %str(windowPropId),label) path = "library://video/Emby - %s/%s_recent_episodes.xml"%(tagname,tagname) WINDOW.setProperty("Emby.nodes.%s.recentepisodes.path" %str(windowPropId),"ActivateWindow(Video,%s,return)"%path) WINDOW.setProperty("Emby.nodes.%s.recentepisodes.content" %str(windowPropId),path) try: etree.ElementTree(root).write(nodefile, xml_declaration=True) except: etree.ElementTree(root).write(nodefile) #create tag node - inprogress items nodefile = os.path.join(libraryPath, tagname + "_progress_episodes.xml") root = etree.Element("node", {"order":"4", "type":"filter"}) label = tagname + " - " + language(30178) etree.SubElement(root, "label").text = label etree.SubElement(root, "match").text = "all" etree.SubElement(root, "content").text = "episodes" etree.SubElement(root, "icon").text = "special://home/addons/plugin.video.emby/icon.png" Rule = etree.SubElement(root, "rule", {"field":"tag","operator":"is"}) etree.SubElement(Rule, "value").text = tagname #set limit to 25 --> currently hardcoded --> TODO: add a setting for this ? etree.SubElement(root, "limit").text = "25" Rule2 = etree.SubElement(root, "rule", {"field":"inprogress","operator":"true"}) WINDOW.setProperty("Emby.nodes.%s.inprogressepisodes.title" %str(windowPropId),label) path = "library://video/Emby - %s/%s_progress_episodes.xml"%(tagname,tagname) WINDOW.setProperty("Emby.nodes.%s.inprogressepisodes.path" %str(windowPropId),"ActivateWindow(Video,%s,return)"%path) WINDOW.setProperty("Emby.nodes.%s.inprogressepisodes.content" %str(windowPropId),path) try: etree.ElementTree(root).write(nodefile, xml_declaration=True) except: etree.ElementTree(root).write(nodefile) #create tag node - nextup items nodefile = os.path.join(libraryPath, tagname + "_nextup_episodes.xml") root = etree.Element("node", {"order":"4", "type":"folder"}) label = tagname + " - " + language(30179) etree.SubElement(root, "label").text = label etree.SubElement(root, "content").text = "episodes" path = "plugin://plugin.video.emby/?id=%s&mode=nextup&limit=25" %tagname etree.SubElement(root, "path").text = path etree.SubElement(root, "icon").text = "special://home/addons/plugin.video.emby/icon.png" WINDOW.setProperty("Emby.nodes.%s.nextepisodes.title" %str(windowPropId),label) path = "library://video/Emby - %s/%s_nextup_episodes.xml"%(tagname,tagname) WINDOW.setProperty("Emby.nodes.%s.nextepisodes.path" %str(windowPropId),"ActivateWindow(Video,%s,return)"%path) WINDOW.setProperty("Emby.nodes.%s.nextepisodes.content" %str(windowPropId),path) try: etree.ElementTree(root).write(nodefile, xml_declaration=True) except: etree.ElementTree(root).write(nodefile) def buildVideoNodesListing(self): try: # the library path doesn't exist on all systems if not xbmcvfs.exists("special://profile/library/"): xbmcvfs.mkdir("special://profile/library") if not xbmcvfs.exists("special://profile/library/video/"): #we need to copy over the default items shutil.copytree(xbmc.translatePath("special://xbmc/system/library/video"), xbmc.translatePath("special://profile/library/video")) #always cleanup existing Emby video nodes first because we don't want old stuff to stay in there path = "special://profile/library/video/" if xbmcvfs.exists(path): allDirs, allFiles = xbmcvfs.listdir(path) for dir in allDirs: if dir.startswith("Emby "): shutil.rmtree(xbmc.translatePath("special://profile/library/video/" + dir)) for file in allFiles: if file.startswith("emby"): xbmcvfs.delete(path + file) #we build up a listing and set window props for all nodes we created #the window props will be used by the main entry point to quickly build up the listing and can be used in skins (like titan) too for quick reference #comment marcelveldt: please leave the window props as-is because I will be referencing them in titan skin... totalNodesCount = 0 #build the listing for all views views_movies = ReadEmbyDB().getCollections("movies") if views_movies: for view in views_movies: self.buildVideoNodeForView(view.get('title'), "movies", totalNodesCount) totalNodesCount +=1 views_shows = ReadEmbyDB().getCollections("tvshows") if views_shows: for view in views_shows: self.buildVideoNodeForView(view.get('title'), "tvshows", totalNodesCount) totalNodesCount +=1 #create tag node for emby channels nodefile = os.path.join(xbmc.translatePath("special://profile/library/video"), "emby_channels.xml") root = etree.Element("node", {"order":"1", "type":"folder"}) label = "Emby - " + language(30173) etree.SubElement(root, "label").text = label etree.SubElement(root, "content").text = "movies" etree.SubElement(root, "path").text = "plugin://plugin.video.emby/?id=0&mode=channels" etree.SubElement(root, "icon").text = "special://home/addons/plugin.video.emby/icon.png" WINDOW.setProperty("Emby.nodes.%s.title" %str(totalNodesCount),label) WINDOW.setProperty("Emby.nodes.%s.type" %str(totalNodesCount),"channels") path = "library://video/emby_channels.xml" WINDOW.setProperty("Emby.nodes.%s.path" %str(totalNodesCount),"ActivateWindow(Video,%s,return)"%path) WINDOW.setProperty("Emby.nodes.%s.content" %str(totalNodesCount),path) totalNodesCount +=1 try: etree.ElementTree(root).write(nodefile, xml_declaration=True) except: etree.ElementTree(root).write(nodefile) #create tag node - favorite shows nodefile = os.path.join(xbmc.translatePath("special://profile/library/video"),"emby_favorite_shows.xml") root = etree.Element("node", {"order":"1", "type":"filter"}) label = "Emby - " + language(30181) etree.SubElement(root, "label").text = label etree.SubElement(root, "match").text = "all" etree.SubElement(root, "content").text = "tvshows" etree.SubElement(root, "icon").text = "special://home/addons/plugin.video.emby/icon.png" etree.SubElement(root, "order", {"direction":"ascending"}).text = "sorttitle" Rule = etree.SubElement(root, "rule", {"field":"tag","operator":"is"}) etree.SubElement(Rule, "value").text = "Favorite tvshows" #do not localize the tagname itself WINDOW.setProperty("Emby.nodes.%s.title" %str(totalNodesCount),label) WINDOW.setProperty("Emby.nodes.%s.type" %str(totalNodesCount),"favourites") path = "library://video/emby_favorite_shows.xml" WINDOW.setProperty("Emby.nodes.%s.path" %str(totalNodesCount),"ActivateWindow(Video,%s,return)"%path) WINDOW.setProperty("Emby.nodes.%s.content" %str(totalNodesCount),path) totalNodesCount +=1 try: etree.ElementTree(root).write(nodefile, xml_declaration=True) except: etree.ElementTree(root).write(nodefile) #create tag node - favorite movies nodefile = os.path.join(xbmc.translatePath("special://profile/library/video"),"emby_favorite_movies.xml") root = etree.Element("node", {"order":"1", "type":"filter"}) label = "Emby - " + language(30180) etree.SubElement(root, "label").text = label etree.SubElement(root, "match").text = "all" etree.SubElement(root, "content").text = "movies" etree.SubElement(root, "icon").text = "special://home/addons/plugin.video.emby/icon.png" etree.SubElement(root, "order", {"direction":"ascending"}).text = "sorttitle" Rule = etree.SubElement(root, "rule", {"field":"tag","operator":"is"}) etree.SubElement(Rule, "value").text = "Favorite movies" #do not localize the tagname itself WINDOW.setProperty("Emby.nodes.%s.title" %str(totalNodesCount),label) WINDOW.setProperty("Emby.nodes.%s.type" %str(totalNodesCount),"favourites") path = "library://video/emby_favorite_movies.xml" WINDOW.setProperty("Emby.nodes.%s.path" %str(totalNodesCount),"ActivateWindow(Video,%s,return)"%path) WINDOW.setProperty("Emby.nodes.%s.content" %str(totalNodesCount),path) totalNodesCount +=1 try: etree.ElementTree(root).write(nodefile, xml_declaration=True) except: etree.ElementTree(root).write(nodefile) WINDOW.setProperty("Emby.nodes.total", str(totalNodesCount)) except Exception as e: utils.logMsg("Emby addon","Error while creating videonodes listings, restart required ?") print e