mirror of
https://github.com/robweber/xbmcbackup.git
synced 2024-11-15 04:45:49 +01:00
249 lines
8.9 KiB
Python
249 lines
8.9 KiB
Python
import xbmc
|
|
import xbmcgui
|
|
import xbmcvfs
|
|
from dropbox import client, rest, session
|
|
import utils as utils
|
|
import os
|
|
import time
|
|
|
|
APP_KEY = 'f5wlmek6aoriqax'
|
|
APP_SECRET = 'b1461sje1kxgzet'
|
|
|
|
class FileManager:
|
|
walk_path = ''
|
|
fileArray = None
|
|
verbose_log = False
|
|
not_dir = ['.zip','.xsp','.rar']
|
|
|
|
def __init__(self,path):
|
|
self.walk_path = path
|
|
|
|
def createFileList(self):
|
|
self.fileArray = []
|
|
self.verbose_log = utils.getSetting("verbose_log") == 'true'
|
|
|
|
#figure out which syncing options to run
|
|
if(utils.getSetting('backup_addons') == 'true'):
|
|
self.addFile("-addons")
|
|
self.walkTree(self.walk_path + "addons/")
|
|
|
|
self.addFile("-userdata")
|
|
|
|
if(utils.getSetting('backup_addon_data') == 'true'):
|
|
self.addFile("-userdata/addon_data")
|
|
self.walkTree(self.walk_path + "userdata/addon_data/")
|
|
|
|
if(utils.getSetting('backup_database') == 'true'):
|
|
self.addFile("-userdata/Database")
|
|
self.walkTree(self.walk_path + "userdata/Database")
|
|
|
|
if(utils.getSetting("backup_playlists") == 'true'):
|
|
self.addFile("-userdata/playlists")
|
|
self.walkTree(self.walk_path + "userdata/playlists")
|
|
|
|
if(utils.getSetting("backup_thumbnails") == "true"):
|
|
self.addFile("-userdata/Thumbnails")
|
|
self.walkTree(self.walk_path + "userdata/Thumbnails")
|
|
|
|
if(utils.getSetting("backup_config") == "true"):
|
|
self.addFile("-userdata/keymaps")
|
|
self.walkTree(self.walk_path + "userdata/keymaps")
|
|
|
|
self.addFile("-userdata/peripheral_data")
|
|
self.walkTree(self.walk_path + "userdata/peripheral_data")
|
|
|
|
#this part is an oddity
|
|
dirs,configFiles = xbmcvfs.listdir(self.walk_path + "userdata/")
|
|
for aFile in configFiles:
|
|
if(aFile.endswith(".xml")):
|
|
self.addFile("userdata/" + aFile)
|
|
|
|
def walkTree(self,directory):
|
|
dirs,files = xbmcvfs.listdir(directory)
|
|
|
|
#create all the subdirs first
|
|
for aDir in dirs:
|
|
dirPath = xbmc.translatePath(directory + "/" + aDir)
|
|
file_ext = aDir.split('.')[-1]
|
|
|
|
self.addFile("-" + dirPath[len(self.walk_path):].decode("UTF-8"))
|
|
#catch for "non directory" type files
|
|
if (not any(file_ext in s for s in self.not_dir)):
|
|
self.walkTree(dirPath)
|
|
|
|
#copy all the files
|
|
for aFile in files:
|
|
filePath = xbmc.translatePath(directory + "/" + aFile)
|
|
self.addFile(filePath[len(self.walk_path):].decode("UTF-8"))
|
|
|
|
def addFile(self,filename):
|
|
#write the full remote path name of this file
|
|
utils.log("Add File: " + filename,xbmc.LOGDEBUG)
|
|
self.fileArray.append(filename)
|
|
|
|
def getFileList(self):
|
|
return self.fileArray
|
|
|
|
class XbmcBackup:
|
|
#constants for initiating a back or restore
|
|
Backup = 0
|
|
Restore = 1
|
|
|
|
local_path = ''
|
|
remote_root = ''
|
|
remote_path = ''
|
|
restoreFile = None
|
|
|
|
#for the progress bar
|
|
progressBar = None
|
|
filesLeft = 0
|
|
filesTotal = 1
|
|
|
|
fileManager = None
|
|
|
|
def __init__(self):
|
|
self.local_path = xbmc.makeLegalFilename(xbmc.translatePath("special://home"),False);
|
|
|
|
if(utils.getSetting('remote_selection') == '1'):
|
|
self.remote_root = utils.getSetting('remote_path_2')
|
|
utils.setSetting("remote_path","")
|
|
elif(utils.getSetting('remote_selection') == '0'):
|
|
self.remote_root = utils.getSetting("remote_path")
|
|
|
|
#fix slashes
|
|
self.remote_root = self.remote_root.replace("\\","/")
|
|
|
|
#check if trailing slash is included
|
|
if(self.remote_root[-1:] != "/"):
|
|
self.remote_root = self.remote_root + "/"
|
|
|
|
utils.log(utils.getString(30046))
|
|
|
|
def run(self,mode=-1,runSilent=False):
|
|
#check if we should use the progress bar
|
|
if(utils.getSetting('run_silent') == 'false' and not runSilent):
|
|
self.progressBar = xbmcgui.DialogProgress()
|
|
self.progressBar.create(utils.getString(30010),utils.getString(30049) + "......")
|
|
|
|
#determine backup mode
|
|
if(mode == -1):
|
|
mode = int(utils.getSetting('addon_mode'))
|
|
|
|
#append backup folder name
|
|
if(mode == self.Backup and self.remote_root != ''):
|
|
#check here if we're using Dropbox or not
|
|
if(utils.getSetting('remote_selection') == 2):
|
|
session = session.DropboxSession(APP_KEY,APP_SECRET,"app_folder")
|
|
token = session.obtain_request_token()
|
|
access_token = session.obtain_access_token(token)
|
|
else:
|
|
self.remote_path = self.remote_root + time.strftime("%Y%m%d") + "/"
|
|
elif(mode == self.Restore and utils.getSetting("backup_name") != '' and self.remote_root != ''):
|
|
self.remote_path = self.remote_root + utils.getSetting("backup_name") + "/"
|
|
else:
|
|
self.remote_path = ""
|
|
|
|
utils.log(utils.getString(30047) + ": " + self.local_path)
|
|
utils.log(utils.getString(30048) + ": " + self.remote_path)
|
|
|
|
#run the correct mode
|
|
if(mode == self.Backup):
|
|
utils.log(utils.getString(30023) + " - " + utils.getString(30016))
|
|
self.fileManager = FileManager(self.local_path)
|
|
|
|
#for backups check if remote path exists
|
|
if(xbmcvfs.exists(self.remote_path)):
|
|
#this will fail - need a disclaimer here
|
|
utils.log(utils.getString(30050))
|
|
|
|
self.syncFiles()
|
|
|
|
#remove old backups
|
|
total_backups = int(utils.getSetting('backup_rotation'))
|
|
if(total_backups > 0):
|
|
|
|
dirs,files = xbmcvfs.listdir(self.remote_root)
|
|
if(len(dirs) > total_backups):
|
|
#remove backups to equal total wanted
|
|
remove_num = len(dirs) - total_backups - 1
|
|
self.filesTotal = self.filesTotal + remove_num + 1
|
|
|
|
#update the progress bar if it is available
|
|
while(remove_num >= 0 and not self.checkCancel()):
|
|
self.updateProgress(utils.getString(30054) + " " + dirs[remove_num])
|
|
utils.log("Removing backup " + dirs[remove_num])
|
|
xbmcvfs.rmdir(self.remote_root + dirs[remove_num] + "/",True)
|
|
remove_num = remove_num - 1
|
|
|
|
|
|
else:
|
|
utils.log(utils.getString(30023) + " - " + utils.getString(30017))
|
|
self.fileManager = FileManager(self.remote_path)
|
|
|
|
#for restores remote path must exist
|
|
if(xbmcvfs.exists(self.remote_path)):
|
|
self.restoreFiles()
|
|
else:
|
|
xbmcgui.Dialog().ok(utils.getString(30010),utils.getString(30045),self.remote_path)
|
|
|
|
if(utils.getSetting('run_silent') == 'false' and not runSilent):
|
|
self.progressBar.close()
|
|
|
|
def syncFiles(self):
|
|
|
|
#make the remote directory
|
|
xbmcvfs.mkdir(self.remote_path)
|
|
|
|
utils.log(utils.getString(30051))
|
|
self.fileManager.createFileList()
|
|
|
|
allFiles = self.fileManager.getFileList()
|
|
|
|
#write list from local to remote
|
|
self.writeFiles(allFiles,self.local_path,self.remote_path)
|
|
|
|
def restoreFiles(self):
|
|
self.fileManager.createFileList()
|
|
|
|
utils.log(utils.getString(30051))
|
|
allFiles = self.fileManager.getFileList()
|
|
|
|
#write list from remote to local
|
|
self.writeFiles(allFiles,self.remote_path,self.local_path)
|
|
|
|
#call update addons to refresh everything
|
|
xbmc.executebuiltin('UpdateLocalAddons')
|
|
|
|
def writeFiles(self,fileList,source,dest):
|
|
utils.log("Writing files to: " + dest)
|
|
self.filesTotal = len(fileList)
|
|
self.filesLeft = self.filesTotal
|
|
|
|
#write each file from source to destination
|
|
for aFile in fileList:
|
|
if(not self.checkCancel()):
|
|
utils.log('Writing file: ' + source + aFile,xbmc.LOGDEBUG)
|
|
self.updateProgress(aFile)
|
|
if (aFile.startswith("-")):
|
|
xbmcvfs.mkdir(xbmc.makeLegalFilename(dest + aFile[1:],False))
|
|
else:
|
|
xbmcvfs.copy(xbmc.makeLegalFilename(source + aFile),xbmc.makeLegalFilename(dest + aFile,False))
|
|
|
|
def updateProgress(self,message=''):
|
|
self.filesLeft = self.filesLeft - 1
|
|
|
|
#update the progress bar
|
|
if(self.progressBar != None):
|
|
self.progressBar.update(int((float(self.filesTotal - self.filesLeft)/float(self.filesTotal)) * 100),message)
|
|
|
|
def checkCancel(self):
|
|
result = False
|
|
|
|
if(self.progressBar != None):
|
|
result = self.progressBar.iscanceled()
|
|
|
|
return result
|
|
|
|
def isReady(self):
|
|
return True if self.remote_root != '' else False
|