From 1971eb8655c96319c6b7386030f06c7a2d05e561 Mon Sep 17 00:00:00 2001 From: shaun Date: Sun, 15 Mar 2015 15:18:34 +1100 Subject: [PATCH] create a empty sources.xml to work with if one does not exist --- resources/lib/Utils.py | 52 ++++++++++++++++++++++-------------------- 1 file changed, 27 insertions(+), 25 deletions(-) diff --git a/resources/lib/Utils.py b/resources/lib/Utils.py index 7ee26cf7..ae73cdc1 100644 --- a/resources/lib/Utils.py +++ b/resources/lib/Utils.py @@ -53,7 +53,7 @@ def checkKodiSources(): xbmcvfs.mkdir(dataPath) if not xbmcvfs.exists(movieLibrary + os.sep): xbmcvfs.mkdir(movieLibrary) - rebootRequired = addKodiSource("mediabrowser_movies",movieLibrary,"movies") + rebootRequired = addKodiSource("mediabrowser_movies",movieLibrary,"movies") if not xbmcvfs.exists(tvLibrary + os.sep): xbmcvfs.mkdir(tvLibrary) rebootRequired = addKodiSource("mediabrowser_tvshows",tvLibrary,"tvshows") @@ -86,31 +86,33 @@ def addKodiSource(name, path, type): else: error = True - if error: - # if adding to the database failed, manually add it to sources.xml - sourcesFile = xbmc.translatePath( "special://profile/sources.xml" ) - if xbmcvfs.exists(sourcesFile): - tree = ET.ElementTree(file=sourcesFile) - root = tree.getroot() - videosources = root.find("video") - #remove any existing entries for this path - allsources = videosources.findall("source") - if allsources != None: - for source in allsources: - if source.find("name").text == name: - videosources.remove(source) - # add the new source - source = SubElement(videosources,'source') - SubElement(source, "name").text = name - SubElement(source, "path").text = path - tree.write(sourcesFile) + # add it to sources.xml + sourcesFile = xbmc.translatePath( "special://profile/sources.xml" ) + + # add an emply sources file to work with + if xbmcvfs.exists(sourcesFile) == False: + sources = Element("sources") + video = SubElement(sources, "video") + ET.ElementTree(sources).write(sourcesFile) + + if xbmcvfs.exists(sourcesFile): + tree = ET.ElementTree(file=sourcesFile) + root = tree.getroot() + videosources = root.find("video") + #remove any existing entries for this path + allsources = videosources.findall("source") + if allsources != None: + for source in allsources: + if source.find("name").text == name: + videosources.remove(source) + # add the new source + source = SubElement(videosources,'source') + SubElement(source, "name").text = name + SubElement(source, "path").text = path + tree.write(sourcesFile) #return bool that reboot is needed and manual add of path to kodi - return True - else: - #return false that no reboot is needed - return False - - + #return True + def checkAuthentication(): #check authentication