0
0
mirror of https://github.com/libretro/Lakka-LibreELEC.git synced 2024-12-15 20:20:30 +00:00
MilhouseVH 74e48d0797 kodi/xml_merge.py: make python3 compatible; bug fixes
* decode byte object
* f was not visible in exception, resulting in an unhanled second exception
* fix indenting
2019-10-29 12:19:07 +00:00

62 lines
2.0 KiB
Python
Executable File

#!/usr/bin/env python3
# taken from http://stackoverflow.com/a/14879370 with minor modifications
from __future__ import print_function
import os
import sys
import xml.dom.minidom
from xml.etree import ElementTree as et
def printerr(*args, **kwargs):
print(*args, file=sys.stderr, **kwargs)
class hashabledict(dict):
def __hash__(self):
return hash(tuple(sorted(self.items())))
class XMLCombiner(object):
def __init__(self, filenames):
if len(filenames) == 0:
raise Exception('No filenames!')
try:
self.roots = []
for f in filenames:
self.roots.append(et.parse(f).getroot())
except xml.etree.ElementTree.ParseError:
printerr("ERROR: Unable to parse XML file %s" % f)
raise
def prettyPrint(self, etree_xml):
minidom = xml.dom.minidom.parseString(et.tostring(etree_xml))
return "\n".join([line for line in minidom.toprettyxml(indent=" ", encoding="utf-8").decode('utf-8').split('\n') if line.strip() != ""])
def combine(self):
for r in self.roots[1:]:
self.combine_element(self.roots[0], r)
return self.prettyPrint(self.roots[0])
def combine_element(self, one, other):
mapping = {(el.tag, hashabledict(el.attrib)): el for el in one}
for el in other:
if len(el) == 0:
try:
mapping[(el.tag, hashabledict(el.attrib))].text = el.text
except KeyError:
mapping[(el.tag, hashabledict(el.attrib))] = el
one.append(el)
else:
try:
self.combine_element(mapping[(el.tag, hashabledict(el.attrib))], el)
except KeyError:
mapping[(el.tag, hashabledict(el.attrib))] = el
one.append(el)
if __name__ == '__main__':
xmlfiles = [file for file in sys.argv[1:] if os.path.exists(file)]
r = XMLCombiner(xmlfiles).combine()
print(r)