def create_tree(thing, parent=None, object_names=False):
"""Create an XML etree from a thing.
Arguments:
thing thing to interpret
parent etree.Element to put subtree into
default: create a new Element tree
object_names experimental feature: if True tag names are object
names instead of types
Returns:
etree.Element instance created
"""
try:
grammar = type(thing).grammar
except AttributeError:
if isinstance(thing, list):
grammar = pypeg2.csl(pypeg2.name())
else:
grammar = pypeg2.word
name = type(thing).__name__
if object_names:
try:
name = str(thing.name)
name = name.replace(" ", "_")
except AttributeError:
pass
if parent is None:
me = etree.Element(name)
else:
me = etree.SubElement(parent, name)
for e in pypeg2.attributes(grammar):
if object_names and e.name == "name":
if name != type(thing).__name__:
continue
key, value = e.name, getattr(thing, e.name, None)
if value is not None:
if pypeg2._issubclass(e.thing, (str, int, pypeg2.Literal)) \
or type(e.thing) == pypeg2._RegEx:
me.set(key, str(value))
else:
create_tree(value, me, object_names)
if isinstance(thing, list):
things = thing
elif isinstance(thing, pypeg2.Namespace):
things = thing.values()
else:
things = []
last = None
for t in things:
if type(t) == str:
if last is not None:
last.tail = str(t)
else:
me.text = str(t)
else:
last = create_tree(t, me, object_names)
if isinstance(thing, str):
me.text = str(thing)
return me
def create_tree(thing, parent=None, object_names=False):
"""Create an XML etree from a thing.
Arguments:
thing thing to interpret
parent etree.Element to put subtree into
default: create a new Element tree
object_names experimental feature: if True tag names are object
names instead of types
Returns:
etree.Element instance created
"""
try:
grammar = type(thing).grammar
except AttributeError:
if isinstance(thing, list):
grammar = pypeg2.csl(pypeg2.name())
else:
grammar = pypeg2.word
name = type(thing).__name__
if object_names:
try:
name = str(thing.name)
name = name.replace(" ", "_")
except AttributeError:
pass
if parent is None:
me = etree.Element(name)
else:
me = etree.SubElement(parent, name)
for e in pypeg2.attributes(grammar):
if object_names and e.name == "name":
if name != type(thing).__name__:
continue
key, value = e.name, getattr(thing, e.name, None)
if value is not None:
if pypeg2._issubclass(e.thing, (str, int, pypeg2.Literal)) \
or type(e.thing) == pypeg2._RegEx:
me.set(key, str(value))
else:
create_tree(value, me, object_names)
if isinstance(thing, list):
things = thing
elif isinstance(thing, pypeg2.Namespace):
things = thing.values()
else:
things = []
last = None
for t in things:
if type(t) == str:
if last is not None:
last.tail = str(t)
else:
me.text = str(t)
else:
last = create_tree(t, me, object_names)
if isinstance(thing, str):
me.text = str(thing)
return me
def create_thing(element, symbol_table):
"""Create thing from an XML element.
Arguments:
element etree.Element instance to read
symbol_table symbol table where the classes can be found
Returns:
thing created
"""
C = symbol_table[element.tag]
if element.text:
thing = C(element.text)
else:
thing = C()
subs = iter(element)
iterated_already = False
try:
grammar = C.grammar
except AttributeError:
pass
else:
for e in pypeg2.attributes(grammar):
key = e.name
if pypeg2._issubclass(e.thing, (str, int, pypeg2.Literal)) \
or type(e.thing) == pypeg2._RegEx:
try:
value = element.attrib[e.name]
except KeyError:
pass
else:
setattr(thing, key, e.thing(value))
else:
try:
if not iterated_already:
iterated_already = True
sub = next(subs)
except StopIteration:
pass
if sub.tag == e.thing.__name__:
iterated_already = False
t = create_thing(sub, symbol_table)
setattr(thing, key, t)
if issubclass(C, list) or issubclass(C, pypeg2.Namespace):
try:
while True:
if iterated_already:
iterated_alread = False
else:
sub = next(subs)
t = create_thing(sub, symbol_table)
if isinstance(thing, pypeg2.List):
thing.append(t)
else:
thing[t.name] = t
except StopIteration:
pass
return thing
def create_thing(element, symbol_table):
"""Create thing from an XML element.
Arguments:
element etree.Element instance to read
symbol_table symbol table where the classes can be found
Returns:
thing created
"""
C = symbol_table[element.tag]
if element.text:
thing = C(element.text)
else:
thing = C()
subs = iter(element)
iterated_already = False
try:
grammar = C.grammar
except AttributeError:
pass
else:
for e in pypeg2.attributes(grammar):
key = e.name
if pypeg2._issubclass(e.thing, (str, int, pypeg2.Literal)) \
or type(e.thing) == pypeg2._RegEx:
try:
value = element.attrib[e.name]
except KeyError:
pass
else:
setattr(thing, key, e.thing(value))
else:
try:
if not iterated_already:
iterated_already = True
sub = next(subs)
except StopIteration:
pass
if sub.tag == e.thing.__name__:
iterated_already = False
t = create_thing(sub, symbol_table)
setattr(thing, key, t)
if issubclass(C, list) or issubclass(C, pypeg2.Namespace):
try:
while True:
if iterated_already:
iterated_alread = False
else:
sub = next(subs)
t = create_thing(sub, symbol_table)
if isinstance(thing, pypeg2.List):
thing.append(t)
else:
thing[t.name] = t
except StopIteration:
pass
return thing
