63 lines
2.2 KiB
Python
63 lines
2.2 KiB
Python
from __future__ import absolute_import, division, unicode_literals
|
|
|
|
from xml.dom.pulldom import START_ELEMENT, END_ELEMENT, \
|
|
COMMENT, IGNORABLE_WHITESPACE, CHARACTERS
|
|
|
|
from . import _base
|
|
|
|
from ..constants import voidElements
|
|
|
|
|
|
class TreeWalker(_base.TreeWalker):
|
|
def __iter__(self):
|
|
ignore_until = None
|
|
previous = None
|
|
for event in self.tree:
|
|
if previous is not None and \
|
|
(ignore_until is None or previous[1] is ignore_until):
|
|
if previous[1] is ignore_until:
|
|
ignore_until = None
|
|
for token in self.tokens(previous, event):
|
|
yield token
|
|
if token["type"] == "EmptyTag":
|
|
ignore_until = previous[1]
|
|
previous = event
|
|
if ignore_until is None or previous[1] is ignore_until:
|
|
for token in self.tokens(previous, None):
|
|
yield token
|
|
elif ignore_until is not None:
|
|
raise ValueError("Illformed DOM event stream: void element without END_ELEMENT")
|
|
|
|
def tokens(self, event, next):
|
|
type, node = event
|
|
if type == START_ELEMENT:
|
|
name = node.nodeName
|
|
namespace = node.namespaceURI
|
|
attrs = {}
|
|
for attr in list(node.attributes.keys()):
|
|
attr = node.getAttributeNode(attr)
|
|
attrs[(attr.namespaceURI, attr.localName)] = attr.value
|
|
if name in voidElements:
|
|
for token in self.emptyTag(namespace,
|
|
name,
|
|
attrs,
|
|
not next or next[1] is not node):
|
|
yield token
|
|
else:
|
|
yield self.startTag(namespace, name, attrs)
|
|
|
|
elif type == END_ELEMENT:
|
|
name = node.nodeName
|
|
namespace = node.namespaceURI
|
|
if name not in voidElements:
|
|
yield self.endTag(namespace, name)
|
|
|
|
elif type == COMMENT:
|
|
yield self.comment(node.nodeValue)
|
|
|
|
elif type in (IGNORABLE_WHITESPACE, CHARACTERS):
|
|
for token in self.text(node.nodeValue):
|
|
yield token
|
|
|
|
else:
|
|
yield self.unknown(type)
|