-
-
Save scottrmercer/e42c396de1ff55b37f6d to your computer and use it in GitHub Desktop.
Added a couple of things missing from the original.
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
import simplejson as json | |
from lxml import etree | |
def is_node_badger_array(node): | |
if len(node) <= 1: | |
return False | |
for (a,b) in zip(node,node[1:]): | |
if a.tag != b.tag: | |
return False | |
return True | |
def body_to_pod(xml): | |
is_arr = is_node_badger_array(xml) | |
if is_arr: | |
return {xml[0].tag: map(body_to_pod,xml)} | |
else: | |
obj = dict() | |
for (key,val) in xml.attrib.iteritems(): | |
obj['@' + key] = val | |
if xml.text is not None and not len(xml): | |
obj['$'] = xml.text | |
for child in xml.iterchildren(): | |
if type(child.tag) is str: #omits commment tags | |
is_simple = (len(child) == 0 and len(child.attrib) == 0) | |
obj[child.tag] = child.text if is_simple else body_to_pod(child) | |
return obj | |
def from_pod_body(data,parent): | |
elements = list() | |
if not isinstance(data,dict): | |
raise Exception('%s is not a dictionary!' % (data)) | |
for (key,value) in data.iteritems(): | |
if isinstance(value,list): | |
for item in value: | |
itemEl = etree.Element(key) | |
itemEl.extend(from_pod_body(item,itemEl)) | |
elements.append(itemEl) | |
elif isinstance(value,dict): | |
itemEl = etree.Element(key) | |
itemEl.extend(from_pod_body(value,itemEl)) | |
elements.append(itemEl) | |
else: | |
if len(key) > 0 and key[0] == '@': | |
parent.attrib[key[1:]] = value | |
elif len(key) == 1 and key[0] == '$': | |
parent.text = value | |
else: | |
itemEl = etree.Element(key) | |
itemEl.text = value | |
elements.append(itemEl) | |
return elements | |
def from_pod(data): | |
els = from_pod_body(data,None) | |
if len(els) == 0: | |
raise Exception('Badgerfish json should be object on top level.') | |
return els[0] | |
def to_pod(xml): | |
return {xml.tag: body_to_pod(xml)} | |
def to_json(xml,**kargs): | |
return json.JSONEncoder(**kargs).encode(to_pod(xml)) | |
def from_json(json_data): | |
return from_pod(json.loads(json_data)) |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
Added: