faster CharacterDataHandler in NodeBuilder
This commit is contained in:
Dimitur Kirov
parent
eacf5fb900
commit
32bfde61e2
|
|
@ -20,362 +20,380 @@ I'm personally using it in many other separate projects. It is designed to be as
|
||||||
import xml.parsers.expat
|
import xml.parsers.expat
|
||||||
|
|
||||||
def XMLescape(txt):
|
def XMLescape(txt):
|
||||||
"""Returns provided string with symbols & < > " replaced by their respective XML entities."""
|
"""Returns provided string with symbols & < > " replaced by their respective XML entities."""
|
||||||
return txt.replace("&", "&").replace("<", "<").replace(">", ">").replace('"', """)
|
return txt.replace("&", "&").replace("<", "<").replace(">", ">").replace('"', """)
|
||||||
|
|
||||||
ENCODING='utf-8'
|
ENCODING='utf-8'
|
||||||
def ustr(what):
|
def ustr(what):
|
||||||
"""Converts object "what" to unicode string using it's own __str__ method if accessible or unicode method otherwise."""
|
"""Converts object "what" to unicode string using it's own __str__ method if accessible or unicode method otherwise."""
|
||||||
if type(what) == type(u''): return what
|
if type(what) == type(u''): return what
|
||||||
try: r=what.__str__()
|
try: r=what.__str__()
|
||||||
except AttributeError: r=str(what)
|
except AttributeError: r=str(what)
|
||||||
if type(r)<>type(u''): return unicode(r,ENCODING)
|
if type(r)<>type(u''): return unicode(r,ENCODING)
|
||||||
return r
|
return r
|
||||||
|
|
||||||
class Node:
|
class Node:
|
||||||
""" Node class describes syntax of separate XML Node. It have a constructor that permits node creation
|
""" Node class describes syntax of separate XML Node. It have a constructor that permits node creation
|
||||||
from set of "namespace name", attributes and payload of text strings and other nodes.
|
from set of "namespace name", attributes and payload of text strings and other nodes.
|
||||||
It does not natively support building node from text string and uses NodeBuilder class for that purpose.
|
It does not natively support building node from text string and uses NodeBuilder class for that purpose.
|
||||||
After creation node can be mangled in many ways so it can be completely changed.
|
After creation node can be mangled in many ways so it can be completely changed.
|
||||||
Also node can be serialised into string in one of two modes: default (where the textual representation
|
Also node can be serialised into string in one of two modes: default (where the textual representation
|
||||||
of node describes it exactly) and "fancy" - with whitespace added to make indentation and thus make
|
of node describes it exactly) and "fancy" - with whitespace added to make indentation and thus make
|
||||||
result more readable by human.
|
result more readable by human.
|
||||||
|
|
||||||
Node class have attribute FORCE_NODE_RECREATION that is defaults to False thus enabling fast node
|
Node class have attribute FORCE_NODE_RECREATION that is defaults to False thus enabling fast node
|
||||||
replication from the some other node. The drawback of the fast way is that new node shares some
|
replication from the some other node. The drawback of the fast way is that new node shares some
|
||||||
info with the "original" node that is changing the one node may influence the other. Though it is
|
info with the "original" node that is changing the one node may influence the other. Though it is
|
||||||
rarely needed (in xmpppy it is never needed at all since I'm usually never using original node after
|
rarely needed (in xmpppy it is never needed at all since I'm usually never using original node after
|
||||||
replication (and using replication only to move upwards on the classes tree).
|
replication (and using replication only to move upwards on the classes tree).
|
||||||
"""
|
"""
|
||||||
FORCE_NODE_RECREATION=0
|
FORCE_NODE_RECREATION=0
|
||||||
def __init__(self, tag=None, attrs={}, payload=[], parent=None, node=None):
|
def __init__(self, tag=None, attrs={}, payload=[], parent=None, node=None):
|
||||||
""" Takes "tag" argument as the name of node (prepended by namespace, if needed and separated from it
|
""" Takes "tag" argument as the name of node (prepended by namespace, if needed and separated from it
|
||||||
by a space), attrs dictionary as the set of arguments, payload list as the set of textual strings
|
by a space), attrs dictionary as the set of arguments, payload list as the set of textual strings
|
||||||
and child nodes that this node carries within itself and "parent" argument that is another node
|
and child nodes that this node carries within itself and "parent" argument that is another node
|
||||||
that this one will be the child of. Also the __init__ can be provided with "node" argument that is
|
that this one will be the child of. Also the __init__ can be provided with "node" argument that is
|
||||||
either a text string containing exactly one node or another Node instance to begin with. If both
|
either a text string containing exactly one node or another Node instance to begin with. If both
|
||||||
"node" and other arguments is provided then the node initially created as replica of "node"
|
"node" and other arguments is provided then the node initially created as replica of "node"
|
||||||
provided and then modified to be compliant with other arguments."""
|
provided and then modified to be compliant with other arguments."""
|
||||||
if node:
|
if node:
|
||||||
if self.FORCE_NODE_RECREATION and type(node)==type(self): node=str(node)
|
if self.FORCE_NODE_RECREATION and type(node)==type(self):
|
||||||
if type(node)<>type(self): node=NodeBuilder(node,self)
|
node=str(node)
|
||||||
else:
|
if type(node)<>type(self):
|
||||||
self.name,self.namespace,self.attrs,self.data,self.kids,self.parent = node.name,node.namespace,{},[],[],node.parent
|
node=NodeBuilder(node,self)
|
||||||
for key in node.attrs.keys(): self.attrs[key]=node.attrs[key]
|
else:
|
||||||
for data in node.data: self.data.append(data)
|
self.name,self.namespace,self.attrs,self.data,self.kids,self.parent = node.name,node.namespace,{},[],[],node.parent
|
||||||
for kid in node.kids: self.kids.append(kid)
|
for key in node.attrs.keys(): self.attrs[key]=node.attrs[key]
|
||||||
else: self.name,self.namespace,self.attrs,self.data,self.kids,self.parent = 'tag','',{},[],[],None
|
for data in node.data: self.data.append(data)
|
||||||
|
for kid in node.kids: self.kids.append(kid)
|
||||||
|
else: self.name,self.namespace,self.attrs,self.data,self.kids,self.parent = 'tag','',{},[],[],None
|
||||||
|
|
||||||
if tag: self.namespace, self.name = ([self.namespace]+tag.split())[-2:]
|
if tag: self.namespace, self.name = ([self.namespace]+tag.split())[-2:]
|
||||||
if parent: self.parent = parent
|
if parent: self.parent = parent
|
||||||
if self.parent and not self.namespace: self.namespace=self.parent.namespace
|
if self.parent and not self.namespace: self.namespace=self.parent.namespace
|
||||||
for attr in attrs.keys():
|
for attr in attrs.keys():
|
||||||
self.attrs[attr]=attrs[attr]
|
self.attrs[attr]=attrs[attr]
|
||||||
if type(payload) in (type(''),type(u'')): payload=[payload]
|
if type(payload) in (type(''),type(u'')): payload=[payload]
|
||||||
for i in payload:
|
for i in payload:
|
||||||
if type(i)==type(self): self.addChild(node=i)
|
if type(i)==type(self): self.addChild(node=i)
|
||||||
else: self.data.append(ustr(i))
|
else: self.data.append(ustr(i))
|
||||||
|
|
||||||
def __str__(self,fancy=0):
|
def __str__(self,fancy=0):
|
||||||
""" Method used to dump node into textual representation.
|
""" Method used to dump node into textual representation.
|
||||||
if "fancy" argument is set to True produces indented output for readability."""
|
if "fancy" argument is set to True produces indented output for readability."""
|
||||||
s = (fancy-1) * 2 * ' ' + "<" + self.name
|
s = (fancy-1) * 2 * ' ' + "<" + self.name
|
||||||
if self.namespace:
|
if self.namespace:
|
||||||
if not self.parent or self.parent.namespace!=self.namespace:
|
if not self.parent or self.parent.namespace!=self.namespace:
|
||||||
s = s + ' xmlns="%s"'%self.namespace
|
s = s + ' xmlns="%s"'%self.namespace
|
||||||
for key in self.attrs.keys():
|
for key in self.attrs.keys():
|
||||||
val = ustr(self.attrs[key])
|
val = ustr(self.attrs[key])
|
||||||
s = s + ' %s="%s"' % ( key, XMLescape(val) )
|
s = s + ' %s="%s"' % ( key, XMLescape(val) )
|
||||||
s = s + ">"
|
s = s + ">"
|
||||||
cnt = 0
|
cnt = 0
|
||||||
if self.kids:
|
if self.kids:
|
||||||
if fancy: s = s + "\n"
|
if fancy: s = s + "\n"
|
||||||
for a in self.kids:
|
for a in self.kids:
|
||||||
if not fancy and (len(self.data)-1)>=cnt: s=s+XMLescape(self.data[cnt])
|
if not fancy and (len(self.data)-1)>=cnt: s=s+XMLescape(self.data[cnt])
|
||||||
elif (len(self.data)-1)>=cnt: s=s+XMLescape(self.data[cnt].strip())
|
elif (len(self.data)-1)>=cnt: s=s+XMLescape(self.data[cnt].strip())
|
||||||
s = s + a.__str__(fancy and fancy+1)
|
s = s + a.__str__(fancy and fancy+1)
|
||||||
cnt=cnt+1
|
cnt=cnt+1
|
||||||
if not fancy and (len(self.data)-1) >= cnt: s = s + XMLescape(self.data[cnt])
|
if not fancy and (len(self.data)-1) >= cnt: s = s + XMLescape(self.data[cnt])
|
||||||
elif (len(self.data)-1) >= cnt: s = s + XMLescape(self.data[cnt].strip())
|
elif (len(self.data)-1) >= cnt: s = s + XMLescape(self.data[cnt].strip())
|
||||||
if not self.kids and s[-1:]=='>':
|
if not self.kids and s[-1:]=='>':
|
||||||
s=s[:-1]+' />'
|
s=s[:-1]+' />'
|
||||||
if fancy: s = s + "\n"
|
if fancy: s = s + "\n"
|
||||||
else:
|
else:
|
||||||
if fancy and not self.data: s = s + (fancy-1) * 2 * ' '
|
if fancy and not self.data: s = s + (fancy-1) * 2 * ' '
|
||||||
s = s + "</" + self.name + ">"
|
s = s + "</" + self.name + ">"
|
||||||
if fancy: s = s + "\n"
|
if fancy: s = s + "\n"
|
||||||
return s
|
return s
|
||||||
def addChild(self, name=None, attrs={}, payload=[], namespace=None, node=None):
|
def addChild(self, name=None, attrs={}, payload=[], namespace=None, node=None):
|
||||||
""" If "node" argument is provided, adds it as child node. Else creates new node from
|
""" If "node" argument is provided, adds it as child node. Else creates new node from
|
||||||
the other arguments' values and adds it as well."""
|
the other arguments' values and adds it as well."""
|
||||||
if namespace: name=namespace+' '+name
|
if namespace: name=namespace+' '+name
|
||||||
if node:
|
if node:
|
||||||
newnode=node
|
newnode=node
|
||||||
node.parent = self
|
node.parent = self
|
||||||
else: newnode=Node(tag=name, parent=self, attrs=attrs, payload=payload)
|
else: newnode=Node(tag=name, parent=self, attrs=attrs, payload=payload)
|
||||||
self.kids.append(newnode)
|
self.kids.append(newnode)
|
||||||
return newnode
|
return newnode
|
||||||
def addData(self, data):
|
def addData(self, data):
|
||||||
""" Adds some CDATA to node. """
|
""" Adds some CDATA to node. """
|
||||||
self.data.append(ustr(data))
|
self.data.append(ustr(data))
|
||||||
def clearData(self):
|
def clearData(self):
|
||||||
""" Removes all CDATA from the node. """
|
""" Removes all CDATA from the node. """
|
||||||
self.data=[]
|
self.data=[]
|
||||||
def delAttr(self, key):
|
def delAttr(self, key):
|
||||||
""" Deletes an attribute "key" """
|
""" Deletes an attribute "key" """
|
||||||
del self.attrs[key]
|
del self.attrs[key]
|
||||||
def delChild(self, node, attrs={}):
|
def delChild(self, node, attrs={}):
|
||||||
""" Deletes the "node" from the node's childs list, if "node" is an instance.
|
""" Deletes the "node" from the node's childs list, if "node" is an instance.
|
||||||
Else deletes the first node that have specified name and (optionally) attributes. """
|
Else deletes the first node that have specified name and (optionally) attributes. """
|
||||||
if type(node)<>type(self): node=self.getTag(node,attrs)
|
if type(node)<>type(self): node=self.getTag(node,attrs)
|
||||||
self.kids.remove(node)
|
self.kids.remove(node)
|
||||||
return node
|
return node
|
||||||
def getAttrs(self):
|
def getAttrs(self):
|
||||||
""" Returns all node's attributes as dictionary. """
|
""" Returns all node's attributes as dictionary. """
|
||||||
return self.attrs
|
return self.attrs
|
||||||
def getAttr(self, key):
|
def getAttr(self, key):
|
||||||
""" Returns value of specified attribute. """
|
""" Returns value of specified attribute. """
|
||||||
try: return self.attrs[key]
|
try: return self.attrs[key]
|
||||||
except: return None
|
except: return None
|
||||||
def getChildren(self):
|
def getChildren(self):
|
||||||
""" Returns all node's child nodes as list. """
|
""" Returns all node's child nodes as list. """
|
||||||
return self.kids
|
return self.kids
|
||||||
def getData(self):
|
def getData(self):
|
||||||
""" Returns all node CDATA as string (concatenated). """
|
""" Returns all node CDATA as string (concatenated). """
|
||||||
return ''.join(self.data)
|
return ''.join(self.data)
|
||||||
def getName(self):
|
def getName(self):
|
||||||
""" Returns the name of node """
|
""" Returns the name of node """
|
||||||
return self.name
|
return self.name
|
||||||
def getNamespace(self):
|
def getNamespace(self):
|
||||||
""" Returns the namespace of node """
|
""" Returns the namespace of node """
|
||||||
return self.namespace
|
return self.namespace
|
||||||
def getParent(self):
|
def getParent(self):
|
||||||
""" Returns the parent of node (if present). """
|
""" Returns the parent of node (if present). """
|
||||||
return self.parent
|
return self.parent
|
||||||
def getPayload(self):
|
def getPayload(self):
|
||||||
""" Return the payload of node i.e. list of child nodes and CDATA entries.
|
""" Return the payload of node i.e. list of child nodes and CDATA entries.
|
||||||
F.e. for "<node>text1<nodea/><nodeb/> text2</node>" will be returned list:
|
F.e. for "<node>text1<nodea/><nodeb/> text2</node>" will be returned list:
|
||||||
['text1', <nodea instance>, <nodeb instance>, ' text2']. """
|
['text1', <nodea instance>, <nodeb instance>, ' text2']. """
|
||||||
ret=[]
|
ret=[]
|
||||||
for i in range(len(self.kids)+len(self.data)+1):
|
for i in range(len(self.kids)+len(self.data)+1):
|
||||||
try:
|
try:
|
||||||
if self.data[i]: ret.append(self.data[i])
|
if self.data[i]: ret.append(self.data[i])
|
||||||
except IndexError: pass
|
except IndexError: pass
|
||||||
try: ret.append(self.kids[i])
|
try: ret.append(self.kids[i])
|
||||||
except IndexError: pass
|
except IndexError: pass
|
||||||
return ret
|
return ret
|
||||||
def getTag(self, name, attrs={}, namespace=None):
|
def getTag(self, name, attrs={}, namespace=None):
|
||||||
""" Filters all child nodes using specified arguments as filter.
|
""" Filters all child nodes using specified arguments as filter.
|
||||||
Returns the first found or None if not found. """
|
Returns the first found or None if not found. """
|
||||||
return self.getTags(name, attrs, namespace, one=1)
|
return self.getTags(name, attrs, namespace, one=1)
|
||||||
def getTagAttr(self,tag,attr):
|
def getTagAttr(self,tag,attr):
|
||||||
""" Returns attribute value of the child with specified name (or None if no such attribute)."""
|
""" Returns attribute value of the child with specified name (or None if no such attribute)."""
|
||||||
try: return self.getTag(tag).attrs[attr]
|
try: return self.getTag(tag).attrs[attr]
|
||||||
except: return None
|
except: return None
|
||||||
def getTagData(self,tag):
|
def getTagData(self,tag):
|
||||||
""" Returns cocatenated CDATA of the child with specified name."""
|
""" Returns cocatenated CDATA of the child with specified name."""
|
||||||
try: return self.getTag(tag).getData()
|
try: return self.getTag(tag).getData()
|
||||||
except: return None
|
except: return None
|
||||||
def getTags(self, name, attrs={}, namespace=None, one=0):
|
def getTags(self, name, attrs={}, namespace=None, one=0):
|
||||||
""" Filters all child nodes using specified arguments as filter.
|
""" Filters all child nodes using specified arguments as filter.
|
||||||
Returns the list of nodes found. """
|
Returns the list of nodes found. """
|
||||||
nodes=[]
|
nodes=[]
|
||||||
for node in self.kids:
|
for node in self.kids:
|
||||||
if namespace and namespace<>node.getNamespace(): continue
|
if namespace and namespace<>node.getNamespace(): continue
|
||||||
if node.getName() == name:
|
if node.getName() == name:
|
||||||
for key in attrs.keys():
|
for key in attrs.keys():
|
||||||
if not node.attrs.has_key(key) or node.attrs[key]<>attrs[key]: break
|
if not node.attrs.has_key(key) or node.attrs[key]<>attrs[key]: break
|
||||||
else: nodes.append(node)
|
else: nodes.append(node)
|
||||||
if one and nodes: return nodes[0]
|
if one and nodes: return nodes[0]
|
||||||
if not one: return nodes
|
if not one: return nodes
|
||||||
def setAttr(self, key, val):
|
def setAttr(self, key, val):
|
||||||
""" Sets attribute "key" with the value "val". """
|
""" Sets attribute "key" with the value "val". """
|
||||||
self.attrs[key]=val
|
self.attrs[key]=val
|
||||||
def setData(self, data):
|
def setData(self, data):
|
||||||
""" Sets node's CDATA to provided string. Resets all previous CDATA!"""
|
""" Sets node's CDATA to provided string. Resets all previous CDATA!"""
|
||||||
self.data=[ustr(data)]
|
self.data=[ustr(data)]
|
||||||
def setName(self,val):
|
def setName(self,val):
|
||||||
""" Changes the node name. """
|
""" Changes the node name. """
|
||||||
self.name = val
|
self.name = val
|
||||||
def setNamespace(self, namespace):
|
def setNamespace(self, namespace):
|
||||||
""" Changes the node namespace. """
|
""" Changes the node namespace. """
|
||||||
self.namespace=namespace
|
self.namespace=namespace
|
||||||
def setParent(self, node):
|
def setParent(self, node):
|
||||||
""" Sets node's parent to "node". WARNING: do not checks if the parent already present
|
""" Sets node's parent to "node". WARNING: do not checks if the parent already present
|
||||||
and not removes the node from the list of childs of previous parent. """
|
and not removes the node from the list of childs of previous parent. """
|
||||||
self.parent = node
|
self.parent = node
|
||||||
def setPayload(self,payload,add=0):
|
def setPayload(self,payload,add=0):
|
||||||
""" Sets node payload according to the list specified. WARNING: completely replaces all node's
|
""" Sets node payload according to the list specified. WARNING: completely replaces all node's
|
||||||
previous content. If you wish just to add child or CDATA - use addData or addChild methods. """
|
previous content. If you wish just to add child or CDATA - use addData or addChild methods. """
|
||||||
if type(payload) in (type(''),type(u'')): payload=[payload]
|
if type(payload) in (type(''),type(u'')): payload=[payload]
|
||||||
if add: self.kids+=payload
|
if add: self.kids+=payload
|
||||||
else: self.kids=payload
|
else: self.kids=payload
|
||||||
def setTag(self, name, attrs={}, namespace=None):
|
def setTag(self, name, attrs={}, namespace=None):
|
||||||
""" Same as getTag but if the node with specified namespace/attributes not found, creates such
|
""" Same as getTag but if the node with specified namespace/attributes not found, creates such
|
||||||
node and returns it. """
|
node and returns it. """
|
||||||
node=self.getTags(name, attrs, namespace=namespace, one=1)
|
node=self.getTags(name, attrs, namespace=namespace, one=1)
|
||||||
if node: return node
|
if node: return node
|
||||||
else: return self.addChild(name, attrs, namespace=namespace)
|
else: return self.addChild(name, attrs, namespace=namespace)
|
||||||
def setTagAttr(self,tag,attr,val):
|
def setTagAttr(self,tag,attr,val):
|
||||||
""" Creates new node (if not already present) with name "tag"
|
""" Creates new node (if not already present) with name "tag"
|
||||||
and sets it's attribute "attr" to value "val". """
|
and sets it's attribute "attr" to value "val". """
|
||||||
try: self.getTag(tag).attrs[attr]=val
|
try: self.getTag(tag).attrs[attr]=val
|
||||||
except: self.addChild(tag,attrs={attr:val})
|
except: self.addChild(tag,attrs={attr:val})
|
||||||
def setTagData(self,tag,val,attrs={}):
|
def setTagData(self,tag,val,attrs={}):
|
||||||
""" Creates new node (if not already present) with name "tag" and (optionally) attributes "attrs"
|
""" Creates new node (if not already present) with name "tag" and (optionally) attributes "attrs"
|
||||||
and sets it's CDATA to string "val". """
|
and sets it's CDATA to string "val". """
|
||||||
try: self.getTag(tag,attrs).setData(ustr(val))
|
try: self.getTag(tag,attrs).setData(ustr(val))
|
||||||
except: self.addChild(tag,attrs,payload=[ustr(val)])
|
except: self.addChild(tag,attrs,payload=[ustr(val)])
|
||||||
def has_attr(self,key):
|
def has_attr(self,key):
|
||||||
""" Checks if node have attribute "key"."""
|
""" Checks if node have attribute "key"."""
|
||||||
return self.attrs.has_key(key)
|
return self.attrs.has_key(key)
|
||||||
def __getitem__(self,item):
|
def __getitem__(self,item):
|
||||||
""" Returns node's attribute "item" value. """
|
""" Returns node's attribute "item" value. """
|
||||||
return self.getAttr(item)
|
return self.getAttr(item)
|
||||||
def __setitem__(self,item,val):
|
def __setitem__(self,item,val):
|
||||||
""" Sets node's attribute "item" value. """
|
""" Sets node's attribute "item" value. """
|
||||||
return self.setAttr(item,val)
|
return self.setAttr(item,val)
|
||||||
def __delitem__(self,item):
|
def __delitem__(self,item):
|
||||||
""" Deletes node's attribute "item". """
|
""" Deletes node's attribute "item". """
|
||||||
return self.delAttr(item,val)
|
return self.delAttr(item,val)
|
||||||
def __getattr__(self,attr):
|
def __getattr__(self,attr):
|
||||||
""" Reduce memory usage caused by T/NT classes - use memory only when needed. """
|
""" Reduce memory usage caused by T/NT classes - use memory only when needed. """
|
||||||
if attr=='T':
|
if attr=='T':
|
||||||
self.T=T(self)
|
self.T=T(self)
|
||||||
return self.T
|
return self.T
|
||||||
if attr=='NT':
|
if attr=='NT':
|
||||||
self.NT=NT(self)
|
self.NT=NT(self)
|
||||||
return self.NT
|
return self.NT
|
||||||
raise AttributeError
|
raise AttributeError
|
||||||
|
|
||||||
class T:
|
class T:
|
||||||
""" Auxiliary class used to quick access to node's child nodes. """
|
""" Auxiliary class used to quick access to node's child nodes. """
|
||||||
def __init__(self,node): self.__dict__['node']=node
|
def __init__(self,node): self.__dict__['node']=node
|
||||||
def __getattr__(self,attr): return self.node.setTag(attr)
|
def __getattr__(self,attr): return self.node.setTag(attr)
|
||||||
def __setattr__(self,attr,val):
|
def __setattr__(self,attr,val):
|
||||||
if isinstance(val,Node): Node.__init__(self.node.setTag(attr),node=val)
|
if isinstance(val,Node): Node.__init__(self.node.setTag(attr),node=val)
|
||||||
else: return self.node.setTagData(attr,val)
|
else: return self.node.setTagData(attr,val)
|
||||||
def __delattr__(self,attr): return self.node.delChild(attr)
|
def __delattr__(self,attr): return self.node.delChild(attr)
|
||||||
|
|
||||||
class NT(T):
|
class NT(T):
|
||||||
""" Auxiliary class used to quick create node's child nodes. """
|
""" Auxiliary class used to quick create node's child nodes. """
|
||||||
def __getattr__(self,attr): return self.node.addChild(attr)
|
def __getattr__(self,attr): return self.node.addChild(attr)
|
||||||
def __setattr__(self,attr,val):
|
def __setattr__(self,attr,val):
|
||||||
if isinstance(val,Node): self.node.addChild(attr,node=val)
|
if isinstance(val,Node): self.node.addChild(attr,node=val)
|
||||||
else: return self.node.addChild(attr,payload=[val])
|
else: return self.node.addChild(attr,payload=[val])
|
||||||
|
|
||||||
DBG_NODEBUILDER = 'nodebuilder'
|
DBG_NODEBUILDER = 'nodebuilder'
|
||||||
class NodeBuilder:
|
class NodeBuilder:
|
||||||
""" Builds a Node class minidom from data parsed to it. This class used for two purposes:
|
""" Builds a Node class minidom from data parsed to it. This class used for two purposes:
|
||||||
1. Creation an XML Node from a textual representation. F.e. reading a config file. See an XML2Node method.
|
1. Creation an XML Node from a textual representation. F.e. reading a config file. See an XML2Node method.
|
||||||
2. Handling an incoming XML stream. This is done by mangling
|
2. Handling an incoming XML stream. This is done by mangling
|
||||||
the __dispatch_depth parameter and redefining the dispatch method.
|
the __dispatch_depth parameter and redefining the dispatch method.
|
||||||
You do not need to use this class directly if you do not designing your own XML handler."""
|
You do not need to use this class directly if you do not designing your own XML handler."""
|
||||||
def __init__(self,data=None,initial_node=None):
|
def __init__(self,data=None,initial_node=None):
|
||||||
""" Takes two optional parameters: "data" and "initial_node".
|
""" Takes two optional parameters: "data" and "initial_node".
|
||||||
By default class initialised with empty Node class instance.
|
By default class initialised with empty Node class instance.
|
||||||
Though, if "initial_node" is provided it used as "starting point".
|
Though, if "initial_node" is provided it used as "starting point".
|
||||||
You can think about it as of "node upgrade".
|
You can think about it as of "node upgrade".
|
||||||
"data" (if provided) feeded to parser immidiatedly after instance init.
|
"data" (if provided) feeded to parser immidiatedly after instance init.
|
||||||
"""
|
"""
|
||||||
self.DEBUG(DBG_NODEBUILDER, "Preparing to handle incoming XML stream.", 'start')
|
self.DEBUG(DBG_NODEBUILDER, "Preparing to handle incoming XML stream.", 'start')
|
||||||
self._parser = xml.parsers.expat.ParserCreate(namespace_separator=' ')
|
self._parser = xml.parsers.expat.ParserCreate(namespace_separator=' ')
|
||||||
self._parser.StartElementHandler = self.starttag
|
self._parser.StartElementHandler = self.starttag
|
||||||
self._parser.EndElementHandler = self.endtag
|
self._parser.EndElementHandler = self.endtag
|
||||||
self._parser.CharacterDataHandler = self.handle_data
|
self._parser.StartNamespaceDeclHandler = self.handle_namespace_start
|
||||||
self._parser.StartNamespaceDeclHandler = self.handle_namespace_start
|
self._parser.CharacterDataHandler = self.handle_cdata
|
||||||
self.Parse = self._parser.Parse
|
self.Parse = self._parser.Parse
|
||||||
|
|
||||||
self.__depth = 0
|
self.__depth = 0
|
||||||
self._dispatch_depth = 1
|
self._dispatch_depth = 1
|
||||||
self._document_attrs = None
|
self._document_attrs = None
|
||||||
self._mini_dom=initial_node
|
self._mini_dom=initial_node
|
||||||
self.last_is_data = 1
|
self.last_is_data = 1
|
||||||
self._ptr=None
|
self._ptr=None
|
||||||
self.namespaces={"http://www.w3.org/XML/1998/namespace":'xml:'}
|
self.data_buffer = None
|
||||||
self.xmlns="http://www.w3.org/XML/1998/namespace"
|
self.namespaces={"http://www.w3.org/XML/1998/namespace":'xml:'}
|
||||||
|
self.xmlns="http://www.w3.org/XML/1998/namespace"
|
||||||
|
|
||||||
if data: self._parser.Parse(data,1)
|
if data:
|
||||||
|
self._parser.Parse(data,1)
|
||||||
|
|
||||||
|
|
||||||
|
def destroy(self):
|
||||||
|
""" Method used to allow class instance to be garbage-collected. """
|
||||||
|
self._parser.StartElementHandler = None
|
||||||
|
self._parser.EndElementHandler = None
|
||||||
|
self._parser.CharacterDataHandler = None
|
||||||
|
self._parser.StartNamespaceDeclHandler = None
|
||||||
|
|
||||||
def destroy(self):
|
def starttag(self, tag, attrs):
|
||||||
""" Method used to allow class instance to be garbage-collected. """
|
"""XML Parser callback. Used internally"""
|
||||||
self._parser.StartElementHandler = None
|
attlist=attrs.keys() #
|
||||||
self._parser.EndElementHandler = None
|
for attr in attlist: # FIXME: Crude hack. And it also slows down the whole library considerably.
|
||||||
self._parser.CharacterDataHandler = None
|
sp=attr.rfind(" ") #
|
||||||
self._parser.StartNamespaceDeclHandler = None
|
if sp==-1: continue #
|
||||||
|
ns=attr[:sp] #
|
||||||
|
attrs[self.namespaces[ns]+attr[sp+1:]]=attrs[attr]
|
||||||
|
del attrs[attr] #
|
||||||
|
self.__depth += 1
|
||||||
|
self.DEBUG(DBG_NODEBUILDER, "DEPTH -> %i , tag -> %s, attrs -> %s" % (self.__depth, tag, `attrs`), 'down')
|
||||||
|
if self.__depth == self._dispatch_depth:
|
||||||
|
if not self._mini_dom :
|
||||||
|
self._mini_dom = Node(tag=tag, attrs=attrs)
|
||||||
|
else:
|
||||||
|
Node.__init__(self._mini_dom,tag=tag, attrs=attrs)
|
||||||
|
self._ptr = self._mini_dom
|
||||||
|
elif self.__depth > self._dispatch_depth:
|
||||||
|
self._ptr.kids.append(Node(tag=tag,parent=self._ptr,attrs=attrs))
|
||||||
|
self._ptr = self._ptr.kids[-1]
|
||||||
|
if self.__depth == 1:
|
||||||
|
self._document_attrs = attrs
|
||||||
|
ns, name = (['']+tag.split())[-2:]
|
||||||
|
self.stream_header_received(ns, name, attrs)
|
||||||
|
if not self.last_is_data and self._ptr.parent:
|
||||||
|
self._ptr.parent.data.append('')
|
||||||
|
self.last_is_data = 0
|
||||||
|
def endtag(self, tag ):
|
||||||
|
"""XML Parser callback. Used internally"""
|
||||||
|
self.DEBUG(DBG_NODEBUILDER, "DEPTH -> %i , tag -> %s" % (self.__depth, tag), 'up')
|
||||||
|
if self.data_buffer:
|
||||||
|
self._ptr.data.append(''.join(self.data_buffer))
|
||||||
|
self.data_buffer = None
|
||||||
|
if self.__depth == self._dispatch_depth:
|
||||||
|
self.dispatch(self._mini_dom)
|
||||||
|
elif self.__depth > self._dispatch_depth:
|
||||||
|
self._ptr = self._ptr.parent
|
||||||
|
else:
|
||||||
|
self.DEBUG(DBG_NODEBUILDER, "Got higher than dispatch level. Stream terminated?", 'stop')
|
||||||
|
self.__depth -= 1
|
||||||
|
self.last_is_data = 0
|
||||||
|
if self.__depth == 0: self.stream_footer_received()
|
||||||
|
|
||||||
|
def handle_cdata(self, data):
|
||||||
|
if self.last_is_data:
|
||||||
|
if self.data_buffer:
|
||||||
|
self.data_buffer.append(data)
|
||||||
|
else:
|
||||||
|
self.data_buffer = [data]
|
||||||
|
self.last_is_data = 1
|
||||||
|
|
||||||
|
def handle_data(self, data, *args):
|
||||||
|
"""XML Parser callback. Used internally"""
|
||||||
|
self.DEBUG(DBG_NODEBUILDER, data, 'data')
|
||||||
|
if not self._ptr: return
|
||||||
|
if self.last_is_data:
|
||||||
|
self._ptr.data[-1] += data
|
||||||
|
else:
|
||||||
|
self._ptr.data.append(data)
|
||||||
|
self.last_is_data = 1
|
||||||
|
|
||||||
def starttag(self, tag, attrs):
|
def handle_namespace_start(self, prefix, uri):
|
||||||
"""XML Parser callback. Used internally"""
|
"""XML Parser callback. Used internally"""
|
||||||
attlist=attrs.keys() #
|
if prefix: self.namespaces[uri]=prefix+':'
|
||||||
for attr in attlist: # FIXME: Crude hack. And it also slows down the whole library considerably.
|
else: self.xmlns=uri
|
||||||
sp=attr.rfind(" ") #
|
def DEBUG(self, level, text, comment=None):
|
||||||
if sp==-1: continue #
|
""" Gets all NodeBuilder walking events. Can be used for debugging if redefined."""
|
||||||
ns=attr[:sp] #
|
def getDom(self):
|
||||||
attrs[self.namespaces[ns]+attr[sp+1:]]=attrs[attr]
|
""" Returns just built Node. """
|
||||||
del attrs[attr] #
|
return self._mini_dom
|
||||||
self.__depth += 1
|
def dispatch(self,stanza):
|
||||||
self.DEBUG(DBG_NODEBUILDER, "DEPTH -> %i , tag -> %s, attrs -> %s" % (self.__depth, tag, `attrs`), 'down')
|
""" Gets called when the NodeBuilder reaches some level of depth on it's way up with the built
|
||||||
if self.__depth == self._dispatch_depth:
|
node as argument. Can be redefined to convert incoming XML stanzas to program events. """
|
||||||
if not self._mini_dom : self._mini_dom = Node(tag=tag, attrs=attrs)
|
def stream_header_received(self,ns,tag,attrs):
|
||||||
else: Node.__init__(self._mini_dom,tag=tag, attrs=attrs)
|
""" Method called when stream just opened. """
|
||||||
self._ptr = self._mini_dom
|
def stream_footer_received(self):
|
||||||
elif self.__depth > self._dispatch_depth:
|
""" Method called when stream just closed. """
|
||||||
self._ptr.kids.append(Node(tag=tag,parent=self._ptr,attrs=attrs))
|
|
||||||
self._ptr = self._ptr.kids[-1]
|
|
||||||
if self.__depth == 1:
|
|
||||||
self._document_attrs = attrs
|
|
||||||
ns, name = (['']+tag.split())[-2:]
|
|
||||||
self.stream_header_received(ns, name, attrs)
|
|
||||||
if not self.last_is_data and self._ptr.parent: self._ptr.parent.data.append('')
|
|
||||||
self.last_is_data = 0
|
|
||||||
|
|
||||||
def endtag(self, tag ):
|
|
||||||
"""XML Parser callback. Used internally"""
|
|
||||||
self.DEBUG(DBG_NODEBUILDER, "DEPTH -> %i , tag -> %s" % (self.__depth, tag), 'up')
|
|
||||||
if self.__depth == self._dispatch_depth:
|
|
||||||
self.dispatch(self._mini_dom)
|
|
||||||
elif self.__depth > self._dispatch_depth:
|
|
||||||
self._ptr = self._ptr.parent
|
|
||||||
else:
|
|
||||||
self.DEBUG(DBG_NODEBUILDER, "Got higher than dispatch level. Stream terminated?", 'stop')
|
|
||||||
self.__depth -= 1
|
|
||||||
self.last_is_data = 0
|
|
||||||
if self.__depth == 0: self.stream_footer_received()
|
|
||||||
|
|
||||||
def handle_data(self, data):
|
|
||||||
"""XML Parser callback. Used internally"""
|
|
||||||
self.DEBUG(DBG_NODEBUILDER, data, 'data')
|
|
||||||
if not self._ptr: return
|
|
||||||
if self.last_is_data:
|
|
||||||
self._ptr.data[-1] += data
|
|
||||||
else:
|
|
||||||
self._ptr.data.append(data)
|
|
||||||
self.last_is_data = 1
|
|
||||||
|
|
||||||
def handle_namespace_start(self, prefix, uri):
|
|
||||||
"""XML Parser callback. Used internally"""
|
|
||||||
if prefix: self.namespaces[uri]=prefix+':'
|
|
||||||
else: self.xmlns=uri
|
|
||||||
def DEBUG(self, level, text, comment=None):
|
|
||||||
""" Gets all NodeBuilder walking events. Can be used for debugging if redefined."""
|
|
||||||
def getDom(self):
|
|
||||||
""" Returns just built Node. """
|
|
||||||
return self._mini_dom
|
|
||||||
def dispatch(self,stanza):
|
|
||||||
""" Gets called when the NodeBuilder reaches some level of depth on it's way up with the built
|
|
||||||
node as argument. Can be redefined to convert incoming XML stanzas to program events. """
|
|
||||||
def stream_header_received(self,ns,tag,attrs):
|
|
||||||
""" Method called when stream just opened. """
|
|
||||||
def stream_footer_received(self):
|
|
||||||
""" Method called when stream just closed. """
|
|
||||||
|
|
||||||
def XML2Node(xml):
|
def XML2Node(xml):
|
||||||
""" Converts supplied textual string into XML node. Handy f.e. for reading configuration file.
|
""" Converts supplied textual string into XML node. Handy f.e. for reading configuration file.
|
||||||
Raises xml.parser.expat.parsererror if provided string is not well-formed XML. """
|
Raises xml.parser.expat.parsererror if provided string is not well-formed XML. """
|
||||||
return NodeBuilder(xml).getDom()
|
return NodeBuilder(xml).getDom()
|
||||||
|
|
||||||
def BadXML2Node(xml):
|
def BadXML2Node(xml):
|
||||||
""" Converts supplied textual string into XML node. Survives if xml data is cutted half way round.
|
""" Converts supplied textual string into XML node. Survives if xml data is cutted half way round.
|
||||||
I.e. "<html>some text <br>some more text". Will raise xml.parser.expat.parsererror on misplaced
|
I.e. "<html>some text <br>some more text". Will raise xml.parser.expat.parsererror on misplaced
|
||||||
tags though. F.e. "<b>some text <br>some more text</b>" will not work."""
|
tags though. F.e. "<b>some text <br>some more text</b>" will not work."""
|
||||||
return NodeBuilder(xml).getDom()
|
return NodeBuilder(xml).getDom()
|
||||||
|
|
|
||||||
Loading…
Reference in New Issue