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