#

# Copyright (c) 2009 Nokia Corporation and/or its subsidiary(-ies).

# All rights reserved.

# This component and the accompanying materials are made available

# under the terms of "Eclipse Public License v1.0"

# which accompanies this distribution, and is available

# at the URL "http://www.eclipse.org/legal/epl-v10.html".

#

# Initial Contributors:

# Nokia Corporation - initial contribution.

#

# Contributors:

#

# Description: 

#

try:

    from elementtree.ElementTree import *

except ImportError:

    from xml.etree.ElementTree import *

class ElementTreeNs(ElementTree):

    """ 

    A class inherited from elementtree.ElementTree that can write

    xml elements with qualified names. The xml namespaces that are defined 

    to the self.namespases are written out to the root element of the given

    root element.

    Example:

    elem = ElementTreeNs.Element('{http://www.test.com}test') 

    etreens = ElementTreeNs(elem, None, {'foo' : 'http://www.test.com'})

    outf = open('test.xml','w')

    ElementTreeNs.write(outf)

    content of test.xml would be

    <foo:test xmlns:foo="http://www.test.com"/>

    """

    def __init__(self, element=None, file=None, namespaces={}):

        ElementTree.__init__(self, element, file)

        self.namespaces = namespaces

    def write(self, file, encoding="us-ascii"):

        """

        Writes the element tree to a file, as XML with the existing namespaces

        @param file A file name, or a file object opened for writing.

        @param encoding Optional output encoding (default is US-ASCII).

        """

        assert self._root is not None

        if not hasattr(file, "write"):

            file = open(file, "wb")

        if not encoding:

            encoding = "us-ascii"

        elif encoding != "utf-8" and encoding != "us-ascii":

            file.write("<?xml version='1.0' encoding='%s'?>\n" % encoding)

        # set the namespaces for the root element

        for ns in self.namespaces:

            self._root.set('xmlns:%s' % self.namespaces[ns], ns)

        self._write(file, self._root, encoding, self.namespaces)

def tostring(element, namespaces, encoding=None):

    """

    """

    class dummy:

        pass

    data = []

    file = dummy()

    file.write = data.append

    ElementTreeNs(element, None, namespaces).write(file, encoding)

    return "".join(data)