root/tools/venkman.py

import os, re, sys
from xml.dom.minidom import parseString as dom
from optparse import OptionParser, OptionValueError
from math import ceil

VERSION = '0.2'
USAGE = "%prog [options] [filename-filter [...]]"
DEFAULT_SORTKEYS = ['avgtime', 'totaltime', 'callcount']
class Item:
    which = ('avgtime', 'mintime', 'maxtime', 'totaltime', 'summary', 'callcount', 'filename', 'startline', 'endline')
    def __init__(self, elem):
        self.attrs = dict()
        for i in range(0, elem.attributes.length):
            x = elem.attributes.item(i)
            if not x.localName in self.which:
                continue
            val = x.nodeValue
            try:
                val = float(val)
            except:
                pass
            self.attrs[x.localName] = val
        self.attrs['avgtime'] = ceil(self['totaltime'] * 1000 / self['callcount']) / 1000.0
    def __repr__(self):
        return self['filename'] + ": " + self['summary']
    def __getitem__(self, x):
        return self.attrs[x]

def printTable(inlist, sortkey, fns):
    s = inlist[:]
    if len(fns):
        s = filter(lambda x: x['filename'] in fns, inlist)
    s.sort(lambda x,y: cmp(y[sortkey], x[sortkey]))
    print "<h1>Sorted by: %s</h1>" % sortkey
    print '<table border="1">'
    for i in s:
        print '<tr><th colspan="2">%s</th></tr>' % i['filename']
        print '<tr><td>%s</td><td>%s</td></tr>' % (str(i[sortkey]), i['summary'])
    print '</table>'


def callback_opts_keys(option, opt, value, parser):
    if value not in Item.which:
        raise OptionValueError("sortkey must be one of the following:\n%s" % ', '.join(Item.which))
    parser.skeys += value,

def main():
    print >> sys.stderr, """Venkman Profiler XML parser/sorter
(C) 2007 Nils Maier - MPL-Tri license
"""
    parser = OptionParser(version=VERSION, usage=USAGE)
    parser.add_option(
        '-f',
        '--file',
        type="string",
        help="file to parse (default: stdin)",
        metavar="FILE"
        )
    parser.add_option(
        '-s',
        '--sortkey',
        metavar='KEY',
        action="callback",
        type="string",
        help="which keys to use for sorting (default: %s)" % ', '.join(DEFAULT_SORTKEYS),
        callback=callback_opts_keys
        )
    parser.skeys = []
    opts, args = parser.parse_args(sys.argv)
    args = args[1:]

    if not opts.file:
        opts.file = sys.stdin
    else:
        opts.file = open(opts.file, "r")
    if not len(parser.skeys):
        parser.skeys = DEFAULT_SORTKEYS

    doc = dom(opts.file.read())

    l = []
    for e in doc.getElementsByTagName('item'):
        l += Item(e),

    for k in parser.skeys:
        printTable(l, k, args)
    

if __name__ == '__main__':
    main()