You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
72 lines
2.3 KiB
72 lines
2.3 KiB
#coding=utf-8
|
|
'''
|
|
Created on Jun 17, 2015
|
|
|
|
@author: danimar
|
|
'''
|
|
|
|
import xml.etree.ElementTree as ET
|
|
from lxml.etree import Element, tostring
|
|
|
|
class DynamicXml(object):
|
|
def __getattr__(self, name):
|
|
try:
|
|
return object.__getattribute__(self,name)
|
|
except:
|
|
self.__setattr__(name, None)
|
|
return object.__getattribute__(self,name)
|
|
|
|
def __setattr__(self, obj, val):
|
|
if(obj=="value" or obj=="atributos" or obj=="_indice"):
|
|
object.__setattr__(self,obj, val)
|
|
else:
|
|
self._indice = self._indice + 1
|
|
object.__setattr__(self,obj, DynamicXml(val, self._indice))
|
|
|
|
def __init__(self, value, indice=0):
|
|
self.value = value
|
|
self.atributos={}
|
|
self._indice = indice
|
|
|
|
def __str__(self):
|
|
return str(self.value)
|
|
def __call__(self, *args, **kw):
|
|
if(len(kw)>0):
|
|
self.atributos=kw
|
|
if(len(args)>0):
|
|
self.value= args[0]
|
|
else:
|
|
return self.value
|
|
|
|
def __getitem__(self, i):
|
|
if not isinstance(self.value, list):
|
|
self.value = []
|
|
if(i+1 > len(self.value)):
|
|
self.value.append(DynamicXml(None))
|
|
return self.value[i]
|
|
|
|
def render(self, pretty_print=False):
|
|
root = Element(self.value)
|
|
self._gerar_xml(root, self)
|
|
return tostring(root, pretty_print=pretty_print)
|
|
|
|
def _gerar_xml(self, xml, objeto):
|
|
items = sorted(
|
|
objeto.__dict__.items(),
|
|
key=lambda x: x[1]._indice if isinstance(x[1], DynamicXml) else 0
|
|
)
|
|
for attr, value in items:
|
|
if(attr!="value" and attr!="atributos" and attr!="_indice"):
|
|
if isinstance(value(), list):
|
|
for item in value():
|
|
sub = ET.SubElement(xml, attr)
|
|
self._gerar_xml(sub, item)
|
|
else:
|
|
sub = ET.SubElement(xml, attr)
|
|
if(str(value)!="None"):
|
|
sub.text = str(value)
|
|
self._gerar_xml(sub, value)
|
|
elif(attr=="atributos"):
|
|
for atr, val in value.items():
|
|
xml.set(atr.replace("__", ":"), str(val))
|
|
|