added a backup set chooser on restore

resources/lib/backup.py

@@ -40,7 +40,6 @@ class XbmcBackup:
     filesLeft = 0
     filesTotal = 1
 
-    fileManager = None
     restore_point = None
     skip_advanced = False   #if we should check for the existance of advancedsettings in the restore
     
@@ -166,10 +165,10 @@ class XbmcBackup:
                 self.remote_vfs.mkdir(self.remote_vfs.root_path)
 
             utils.log(utils.getString(30051))
+            utils.log('File Selection Type: ' + str(utils.getSetting('backup_selection_type')))
             allFiles = []
-            fileManager = FileManager(self.xbmc_vfs)
 
-            if(utils.getSetting('backup_selection_type') == 0):
+            if(int(utils.getSetting('backup_selection_type')) == 0):
                 #read in a list of the directories to backup
                 selectedDirs = self._readBackupConfig(utils.addon_dir() + "/resources/data/default_files.json")
                 
@@ -222,6 +221,8 @@ class XbmcBackup:
             self.remote_vfs.set_root(orig_base_path)
 
             if(utils.getSetting("compress_backups") == 'true'):
+                fileManager = FileManager(self.xbmc_vfs)
+                
                 #send the zip file to the real remote vfs
                 zip_name = self.remote_vfs.root_path[:-1] + ".zip"
                 self.remote_vfs.cleanup()
@@ -315,8 +316,17 @@ class XbmcBackup:
                     xbmcgui.Dialog().ok(utils.getString(30077),utils.getString(30078)) 
                     return
 
+            restoreSets = [n['name'] for n in valFile['directories']]
+            selectedSets = xbmcgui.Dialog().multiselect('Choose Sets to Restore',restoreSets)
+
+
+            if(selectedSets != None):
                 #go through each of the directories in the backup and write them to the correct location
-            for aDir in valFile['directories']:
+                for index in selectedSets:
+
+                    #add this directory
+                    aDir = valFile['directories'][index]
+                
                     self.xbmc_vfs.set_root(aDir['path'])
                     if(self.remote_vfs.exists(self.remote_vfs.root_path + aDir['name'] + '/')):
                         #walk the directory
@@ -359,8 +369,8 @@ class XbmcBackup:
     def backupFiles(self,fileList,source,dest):
         result = True
 
-        utils.log("Writing files to: " + dest.root_path)
         utils.log("Source: " + source.root_path)
+        utils.log("Desintation: " + dest.root_path)
         for aFile in fileList:
             if(not self.progressBar.checkCancel()):
                 utils.log('Writing file: ' + aFile,xbmc.LOGDEBUG)
@@ -385,6 +395,7 @@ class XbmcBackup:
         return result
 
     def _addBackupDir(self,folder_name,root_path,dirList):
+        utils.log('Backup set: ' + folder_name)
         fileManager = FileManager(self.xbmc_vfs)
   
         self.xbmc_vfs.set_root(root_path)
@@ -511,14 +522,17 @@ class FileManager:
     def __init__(self,vfs):
         self.vfs = vfs
         self.fileArray = []
+        self.exclude_dir = []
+        self.root_dirs = []
 
     def walk(self):
+        
         for aDir in self.root_dirs:
             self.addFile('-' + xbmc.translatePath(aDir['path']))
             self.walkTree(xbmc.translatePath(aDir['path']),aDir['recurse'])
 
     def walkTree(self,directory,recurse=True):
-        
+        utils.log('walking ' + directory + ', recurse: ' + str(recurse))
         if(directory[-1:] == '/' or directory[-1:] == '\\'):
             directory = directory[:-1]
        
@@ -583,7 +597,7 @@ class FileManager:
     def getFiles(self):
         result = self.fileArray
         self.fileArray = []
-        self.root_paths = []
+        self.root_dirs = []
         self.exclude_dir = []
                                   
         return result