import strings

import util

def escape(str_):

    return strings.to_unicode(cgi.escape(str_, True)).encode('ascii',

        'xmlcharrefreplace')

def unescape(str_): return HTMLParser().unescape(str_)

def get_id(node): return node.getAttribute('id')

def set_id(node, id_): node.setAttribute('id', id_)

def is_completely_empty(node): return node.firstChild == None

def has_one_child(node):

    return node.firstChild != None and node.firstChild.nextSibling == None

class NodeIter:

    def __init__(self, node): self.child = node.firstChild

        self.child = self.child.nextSibling

        return child

def is_comment(node): return node.nodeType == Node.COMMENT_NODE

        if not is_comment(child): return False

    return True

class NodeElemIter:

    def __init__(self, node): self.child = node.firstChild

def last_elem(node): return NodeElemReverseIter(node).next()

class NodeParentIter:

    def __init__(self, node): self.node = node

        self.node = self.node.parentNode

        return node

def is_text_node(node): return node.nodeType == Node.TEXT_NODE

def is_text(node): return has_one_child(node) and is_text_node(node.firstChild)

    def __init__(self, iterable):

        util.CheckedIter.__init__(self, is_text_node_entry, iterable)

def set_child(node, name, value):

    '''Note: does not remove any existing child of the same name'''

    if util.is_str(new): new = node.ownerDocument.createTextNode(new)

    replace(node, new)

def by_tag_name(node, name, last_only=False):

    '''last_only optimization returns last matching node'''

            if last_only: break

    return children

def create_doc(root='_'):

    return minidom.getDOMImplementation().createDocument(None, root, None)

prettyxml_config = dict(addindent='    ', newl='\n')

toprettyxml_config = prettyxml_config.copy()

util.rename_key(toprettyxml_config, 'addindent', 'indent')

minidom._write_data = lambda writer, data: writer.write(escape(data))

minidom.Node.__iter__ = lambda self: NodeIter(self)