PyCIM/PrettyPrintXML.py

   1 # Copyright (C) 2010 Richard Lincoln
   2 #
   3 # This library is free software; you can redistribute it and/or
   4 # modify it under the terms of the GNU Lesser General Public
   5 # License as published by the Free Software Foundation; either
   6 # version 2.1 of the License, or (at your option) any later version.
   7 #
   8 # This library is distributed in the hope that it will be useful,
   9 # but WITHOUT ANY WARRANTY; without even the implied warranty of
  10 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
  11 # Lesser General Public License for more details.
  12 #
  13 # You should have received a copy of the GNU Lesser General Public
  14 # License along with this library; if not, write to the Free Software
  15 # Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA, USA
  16
  17 from xml.etree.cElementTree import parse, tostring
  18
  19 def xmlpp(source):
  20     root = parse(source).getroot()
  21     indent(root)
  22     return tostring(root)
  23
  24 def indent(elem, level=0):
  25     i = "\n" + level*"  "
  26     if len(elem):
  27         if not elem.text or not elem.text.strip():
  28             elem.text = i + "  "
  29         if not elem.tail or not elem.tail.strip():
  30             elem.tail = i
  31         for elem in elem:
  32             indent(elem, level+1)
  33         if not elem.tail or not elem.tail.strip():
  34             elem.tail = i
  35     else:
  36         if level and (not elem.tail or not elem.tail.strip()):
  37             elem.tail = i