323 lines
18 KiB
Python
323 lines
18 KiB
Python
|
#################################################################################################
|
||
|
|
# 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
|
||
|
|
from xml.etree.ElementTree import Element, SubElement, Comment, tostring
|
||
|
|
from xml.etree import ElementTree
|
||
|
|
from xml.dom import minidom
|
||
|
|
import xml.etree.cElementTree as ET
|
||
|
|
|
||
|
|
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 = Element("node", {"order":"0"})
|
||
|
|
SubElement(root, "label").text = "Emby - " + tagname
|
||
|
|
SubElement(root, "icon").text = "special://home/addons/plugin.video.emby/icon.png"
|
||
|
|
try:
|
||
|
|
ET.ElementTree(root).write(nodefile, xml_declaration=True)
|
||
|
|
except:
|
||
|
|
ET.ElementTree(root).write(nodefile)
|
||
|
|
|
||
|
|
#create tag node - all items
|
||
|
|
nodefile = os.path.join(libraryPath, tagname + "_all.xml")
|
||
|
|
root = Element("node", {"order":"1", "type":"filter"})
|
||
|
|
SubElement(root, "label").text = tagname
|
||
|
|
SubElement(root, "match").text = "all"
|
||
|
|
SubElement(root, "content").text = type
|
||
|
|
SubElement(root, "icon").text = "special://home/addons/plugin.video.emby/icon.png"
|
||
|
|
SubElement(root, "order", {"direction":"ascending"}).text = "sorttitle"
|
||
|
|
Rule = 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)
|
||
|
|
SubElement(Rule, "value").text = tagname
|
||
|
|
try:
|
||
|
|
ET.ElementTree(root).write(nodefile, xml_declaration=True)
|
||
|
|
except:
|
||
|
|
ET.ElementTree(root).write(nodefile)
|
||
|
|
|
||
|
|
#create tag node - recent items
|
||
|
|
nodefile = os.path.join(libraryPath, tagname + "_recent.xml")
|
||
|
|
root = Element("node", {"order":"2", "type":"filter"})
|
||
|
|
if type == "tvshows":
|
||
|
|
label = tagname + " - " + language(30170)
|
||
|
|
else:
|
||
|
|
label = tagname + " - " + language(30174)
|
||
|
|
SubElement(root, "label").text = label
|
||
|
|
SubElement(root, "match").text = "all"
|
||
|
|
SubElement(root, "content").text = type
|
||
|
|
SubElement(root, "icon").text = "special://home/addons/plugin.video.emby/icon.png"
|
||
|
|
Rule = SubElement(root, "rule", {"field":"tag","operator":"is"})
|
||
|
|
SubElement(Rule, "value").text = tagname
|
||
|
|
SubElement(root, "order", {"direction":"descending"}).text = "dateadded"
|
||
|
|
#set limit to 25 --> currently hardcoded --> TODO: add a setting for this ?
|
||
|
|
SubElement(root, "limit").text = "25"
|
||
|
|
#exclude watched items --> currently hardcoded --> TODO: add a setting for this ?
|
||
|
|
Rule2 = SubElement(root, "rule", {"field":"playcount","operator":"is"})
|
||
|
|
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:
|
||
|
|
ET.ElementTree(root).write(nodefile, xml_declaration=True)
|
||
|
|
except:
|
||
|
|
ET.ElementTree(root).write(nodefile)
|
||
|
|
|
||
|
|
#create tag node - inprogress items
|
||
|
|
nodefile = os.path.join(libraryPath, tagname + "_progress.xml")
|
||
|
|
root = Element("node", {"order":"3", "type":"filter"})
|
||
|
|
if type == "tvshows":
|
||
|
|
label = tagname + " - " + language(30171)
|
||
|
|
else:
|
||
|
|
label = tagname + " - " + language(30177)
|
||
|
|
SubElement(root, "label").text = label
|
||
|
|
SubElement(root, "match").text = "all"
|
||
|
|
SubElement(root, "content").text = type
|
||
|
|
SubElement(root, "icon").text = "special://home/addons/plugin.video.emby/icon.png"
|
||
|
|
Rule = SubElement(root, "rule", {"field":"tag","operator":"is"})
|
||
|
|
SubElement(Rule, "value").text = tagname
|
||
|
|
#set limit to 25 --> currently hardcoded --> TODO: add a setting for this ?
|
||
|
|
SubElement(root, "limit").text = "25"
|
||
|
|
Rule2 = 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:
|
||
|
|
ET.ElementTree(root).write(nodefile, xml_declaration=True)
|
||
|
|
except:
|
||
|
|
ET.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 = Element("node", {"order":"4", "type":"filter"})
|
||
|
|
label = tagname + " - " + language(30189)
|
||
|
|
SubElement(root, "label").text = label
|
||
|
|
SubElement(root, "match").text = "all"
|
||
|
|
SubElement(root, "content").text = "movies"
|
||
|
|
SubElement(root, "icon").text = "special://home/addons/plugin.video.emby/icon.png"
|
||
|
|
Rule = SubElement(root, "rule", {"field":"tag","operator":"is"})
|
||
|
|
SubElement(Rule, "value").text = tagname
|
||
|
|
Rule = SubElement(root, "rule", {"field":"playcount","operator":"is"})
|
||
|
|
SubElement(Rule, "value").text = "0"
|
||
|
|
SubElement(root, "order", {"direction":"ascending"}).text = "sorttitle"
|
||
|
|
#set limit to 25 --> currently hardcoded --> TODO: add a setting for this ?
|
||
|
|
SubElement(root, "limit").text = "25"
|
||
|
|
#exclude watched items --> currently hardcoded --> TODO: add a setting for this ?
|
||
|
|
Rule2 = SubElement(root, "rule", {"field":"playcount","operator":"is"})
|
||
|
|
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:
|
||
|
|
ET.ElementTree(root).write(nodefile, xml_declaration=True)
|
||
|
|
except:
|
||
|
|
ET.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 = Element("node", {"order":"3", "type":"filter"})
|
||
|
|
label = tagname + " - " + language(30175)
|
||
|
|
SubElement(root, "label").text = label
|
||
|
|
SubElement(root, "match").text = "all"
|
||
|
|
SubElement(root, "content").text = "episodes"
|
||
|
|
SubElement(root, "icon").text = "special://home/addons/plugin.video.emby/icon.png"
|
||
|
|
Rule = SubElement(root, "rule", {"field":"tag","operator":"is"})
|
||
|
|
SubElement(Rule, "value").text = tagname
|
||
|
|
SubElement(root, "order", {"direction":"descending"}).text = "dateadded"
|
||
|
|
#set limit to 25 --> currently hardcoded --> TODO: add a setting for this ?
|
||
|
|
SubElement(root, "limit").text = "25"
|
||
|
|
#exclude watched items --> currently hardcoded --> TODO: add a setting for this ?
|
||
|
|
Rule2 = SubElement(root, "rule", {"field":"playcount","operator":"is"})
|
||
|
|
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:
|
||
|
|
ET.ElementTree(root).write(nodefile, xml_declaration=True)
|
||
|
|
except:
|
||
|
|
ET.ElementTree(root).write(nodefile)
|
||
|
|
|
||
|
|
#create tag node - inprogress items
|
||
|
|
nodefile = os.path.join(libraryPath, tagname + "_progress_episodes.xml")
|
||
|
|
root = Element("node", {"order":"4", "type":"filter"})
|
||
|
|
label = tagname + " - " + language(30178)
|
||
|
|
SubElement(root, "label").text = label
|
||
|
|
SubElement(root, "match").text = "all"
|
||
|
|
SubElement(root, "content").text = "episodes"
|
||
|
|
SubElement(root, "icon").text = "special://home/addons/plugin.video.emby/icon.png"
|
||
|
|
Rule = SubElement(root, "rule", {"field":"tag","operator":"is"})
|
||
|
|
SubElement(Rule, "value").text = tagname
|
||
|
|
#set limit to 25 --> currently hardcoded --> TODO: add a setting for this ?
|
||
|
|
SubElement(root, "limit").text = "25"
|
||
|
|
Rule2 = 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:
|
||
|
|
ET.ElementTree(root).write(nodefile, xml_declaration=True)
|
||
|
|
except:
|
||
|
|
ET.ElementTree(root).write(nodefile)
|
||
|
|
|
||
|
|
#create tag node - nextup items
|
||
|
|
nodefile = os.path.join(libraryPath, tagname + "_nextup_episodes.xml")
|
||
|
|
root = Element("node", {"order":"4", "type":"folder"})
|
||
|
|
label = tagname + " - " + language(30179)
|
||
|
|
SubElement(root, "label").text = label
|
||
|
|
SubElement(root, "content").text = "episodes"
|
||
|
|
path = "plugin://plugin.video.emby/?id=%s&mode=nextup&limit=25" %tagname
|
||
|
|
SubElement(root, "path").text = path
|
||
|
|
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:
|
||
|
|
ET.ElementTree(root).write(nodefile, xml_declaration=True)
|
||
|
|
except:
|
||
|
|
ET.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 = Element("node", {"order":"1", "type":"folder"})
|
||
|
|
label = "Emby - " + language(30173)
|
||
|
|
SubElement(root, "label").text = label
|
||
|
|
SubElement(root, "content").text = "movies"
|
||
|
|
SubElement(root, "path").text = "plugin://plugin.video.emby/?id=0&mode=channels"
|
||
|
|
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:
|
||
|
|
ET.ElementTree(root).write(nodefile, xml_declaration=True)
|
||
|
|
except:
|
||
|
|
ET.ElementTree(root).write(nodefile)
|
||
|
|
|
||
|
|
#create tag node - favorite shows
|
||
|
|
nodefile = os.path.join(xbmc.translatePath("special://profile/library/video"),"emby_favorite_shows.xml")
|
||
|
|
root = Element("node", {"order":"1", "type":"filter"})
|
||
|
|
label = "Emby - " + language(30181)
|
||
|
|
SubElement(root, "label").text = label
|
||
|
|
SubElement(root, "match").text = "all"
|
||
|
|
SubElement(root, "content").text = "tvshows"
|
||
|
|
SubElement(root, "icon").text = "special://home/addons/plugin.video.emby/icon.png"
|
||
|
|
SubElement(root, "order", {"direction":"ascending"}).text = "sorttitle"
|
||
|
|
Rule = SubElement(root, "rule", {"field":"tag","operator":"is"})
|
||
|
|
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:
|
||
|
|
ET.ElementTree(root).write(nodefile, xml_declaration=True)
|
||
|
|
except:
|
||
|
|
ET.ElementTree(root).write(nodefile)
|
||
|
|
|
||
|
|
#create tag node - favorite movies
|
||
|
|
nodefile = os.path.join(xbmc.translatePath("special://profile/library/video"),"emby_favorite_movies.xml")
|
||
|
|
root = Element("node", {"order":"1", "type":"filter"})
|
||
|
|
label = "Emby - " + language(30180)
|
||
|
|
SubElement(root, "label").text = label
|
||
|
|
SubElement(root, "match").text = "all"
|
||
|
|
SubElement(root, "content").text = "movies"
|
||
|
|
SubElement(root, "icon").text = "special://home/addons/plugin.video.emby/icon.png"
|
||
|
|
SubElement(root, "order", {"direction":"ascending"}).text = "sorttitle"
|
||
|
|
Rule = SubElement(root, "rule", {"field":"tag","operator":"is"})
|
||
|
|
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:
|
||
|
|
ET.ElementTree(root).write(nodefile, xml_declaration=True)
|
||
|
|
except:
|
||
|
|
ET.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
|
||
|
|
|