From ec040e459c785fd76974c02e9b3d062cae83d3e0 Mon Sep 17 00:00:00 2001 From: Suan-Aik Yeo Date: Sun, 23 Oct 2011 02:24:59 -0500 Subject: [PATCH] enable listnames for list2xml and list2et --- huTools/structured.py | 9 +++++---- 1 file changed, 5 insertions(+), 4 deletions(-) mode change 100644 => 100755 huTools/structured.py diff --git a/huTools/structured.py b/huTools/structured.py old mode 100644 new mode 100755 index c5984e1..cf6daef --- a/huTools/structured.py +++ b/huTools/structured.py @@ -290,13 +290,14 @@ def dict2et(xmldict, roottag='data', listnames=None): return root -def list2et(xmllist, root, elementname): +def list2et(xmllist, root, elementname, listnames={}): """Converts a list to an Elementtree. See also dict2et() """ - basexml = dict2et({root: xmllist}, 'xml', listnames={root: elementname}) + listnames[root] = elementname + basexml = dict2et({root: xmllist}, 'xml', listnames) return basexml.find(root) @@ -311,12 +312,12 @@ def dict2xml(datadict, roottag='data', listnames=None, pretty=False): return ET.tostring(tree, 'utf-8') -def list2xml(datalist, root, elementname, pretty=False): +def list2xml(datalist, root, elementname, listnames=None, pretty=False): """Converts a list to an UTF-8 encoded XML string. See also dict2et() """ - tree = list2et(datalist, root, elementname) + tree = list2et(datalist, root, elementname, listnames) if pretty: indent(tree) return ET.tostring(tree, 'utf-8')