|
1 # |
|
2 # Copyright (c) 2009 Nokia Corporation and/or its subsidiary(-ies). |
|
3 # All rights reserved. |
|
4 # This component and the accompanying materials are made available |
|
5 # under the terms of "Eclipse Public License v1.0" |
|
6 # which accompanies this distribution, and is available |
|
7 # at the URL "http://www.eclipse.org/legal/epl-v10.html". |
|
8 # |
|
9 # Initial Contributors: |
|
10 # Nokia Corporation - initial contribution. |
|
11 # |
|
12 # Contributors: |
|
13 # |
|
14 # Description: |
|
15 # |
|
16 |
|
17 try: |
|
18 from elementtree.ElementTree import * |
|
19 except ImportError: |
|
20 from xml.etree.ElementTree import * |
|
21 |
|
22 |
|
23 class ElementTreeNs(ElementTree): |
|
24 """ |
|
25 A class inherited from elementtree.ElementTree that can write |
|
26 xml elements with qualified names. The xml namespaces that are defined |
|
27 to the self.namespases are written out to the root element of the given |
|
28 root element. |
|
29 |
|
30 Example: |
|
31 |
|
32 elem = ElementTreeNs.Element('{http://www.test.com}test') |
|
33 etreens = ElementTreeNs(elem, None, {'foo' : 'http://www.test.com'}) |
|
34 outf = open('test.xml','w') |
|
35 ElementTreeNs.write(outf) |
|
36 |
|
37 content of test.xml would be |
|
38 <foo:test xmlns:foo="http://www.test.com"/> |
|
39 """ |
|
40 def __init__(self, element=None, file=None, namespaces={}): |
|
41 ElementTree.__init__(self, element, file) |
|
42 self.namespaces = namespaces |
|
43 |
|
44 def write(self, file, encoding="us-ascii"): |
|
45 """ |
|
46 Writes the element tree to a file, as XML with the existing namespaces |
|
47 |
|
48 @param file A file name, or a file object opened for writing. |
|
49 @param encoding Optional output encoding (default is US-ASCII). |
|
50 """ |
|
51 assert self._root is not None |
|
52 if not hasattr(file, "write"): |
|
53 file = open(file, "wb") |
|
54 if not encoding: |
|
55 encoding = "us-ascii" |
|
56 elif encoding != "utf-8" and encoding != "us-ascii": |
|
57 file.write("<?xml version='1.0' encoding='%s'?>\n" % encoding) |
|
58 # set the namespaces for the root element |
|
59 for ns in self.namespaces: |
|
60 self._root.set('xmlns:%s' % self.namespaces[ns], ns) |
|
61 self._write(file, self._root, encoding, self.namespaces) |
|
62 |
|
63 |
|
64 def tostring(element, namespaces, encoding=None): |
|
65 """ |
|
66 """ |
|
67 class dummy: |
|
68 pass |
|
69 data = [] |
|
70 file = dummy() |
|
71 file.write = data.append |
|
72 ElementTreeNs(element, None, namespaces).write(file, encoding) |
|
73 return "".join(data) |
|
74 |