104 lines
2.6 KiB
Python
104 lines
2.6 KiB
Python
|
# coding=utf-8
|
||
|
|
|
||
|
|
import xml.sax
|
||
|
|
from datetime import datetime
|
||
|
|
from datetime import date
|
||
|
|
import time
|
||
|
|
|
||
|
|
class ModuleState:
|
||
|
|
def __init__(self, name):
|
||
|
|
self.name = name
|
||
|
|
self.attributes = dict()
|
||
|
|
self.childs = list()
|
||
|
|
self.index = dict()
|
||
|
|
self.index_fieldname = None
|
||
|
|
self.index_tagname = None
|
||
|
|
|
||
|
|
def getName(self):
|
||
|
|
return self.name
|
||
|
|
|
||
|
|
def __getitem__(self, i):
|
||
|
|
return self.getAttribute(i)
|
||
|
|
|
||
|
|
def __contains__(self, i):
|
||
|
|
return i in self.index
|
||
|
|
|
||
|
|
def getAttribute(self, name):
|
||
|
|
if name in self.attributes:
|
||
|
|
return self.attributes[name]
|
||
|
|
else:
|
||
|
|
return None
|
||
|
|
|
||
|
|
def getDate(self, name):
|
||
|
|
if name in self.attributes.keys():
|
||
|
|
if isinstance(self.attributes[name], datetime):
|
||
|
|
return self.attributes[name]
|
||
|
|
else:
|
||
|
|
return date.fromtimestamp(float(self.attributes[name]))
|
||
|
|
else:
|
||
|
|
return None
|
||
|
|
|
||
|
|
def getInt(self, name):
|
||
|
|
if name in self.attributes.keys():
|
||
|
|
return int(self.attributes[name])
|
||
|
|
else:
|
||
|
|
return None
|
||
|
|
|
||
|
|
def setIndex(self, fieldname = "name", tagname = None):
|
||
|
|
self.index_fieldname = fieldname
|
||
|
|
self.index_tagname = tagname
|
||
|
|
for child in self.childs:
|
||
|
|
if (tagname is None or tagname == child.name) and child.hasAttribute(fieldname):
|
||
|
|
self.index[child[fieldname]] = child
|
||
|
|
|
||
|
|
def hasAttribute(self, name):
|
||
|
|
return (name in self.attributes)
|
||
|
|
|
||
|
|
def setAttribute(self, name, value):
|
||
|
|
self.attributes[name] = value
|
||
|
|
|
||
|
|
def getChilds(self):
|
||
|
|
return self.childs
|
||
|
|
|
||
|
|
def getNode(self, tagname):
|
||
|
|
ret = None
|
||
|
|
for child in self.childs:
|
||
|
|
if tagname is None or tagname == child.name:
|
||
|
|
ret = child
|
||
|
|
return ret
|
||
|
|
|
||
|
|
def getNodes(self, tagname):
|
||
|
|
ret = list()
|
||
|
|
for child in self.childs:
|
||
|
|
if tagname is None or tagname == child.name:
|
||
|
|
ret.append(child)
|
||
|
|
return ret
|
||
|
|
|
||
|
|
def addChild(self, child):
|
||
|
|
self.childs.append(child)
|
||
|
|
if self.index_fieldname is not None:
|
||
|
|
self.setIndex(self.index_fieldname, self.index_tagname)
|
||
|
|
|
||
|
|
def save_node(self, gen):
|
||
|
|
attribs = {}
|
||
|
|
for att in self.attributes.keys():
|
||
|
|
if isinstance(self.attributes[att], datetime):
|
||
|
|
attribs[att] = str(time.mktime(self.attributes[att].timetuple()))
|
||
|
|
else:
|
||
|
|
attribs[att] = str(self.attributes[att])
|
||
|
|
attrs = xml.sax.xmlreader.AttributesImpl(attribs)
|
||
|
|
|
||
|
|
gen.startElement(self.name, attrs)
|
||
|
|
|
||
|
|
for child in self.childs:
|
||
|
|
child.save_node(gen)
|
||
|
|
|
||
|
|
gen.endElement(self.name)
|
||
|
|
|
||
|
|
def save(self, filename):
|
||
|
|
with open(filename,"w") as f:
|
||
|
|
gen = xml.sax.saxutils.XMLGenerator(f, "utf-8")
|
||
|
|
gen.startDocument()
|
||
|
|
self.save_node(gen)
|
||
|
|
gen.endDocument()
|