Is there a library or mechanism I can use to flatten the XML file?
Existing:
<A>
<B>
<ConnectionType>a</ConnectionType>
<StartTime>00:00:00</StartTime>
<EndTime>00:00:00</EndTime>
<UseDataDictionary>N</UseDataDictionary>
Desired:
A.B.ConnectionType = a
A.B.StartTime = 00:00:00
A.B.EndTime = 00:00:00
A.B.UseDataDictionary = N
By using xmltodict to transform your XML file to a dictionary, in combination with this answer to flatten a dict, this should be possible.
Example:
# Original code: https://codereview.stackexchange.com/a/21035
from collections import OrderedDict
def flatten_dict(d):
def items():
for key, value in d.items():
if isinstance(value, dict):
for subkey, subvalue in flatten_dict(value).items():
yield key + "." + subkey, subvalue
else:
yield key, value
return OrderedDict(items())
import xmltodict
# Convert to dict
with open('test.xml', 'rb') as f:
xml_content = xmltodict.parse(f)
# Flatten dict
flattened_xml = flatten_dict(xml_content)
# Print in desired format
for k,v in flattened_xml.items():
print('{} = {}'.format(k,v))
Output:
A.B.ConnectionType = a
A.B.StartTime = 00:00:00
A.B.EndTime = 00:00:00
A.B.UseDataDictionary = N
pip install xmltodict and pip install OrderedDictThis is not a complete implementation but you could take advantage of lxmls's getpath:
xml = """<A>
<B>
<ConnectionType>a</ConnectionType>
<StartTime>00:00:00</StartTime>
<EndTime>00:00:00</EndTime>
<UseDataDictionary>N
<UseDataDictionary2>G</UseDataDictionary2>
</UseDataDictionary>
</B>
</A>"""
from lxml import etree
from io import StringIO
tree = etree.parse(StringIO(xml))
root = tree.getroot().tag
for node in tree.iter():
for child in node.getchildren():
if child and child.text.strip():
print("{}.{} = {}".format(root, ".".join(tree.getelementpath(child).split("/")), child.text.strip()))
Which gives you:
A.B.ConnectionType = a
A.B.StartTime = 00:00:00
A.B.EndTime = 00:00:00
A.B.UseDataDictionary = N
A.B.UseDataDictionary.UseDataDictionary2 = G
Here's an improved version from ƘɌỈsƬƠƑ that also handles nested lists:
def flatten_dict(d):
def items():
for key, value in d.items():
# nested subtree
if isinstance(value, dict):
for subkey, subvalue in flatten_dict(value).items():
yield '{}.{}'.format(key, subkey), subvalue
# nested list
elif isinstance(value, list):
for num, elem in enumerate(value):
for subkey, subvalue in flatten_dict(elem).items():
yield '{}.[{}].{}'.format(key, num, subkey), subvalue
# everything else (only leafs should remain)
else:
yield key, value
xmltodictlibrary in combination with this answer to flatten adict.