ardupilot/Tools/CodeStyle/xmlpretty.py

62 lines
2.2 KiB
Python
Raw Normal View History

2015-12-30 00:38:19 -04:00
#!/usr/bin/python
import xml.dom.minidom as minidom
from sys import exit, argv, stderr
import re
dom = minidom.parse(argv[1])
def foreach_tree(doc, root, func, level=0):
func(doc, root, level)
childNodes = root.childNodes[:]
for node in childNodes:
foreach_tree(doc, node, func, level+1)
def strip_indent(doc, node, level):
if node.nodeType == node.TEXT_NODE and re.match(r"^\s+$", node.nodeValue):
node.parentNode.removeChild(node)
node.unlink()
def strip_comment_whitespace(doc, node, level):
if node.nodeType == node.COMMENT_NODE:
node.nodeValue = re.sub(r"\s+", " ", node.nodeValue)
def strip_comments_completely(doc, node, level):
if node.nodeType == node.COMMENT_NODE:
node.parentNode.removeChild(node)
node.unlink()
def strip_text_whitespace(doc, node, level):
if node.nodeType == node.TEXT_NODE:
node.nodeValue = re.sub(r"\s+", " ", node.nodeValue).strip()
def strip_text_completely(doc, node, level):
if node.nodeType == node.TEXT_NODE:
node.parentNode.removeChild(node)
node.unlink()
def auto_indent(doc, node, level):
if level > 0 and node.parentNode.nodeName not in ("description", "field", "param", "include"):
node.parentNode.insertBefore(doc.createTextNode("\n%s" % (" "*4*level)), node)
if node.nextSibling is None:
node.parentNode.appendChild(doc.createTextNode("\n%s" % (" "*4*(level-1))))
def next_non_text_sibling(node):
ret = node.nextSibling
while ret is not None and ret.nodeType == node.TEXT_NODE:
ret = ret.nextSibling
return ret
def auto_space(doc, node, level):
if level > 0 and node.childNodes is not None and len(node.childNodes) > 1 and next_non_text_sibling(node) is not None:
node.parentNode.insertBefore(doc.createTextNode("\n"), node.nextSibling)
foreach_tree(dom, dom.documentElement, strip_indent)
foreach_tree(dom, dom.documentElement, strip_comment_whitespace)
foreach_tree(dom, dom.documentElement, strip_text_whitespace)
foreach_tree(dom, dom.documentElement, auto_indent)
foreach_tree(dom, dom.documentElement, auto_space)
print "<?xml version='1.0'?>"
print dom.documentElement.toxml()