| from __future__ import absolute_import, division, unicode_literals | |
| from xml.dom import Node | |
| from . import base | |
| class TreeWalker(base.NonRecursiveTreeWalker): | |
| def getNodeDetails(self, node): | |
| if node.nodeType == Node.DOCUMENT_TYPE_NODE: | |
| return base.DOCTYPE, node.name, node.publicId, node.systemId | |
| elif node.nodeType in (Node.TEXT_NODE, Node.CDATA_SECTION_NODE): | |
| return base.TEXT, node.nodeValue | |
| elif node.nodeType == Node.ELEMENT_NODE: | |
| attrs = {} | |
| for attr in list(node.attributes.keys()): | |
| attr = node.getAttributeNode(attr) | |
| if attr.namespaceURI: | |
| attrs[(attr.namespaceURI, attr.localName)] = attr.value | |
| else: | |
| attrs[(None, attr.name)] = attr.value | |
| return (base.ELEMENT, node.namespaceURI, node.nodeName, | |
| attrs, node.hasChildNodes()) | |
| elif node.nodeType == Node.COMMENT_NODE: | |
| return base.COMMENT, node.nodeValue | |
| elif node.nodeType in (Node.DOCUMENT_NODE, Node.DOCUMENT_FRAGMENT_NODE): | |
| return (base.DOCUMENT,) | |
| else: | |
| return base.UNKNOWN, node.nodeType | |
| def getFirstChild(self, node): | |
| return node.firstChild | |
| def getNextSibling(self, node): | |
| return node.nextSibling | |
| def getParentNode(self, node): | |
| return node.parentNode | |