Viewing file:  NodeIterator.py (3.9 KB) -rw-r--r--
Select action/file-type:
 (+) |  (+) |  (+) | Code (+) | Session (+) |  (+) | SDB (+) |  (+) |  (+) |  (+) |  (+) |  (+) |
########################################################################
#
# File Name: NodeIterator.py
#
#
"""
Node Iterators from DOM Level 2. Allows "flat" iteration over nodes.
WWW: http://4suite.com/4DOM e-mail: support@4suite.com
Copyright (c) 2000 Fourthought Inc, USA. All Rights Reserved.
See http://4suite.com/COPYRIGHT for license and copyright information
"""
from NodeFilter import NodeFilter
from xml.dom import NoModificationAllowedErr
from xml.dom import InvalidStateErr
class NodeIterator:
def __init__(self, root, whatToShow, filter, expandEntityReferences):
self.__dict__['root'] = root
self.__dict__['filter'] = filter
self.__dict__['expandEntityReferences'] = expandEntityReferences
self.__dict__['whatToShow'] = whatToShow
self.__dict__['_atStart'] = 1
self.__dict__['_atEnd'] = 0
self.__dict__['_current'] = root
self.__dict__['_nodeStack'] = []
self.__dict__['_detached'] = 0
def __setattr__(self, name, value):
if name in ['root', 'filter', 'expandEntityReferences', 'whatToShow']:
raise NoModificationAllowedErr()
self.__dict__[name] = value
def _get_root(self):
return self.root
def _get_filter(self):
return self.filter
def _get_expandEntityReferences(self):
return self.expandEntityReferences
def _get_whatToShow(self):
return self.whatToShow
def nextNode(self):
if self._detached:
raise InvalidStateErr()
next_node = self._advance()
while (next_node and not (
self._checkWhatToShow(next_node) and
self._checkFilter(next_node) == NodeFilter.FILTER_ACCEPT)):
next_node = self._advance()
return next_node
def previousNode(self):
if self._detached:
raise InvalidStateErr()
prev_node = self._regress()
while (prev_node and not (
self._checkWhatToShow(prev_node) and
self._checkFilter(prev_node) == NodeFilter.FILTER_ACCEPT)):
prev_node = self._regress()
return prev_node
def detach(self):
self._detached = 1
def _advance(self):
node = None
if self._atStart:
# First time through
self._atStart = 0
node = self._current
elif not self._atEnd:
current = self._current
if current.firstChild:
# Do children first
node = current.firstChild
else:
# Now try the siblings
while current is not self.root:
if current.nextSibling:
node = current.nextSibling
break
# We are at the end of a branch, starting going back up
current = current.parentNode
else:
node = None
if node:
self._current = node
else:
self._atEnd = 1
return node
def _regress(self):
node = None
if self._atEnd:
self._atEnd = 0
node = self._current
elif not self._atStart:
current = self._current
if current is self.root:
node = None
elif current.previousSibling:
node = current.previousSibling
if node.lastChild:
node = node.lastChild
else:
node = current.parentNode
if node:
self._current = node
else:
self._atStart = 1
return node
def _checkWhatToShow(self, node):
show_bit = 1 << (node.nodeType - 1)
return self.whatToShow & show_bit
def _checkFilter(self, node):
if self.filter:
return self.filter.acceptNode(node)
else:
return NodeFilter.FILTER_ACCEPT
|