mirror of
https://github.com/jellyfin/jellyfin-kodi.git
synced 2024-11-13 21:56:11 +00:00
392 lines
No EOL
16 KiB
Python
392 lines
No EOL
16 KiB
Python
# -*- coding: utf-8 -*-
|
|
|
|
#################################################################################################
|
|
|
|
import logging
|
|
import shutil
|
|
import xml.etree.ElementTree as etree
|
|
|
|
import xbmc
|
|
import xbmcaddon
|
|
import xbmcvfs
|
|
|
|
import utils
|
|
from utils import window, language as lang
|
|
|
|
#################################################################################################
|
|
|
|
log = logging.getLogger("EMBY."+__name__)
|
|
|
|
#################################################################################################
|
|
|
|
|
|
class VideoNodes(object):
|
|
|
|
|
|
def __init__(self):
|
|
|
|
self.kodiversion = int(xbmc.getInfoLabel('System.BuildVersion')[:2])
|
|
|
|
|
|
def commonRoot(self, order, label, tagname, roottype=1):
|
|
|
|
if roottype == 0:
|
|
# Index
|
|
root = etree.Element('node', attrib={'order': "%s" % order})
|
|
elif roottype == 1:
|
|
# Filter
|
|
root = etree.Element('node', attrib={'order': "%s" % order, 'type': "filter"})
|
|
etree.SubElement(root, 'match').text = "all"
|
|
# Add tag rule
|
|
rule = etree.SubElement(root, 'rule', attrib={'field': "tag", 'operator': "is"})
|
|
etree.SubElement(rule, 'value').text = tagname
|
|
else:
|
|
# Folder
|
|
root = etree.Element('node', attrib={'order': "%s" % order, 'type': "folder"})
|
|
|
|
etree.SubElement(root, 'label').text = label
|
|
etree.SubElement(root, 'icon').text = "special://home/addons/plugin.video.emby/icon.png"
|
|
|
|
return root
|
|
|
|
def viewNode(self, indexnumber, tagname, mediatype, viewtype, viewid, delete=False):
|
|
|
|
if viewtype == "mixed":
|
|
dirname = "%s - %s" % (viewid, mediatype)
|
|
else:
|
|
dirname = viewid
|
|
|
|
path = xbmc.translatePath("special://profile/library/video/").decode('utf-8')
|
|
nodepath = xbmc.translatePath(
|
|
"special://profile/library/video/emby/%s/" % dirname).decode('utf-8')
|
|
|
|
# Verify the video directory
|
|
if not xbmcvfs.exists(path):
|
|
try:
|
|
shutil.copytree(
|
|
src=xbmc.translatePath("special://xbmc/system/library/video").decode('utf-8'),
|
|
dst=xbmc.translatePath("special://profile/library/video").decode('utf-8'))
|
|
except Exception as error:
|
|
log.error(error)
|
|
|
|
xbmcvfs.exists(path)
|
|
|
|
if delete:
|
|
dirs, files = xbmcvfs.listdir(nodepath)
|
|
for file in files:
|
|
xbmcvfs.delete(nodepath + file)
|
|
|
|
log.info("Sucessfully removed videonode: %s." % tagname)
|
|
return
|
|
# Create the node directory
|
|
if not xbmcvfs.exists(nodepath) and not mediatype == "photos":
|
|
# We need to copy over the default items
|
|
xbmcvfs.mkdirs(nodepath)
|
|
|
|
# Create index entry
|
|
nodeXML = "%sindex.xml" % nodepath
|
|
# Set windows property
|
|
path = "library://video/emby/%s/" % dirname
|
|
for i in range(1, indexnumber):
|
|
# Verify to make sure we don't create duplicates
|
|
if window('Emby.nodes.%s.index' % i) == path:
|
|
return
|
|
|
|
if mediatype == "photos":
|
|
path = "plugin://plugin.video.emby/?id=%s&mode=getsubfolders" % indexnumber
|
|
|
|
window('Emby.nodes.%s.index' % indexnumber, value=path)
|
|
|
|
# Root
|
|
if not mediatype == "photos":
|
|
if viewtype == "mixed":
|
|
specialtag = "%s - %s" % (tagname, mediatype)
|
|
root = self.commonRoot(order=0, label=specialtag, tagname=tagname, roottype=0)
|
|
else:
|
|
root = self.commonRoot(order=0, label=tagname, tagname=tagname, roottype=0)
|
|
try:
|
|
utils.indent(root)
|
|
except: pass
|
|
etree.ElementTree(root).write(nodeXML)
|
|
|
|
nodetypes = {
|
|
|
|
'1': "all",
|
|
'2': "recent",
|
|
'3': "recentepisodes",
|
|
'4': "inprogress",
|
|
'5': "inprogressepisodes",
|
|
'6': "unwatched",
|
|
'7': "nextepisodes",
|
|
'8': "sets",
|
|
'9': "genres",
|
|
'10': "random",
|
|
'11': "recommended",
|
|
}
|
|
mediatypes = {
|
|
# label according to nodetype per mediatype
|
|
'movies':
|
|
{
|
|
'1': tagname,
|
|
'2': 30174,
|
|
'4': 30177,
|
|
'6': 30189,
|
|
'8': 20434,
|
|
'9': 135,
|
|
'10': 30229,
|
|
'11': 30230
|
|
},
|
|
|
|
'tvshows':
|
|
{
|
|
'1': tagname,
|
|
'2': 30170,
|
|
'3': 30175,
|
|
'4': 30171,
|
|
'5': 30178,
|
|
'7': 30179,
|
|
'9': 135,
|
|
'10': 30229,
|
|
'11': 30230
|
|
},
|
|
|
|
'homevideos':
|
|
{
|
|
'1': tagname,
|
|
'2': 30251,
|
|
'11': 30253
|
|
},
|
|
|
|
'photos':
|
|
{
|
|
'1': tagname,
|
|
'2': 30252,
|
|
'8': 30255,
|
|
'11': 30254
|
|
},
|
|
|
|
'musicvideos':
|
|
{
|
|
'1': tagname,
|
|
'2': 30256,
|
|
'4': 30257,
|
|
'6': 30258
|
|
}
|
|
}
|
|
|
|
nodes = mediatypes[mediatype]
|
|
for node in nodes:
|
|
|
|
nodetype = nodetypes[node]
|
|
nodeXML = "%s%s.xml" % (nodepath, nodetype)
|
|
# Get label
|
|
stringid = nodes[node]
|
|
if node != "1":
|
|
label = lang(stringid)
|
|
if not label:
|
|
label = xbmc.getLocalizedString(stringid)
|
|
else:
|
|
label = stringid
|
|
|
|
# Set window properties
|
|
if (mediatype == "homevideos" or mediatype == "photos") and nodetype == "all":
|
|
# Custom query
|
|
path = ("plugin://plugin.video.emby/?id=%s&mode=browsecontent&type=%s"
|
|
% (tagname, mediatype))
|
|
elif (mediatype == "homevideos" or mediatype == "photos"):
|
|
# Custom query
|
|
path = ("plugin://plugin.video.emby/?id=%s&mode=browsecontent&type=%s&folderid=%s"
|
|
% (tagname, mediatype, nodetype))
|
|
elif nodetype == "nextepisodes":
|
|
# Custom query
|
|
path = "plugin://plugin.video.emby/?id=%s&mode=nextup&limit=25" % tagname
|
|
elif self.kodiversion == 14 and nodetype == "recentepisodes":
|
|
# Custom query
|
|
path = "plugin://plugin.video.emby/?id=%s&mode=recentepisodes&limit=25" % tagname
|
|
elif self.kodiversion == 14 and nodetype == "inprogressepisodes":
|
|
# Custom query
|
|
path = "plugin://plugin.video.emby/?id=%s&mode=inprogressepisodes&limit=25"% tagname
|
|
else:
|
|
path = "library://video/emby/%s/%s.xml" % (viewid, nodetype)
|
|
|
|
if mediatype == "photos":
|
|
windowpath = "ActivateWindow(Pictures,%s,return)" % path
|
|
else:
|
|
windowpath = "ActivateWindow(Videos,%s,return)" % path
|
|
|
|
if nodetype == "all":
|
|
|
|
if viewtype == "mixed":
|
|
templabel = "%s - %s" % (tagname, mediatype)
|
|
else:
|
|
templabel = label
|
|
|
|
embynode = "Emby.nodes.%s" % indexnumber
|
|
window('%s.title' % embynode, value=templabel)
|
|
window('%s.path' % embynode, value=windowpath)
|
|
window('%s.content' % embynode, value=path)
|
|
window('%s.type' % embynode, value=mediatype)
|
|
else:
|
|
embynode = "Emby.nodes.%s.%s" % (indexnumber, nodetype)
|
|
window('%s.title' % embynode, value=label)
|
|
window('%s.path' % embynode, value=windowpath)
|
|
window('%s.content' % embynode, value=path)
|
|
|
|
if mediatype == "photos":
|
|
# For photos, we do not create a node in videos but we do want the window props
|
|
# to be created.
|
|
# To do: add our photos nodes to kodi picture sources somehow
|
|
continue
|
|
|
|
if xbmcvfs.exists(nodeXML):
|
|
# Don't recreate xml if already exists
|
|
continue
|
|
|
|
# Create the root
|
|
if (nodetype == "nextepisodes" or mediatype == "homevideos" or
|
|
(self.kodiversion == 14 and nodetype in ('recentepisodes', 'inprogressepisodes'))):
|
|
# Folder type with plugin path
|
|
root = self.commonRoot(order=node, label=label, tagname=tagname, roottype=2)
|
|
etree.SubElement(root, 'path').text = path
|
|
etree.SubElement(root, 'content').text = "episodes"
|
|
else:
|
|
root = self.commonRoot(order=node, label=label, tagname=tagname)
|
|
if nodetype in ('recentepisodes', 'inprogressepisodes'):
|
|
etree.SubElement(root, 'content').text = "episodes"
|
|
else:
|
|
etree.SubElement(root, 'content').text = mediatype
|
|
|
|
limit = "25"
|
|
# Elements per nodetype
|
|
if nodetype == "all":
|
|
etree.SubElement(root, 'order', {'direction': "ascending"}).text = "sorttitle"
|
|
|
|
elif nodetype == "recent":
|
|
etree.SubElement(root, 'order', {'direction': "descending"}).text = "dateadded"
|
|
etree.SubElement(root, 'limit').text = limit
|
|
rule = etree.SubElement(root, 'rule', {'field': "playcount", 'operator': "is"})
|
|
etree.SubElement(rule, 'value').text = "0"
|
|
|
|
elif nodetype == "inprogress":
|
|
etree.SubElement(root, 'rule', {'field': "inprogress", 'operator': "true"})
|
|
etree.SubElement(root, 'limit').text = limit
|
|
|
|
elif nodetype == "genres":
|
|
etree.SubElement(root, 'order', {'direction': "ascending"}).text = "sorttitle"
|
|
etree.SubElement(root, 'group').text = "genres"
|
|
|
|
elif nodetype == "unwatched":
|
|
etree.SubElement(root, 'order', {'direction': "ascending"}).text = "sorttitle"
|
|
rule = etree.SubElement(root, "rule", {'field': "playcount", 'operator': "is"})
|
|
etree.SubElement(rule, 'value').text = "0"
|
|
|
|
elif nodetype == "sets":
|
|
etree.SubElement(root, 'order', {'direction': "ascending"}).text = "sorttitle"
|
|
etree.SubElement(root, 'group').text = "sets"
|
|
|
|
elif nodetype == "random":
|
|
etree.SubElement(root, 'order', {'direction': "ascending"}).text = "random"
|
|
etree.SubElement(root, 'limit').text = limit
|
|
|
|
elif nodetype == "recommended":
|
|
etree.SubElement(root, 'order', {'direction': "descending"}).text = "rating"
|
|
etree.SubElement(root, 'limit').text = limit
|
|
rule = etree.SubElement(root, 'rule', {'field': "playcount", 'operator': "is"})
|
|
etree.SubElement(rule, 'value').text = "0"
|
|
rule2 = etree.SubElement(root, 'rule',
|
|
attrib={'field': "rating", 'operator': "greaterthan"})
|
|
etree.SubElement(rule2, 'value').text = "7"
|
|
|
|
elif nodetype == "recentepisodes":
|
|
# Kodi Isengard, Jarvis
|
|
etree.SubElement(root, 'order', {'direction': "descending"}).text = "dateadded"
|
|
etree.SubElement(root, 'limit').text = limit
|
|
rule = etree.SubElement(root, 'rule', {'field': "playcount", 'operator': "is"})
|
|
etree.SubElement(rule, 'value').text = "0"
|
|
|
|
elif nodetype == "inprogressepisodes":
|
|
# Kodi Isengard, Jarvis
|
|
etree.SubElement(root, 'limit').text = "25"
|
|
rule = etree.SubElement(root, 'rule',
|
|
attrib={'field': "inprogress", 'operator':"true"})
|
|
|
|
try:
|
|
utils.indent(root)
|
|
except: pass
|
|
etree.ElementTree(root).write(nodeXML)
|
|
|
|
def singleNode(self, indexnumber, tagname, mediatype, itemtype):
|
|
|
|
tagname = tagname.encode('utf-8')
|
|
cleantagname = utils.normalize_nodes(tagname)
|
|
nodepath = xbmc.translatePath("special://profile/library/video/").decode('utf-8')
|
|
nodeXML = "%semby_%s.xml" % (nodepath, cleantagname)
|
|
path = "library://video/emby_%s.xml" % cleantagname
|
|
windowpath = "ActivateWindow(Videos,%s,return)" % path
|
|
|
|
# Create the video node directory
|
|
if not xbmcvfs.exists(nodepath):
|
|
# We need to copy over the default items
|
|
shutil.copytree(
|
|
src=xbmc.translatePath("special://xbmc/system/library/video").decode('utf-8'),
|
|
dst=xbmc.translatePath("special://profile/library/video").decode('utf-8'))
|
|
xbmcvfs.exists(path)
|
|
|
|
labels = {
|
|
|
|
'Favorite movies': 30180,
|
|
'Favorite tvshows': 30181,
|
|
'Favorite episodes': 30182,
|
|
'channels': 30173
|
|
}
|
|
label = lang(labels[tagname])
|
|
embynode = "Emby.nodes.%s" % indexnumber
|
|
window('%s.title' % embynode, value=label)
|
|
window('%s.path' % embynode, value=windowpath)
|
|
window('%s.content' % embynode, value=path)
|
|
window('%s.type' % embynode, value=itemtype)
|
|
|
|
if xbmcvfs.exists(nodeXML):
|
|
# Don't recreate xml if already exists
|
|
return
|
|
|
|
if itemtype == "channels":
|
|
root = self.commonRoot(order=1, label=label, tagname=tagname, roottype=2)
|
|
etree.SubElement(root, 'path').text = "plugin://plugin.video.emby/?id=0&mode=channels"
|
|
elif itemtype == "favourites" and mediatype == "episodes":
|
|
root = self.commonRoot(order=1, label=label, tagname=tagname, roottype=2)
|
|
etree.SubElement(root, 'path').text = "plugin://plugin.video.emby/?id=%s&mode=browsecontent&type=%s&folderid=favepisodes" %(tagname, mediatype)
|
|
else:
|
|
root = self.commonRoot(order=1, label=label, tagname=tagname)
|
|
etree.SubElement(root, 'order', {'direction': "ascending"}).text = "sorttitle"
|
|
|
|
etree.SubElement(root, 'content').text = mediatype
|
|
|
|
try:
|
|
utils.indent(root)
|
|
except: pass
|
|
etree.ElementTree(root).write(nodeXML)
|
|
|
|
def clearProperties(self):
|
|
|
|
log.info("Clearing nodes properties.")
|
|
embyprops = window('Emby.nodes.total')
|
|
propnames = [
|
|
|
|
"index","path","title","content",
|
|
"inprogress.content","inprogress.title",
|
|
"inprogress.content","inprogress.path",
|
|
"nextepisodes.title","nextepisodes.content",
|
|
"nextepisodes.path","unwatched.title",
|
|
"unwatched.content","unwatched.path",
|
|
"recent.title","recent.content","recent.path",
|
|
"recentepisodes.title","recentepisodes.content",
|
|
"recentepisodes.path","inprogressepisodes.title",
|
|
"inprogressepisodes.content","inprogressepisodes.path"
|
|
]
|
|
|
|
if embyprops:
|
|
totalnodes = int(embyprops)
|
|
for i in range(totalnodes):
|
|
for prop in propnames:
|
|
window('Emby.nodes.%s.%s' % (str(i), prop), clear=True) |