################################################################################ # Copyright (C) 2019 drinfernoo # # # # This Program is free software; you can redistribute it and/or modify # # it under the terms of the GNU General Public License as published by # # the Free Software Foundation; either version 2, or (at your option) # # any later version. # # # # This Program is distributed in the hope that it will be useful, # # but WITHOUT ANY WARRANTY; without even the implied warranty of # # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the # # GNU General Public License for more details. # # # # You should have received a copy of the GNU General Public License # # along with XBMC; see the file COPYING. If not, write to # # the Free Software Foundation, 675 Mass Ave, Cambridge, MA 02139, USA. # # http://www.gnu.org/copyleft/gpl.html # ################################################################################ import xbmc import xbmcgui import xbmcvfs import glob import os import re from resources.libs.common.config import CONFIG from resources.libs.common import logging from resources.libs.common import tools from resources.libs.gui import window def parse(file, foldername): getid = tools.parse_dom(file, 'addon', ret='id') getname = tools.parse_dom(file, 'addon', ret='name') addid = foldername if len(getid) == 0 else getid[0] title = foldername if len(getname) == 0 else getname[0] temp = title.replace('[', '<').replace(']', '>') temp = re.sub('<[^<]+?>', '', temp) return temp, addid def whitelist(do): addonnames = [] addonids = [] addonfolds = [] dialog = xbmcgui.Dialog() if do == 'edit': fold = glob.glob(os.path.join(CONFIG.ADDONS, '*/')) for folder in sorted(fold, key=lambda x: x): foldername = os.path.split(folder[:-1])[1] if foldername in CONFIG.EXCLUDES: continue elif foldername in CONFIG.DEFAULTPLUGINS: continue elif foldername == 'packages': continue xml = os.path.join(folder, 'addon.xml') if os.path.exists(xml): a = tools.read_from_file(xml) temp, addid = parse(a, foldername) addonnames.append(temp) addonids.append(addid) addonfolds.append(foldername) fold2 = glob.glob(os.path.join(CONFIG.ADDON_DATA, '*/')) for folder in sorted(fold2, key=lambda x: x): foldername = os.path.split(folder[:-1])[1] if foldername in addonfolds: continue if foldername in CONFIG.EXCLUDES: continue xml = os.path.join(CONFIG.ADDONS, foldername, 'addon.xml') xml2 = os.path.join(CONFIG.XBMC, 'addons', foldername, 'addon.xml') if os.path.exists(xml): a = tools.read_from_file(xml) elif os.path.exists(xml2): a = tools.read_from_file(xml2) else: continue temp, addid = parse(a, foldername) addonnames.append(temp) addonids.append(addid) addonfolds.append(foldername) selected = [] tempaddonnames = ["-- Click here to Continue --"] + addonnames currentWhite = whitelist(do='read') for item in currentWhite: logging.log(str(item)) try: name, id, fold = item except Exception as e: logging.log(str(e)) if id in addonids: pos = addonids.index(id)+1 selected.append(pos-1) tempaddonnames[pos] = "[B][COLOR {0}]{1}[/COLOR][/B]".format(CONFIG.COLOR1, name) else: addonids.append(id) addonnames.append(name) tempaddonnames.append("[B][COLOR {0}]{1}[/COLOR][/B]".format(CONFIG.COLOR1, name)) choice = 1 while choice not in [-1, 0]: choice = dialog.select("{0}: Seleccione los complementos que desea incluir en la lista blanca.".format(CONFIG.ADDONTITLE), tempaddonnames) if choice == -1: break elif choice == 0: break else: choice2 = (choice-1) if choice2 in selected: selected.remove(choice2) tempaddonnames[choice] = addonnames[choice2] else: selected.append(choice2) tempaddonnames[choice] = "[B][COLOR {0}]{1}[/COLOR][/B]".format(CONFIG.COLOR1, addonnames[choice2]) white_list = [] if len(selected) > 0: for addon in selected: white_list.append("['%s', '%s', '%s']" % (addonnames[addon], addonids[addon], addonfolds[addon])) writing = '\n'.join(white_list) tools.write_to_file(CONFIG.WHITELIST, writing) else: try: os.remove(CONFIG.WHITELIST) except: pass logging.log_notify(CONFIG.ADDONTITLE, "[COLOR {0}]{1} Add-ons en Lista Blanca[/COLOR]".format(CONFIG.COLOR2, len(selected))) elif do == 'read': white_list = [] if os.path.exists(CONFIG.WHITELIST): lines = tools.read_from_file(CONFIG.WHITELIST).split('\n') for item in lines: try: name, id, fold = eval(item) white_list.append(eval(item)) except: pass return white_list elif do == 'view': list = whitelist(do='read') if len(list) > 0: msg = "Aquí hay una lista de los elementos de su lista blanca, estos elementos(junto con las dependencias) no se eliminará cuando se realice un nuevo comienzo o los datos del usuario se sobrescriban en una instalación de Build.[CR][CR]" for item in list: try: name, id, fold = item except Exception as e: logging.log(str(e)) msg += "[COLOR {0}]{1}[/COLOR] [COLOR {2}]\"{3}\"[/COLOR][CR]".format(CONFIG.COLOR1, name, CONFIG.COLOR2, id) window.show_text_box("Viewing Whitelisted Items", msg) else: logging.log_notify(CONFIG.ADDONTITLE, "[COLOR {0}]No items in whitelist[/COLOR]".format(CONFIG.COLOR2)) elif do == 'import': source = dialog.browse(1, '[COLOR {0}]Seleccione el archivo de la lista blanca para importar[/COLOR]'.format(CONFIG.COLOR2), 'files', '.txt', False, False, CONFIG.HOME) logging.log(str(source)) if not source.endswith('.txt'): logging.log_notify(CONFIG.ADDONTITLE, "[COLOR {0}]Cancelada[/COLOR]".format(CONFIG.COLOR2)) return current = whitelist(do='read') idList = [] count = 0 for item in current: name, id, fold = item idList.append(id) lines = tools.read_from_file(xbmcvfs.File(source)).split('\n') with open(CONFIG.WHITELIST, 'a') as f: for item in lines: try: name, id, folder = eval(item) except Exception as e: logging.log("Error Adding: '{0}' / {1}".format(item, str(e)), level=xbmc.LOGERROR) continue logging.log("{0} / {1} / {2}".format(name, id, folder)) if id not in idList: count += 1 writing = "['{0}', '{1}', '{2}']".format(name, id, folder) if len(idList) + count > 1: writing = "\n%s" % writing f.write(writing) logging.log_notify(CONFIG.ADDONTITLE, "[COLOR {0}]{1} Item(s) Added[/COLOR]".format(CONFIG.COLOR2, count)) elif do == 'export': source = dialog.browse(3, '[COLOR {0}]Seleccione dónde desea exportar el archivo de la lista blanca[/COLOR]'.format(CONFIG.COLOR2), 'files', '.txt', False, False, CONFIG.HOME) logging.log(str(source)) try: xbmcvfs.copy(CONFIG.WHITELIST, os.path.join(source, 'whitelist.txt')) dialog.ok(CONFIG.ADDONTITLE, "[COLOR {0}]La lista blanca se ha exportado a:[/COLOR]".format(CONFIG.COLOR2), "[COLOR {0}]{1}[/COLOR]".format(CONFIG.COLOR1, os.path.join(source, 'whitelist.txt'))) logging.log_notify(CONFIG.ADDONTITLE, "[COLOR {0}]Lista Blanca Exportada[/COLOR]".format(CONFIG.COLOR2)) except Exception as e: logging.log("Export Error: {0}".format(str(e)), level=xbmc.LOGERROR) if not dialog.yesno(CONFIG.ADDONTITLE, "[COLOR {0}]The location you selected isn\'t writable would you like to select another one?[/COLOR]".format(CONFIG.COLOR2), yeslabel="[B][COLOR springgreen]Cambiar Localizacion[/COLOR][/B]", nolabel="[B][COLOR blue]No Cancelar[/COLOR][/B]"): logging.log_notify(CONFIG.ADDONTITLE, "[COLOR {0}]Exportacion de Lista Blanca Cancelada[/COLOR]".format(CONFIG.COLOR2, e)) else: whitelist(do='export') elif do == 'clear': if not dialog.yesno(CONFIG.ADDONTITLE, "[COLOR {0}]Are you sure you want to clear your whitelist?".format(CONFIG.COLOR2), "This process can't be undone.[/COLOR]", yeslabel="[B][COLOR springgreen]Yes Remove[/COLOR][/B]", nolabel="[B][COLOR blue]No Cancel[/COLOR][/B]"): logging.log_notify(CONFIG.ADDONTITLE, "[COLOR {0}]Clear Whitelist Cancelled[/COLOR]".format(CONFIG.COLOR2)) return try: os.remove(CONFIG.WHITELIST) logging.log_notify(CONFIG.ADDONTITLE, "[COLOR {0}]Whitelist Cleared[/COLOR]".format(CONFIG.COLOR2)) except: logging.log_notify(CONFIG.ADDONTITLE, "[COLOR {0}]Error Clearing Whitelist![/COLOR]".format(CONFIG.COLOR2)) ######################################################### # PARTE DE ESTE CÓDIGO FUE TOMADO DE CHIKIRY # # A QUIEN DAMOS LAS GRACIAS # #########################################################