1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
|
#----------------------------------------------------------------------------
# Name: xmlprettyprinter.py
# Purpose:
#
# Author: John Spurling
#
# Created: 9/21/04
# CVS-ID: $Id$
# Copyright: (c) 2004-2005 ActiveGrid, Inc.
# License: wxWindows License
#----------------------------------------------------------------------------
from activegrid.util.lang import *
ifDefPy()
import xml.sax
endIfDef()
class XMLPrettyPrinter(xml.sax.ContentHandler):
def __init__(self, indentationChar=' ', newlineChar='\n'):
self.xmlOutput = ''
self.indentationLevel = 0
self.indentationChar = indentationChar
self.elementStack = []
self.newlineChar = newlineChar
self.hitCharData = False
## ContentHandler methods
def startElement(self, name, attrs):
indentation = self.newlineChar + (self.indentationChar * self.indentationLevel)
# build attribute string
attrstring = ''
for attr in attrs.getNames():
value = attrs[attr]
attrstring += ' %s="%s"' % (attr, value)
self.xmlOutput += '%s<%s%s>' % (indentation, name, attrstring)
self.indentationLevel += 1
self.elementStack.append(name)
self.hitCharData = False
def characters(self, content):
## print "--> characters(%s)" % content
self.xmlOutput += content
self.hitCharData = True
def endElement(self, name):
self.indentationLevel -= 1
indentation = ''
if not self.hitCharData:
indentation += self.newlineChar + (self.indentationChar * self.indentationLevel)
## indentation += self.indentationChar * self.indentationLevel
else:
self.hitCharData = False
## self.xmlOutput += '%s</%s>%s' % (indentation, self.elementStack.pop(), self.newlineChar)
self.xmlOutput += '%s</%s>' % (indentation, self.elementStack.pop())
def getXMLString(self):
return self.xmlOutput[1:]
def xmlprettyprint(xmlstr, spaces=4):
xpp = XMLPrettyPrinter(indentationChar=' ' * spaces)
xml.sax.parseString(xmlstr, xpp)
return xpp.getXMLString()
if isMain(__name__):
simpleTestString = """<one>some text<two anattr="booga">two's data</two></one>"""
print xmlprettyprint(simpleTestString)
|
