Robot Framework output file size statistics

print_output_stats.py

import xml.etree.ElementTree as ET


def sizes(element, result):
    if element.tag not in result:
        result[element.tag] = [0, 0]
    result[element.tag][1] += 1
    size_before = result[element.tag][0]
    this_size = 5 + 2*len(element.tag) + \
            len(element.text or "") + \
            sum(4 + len(k) + len(v) for k, v in element.attrib.items()) + \
            sum(1 + sizes(child, result) for child in element)
    result[element.tag][0] = size_before + this_size
    return this_size

def print_stats(filename):
    root = ET.parse(filename).getroot()
    result = {}
    sizes(root, result)
    res = reversed(sorted([(size, calls, k) for (k, [size, calls]) in  result.items()]))
    print("Element\t\tSize\t\tCount")
    print("="*80)
    for s, c, name in res:
        print("%s\t\t%s\t\t%s" % (name, s, c))
    print("-"*80)

if __name__ == "__main__":
    import sys
    print_stats(sys.argv[1])