app/django/bin/profiling/gather_profile_stats.py
author Todd Larsen <tlarsen@google.com>
Wed, 13 Aug 2008 21:45:59 +0000
changeset 70 0e1704e650ad
parent 54 03e267d67478
child 323 ff1a9aa48cfd
permissions -rw-r--r--
Re-arrange mock-up paths for program/docs so that they go from less-specific (e.g. ghop2008, gsoc2009 program linknames) to more-specific (tos, rules, faqs document linknames). This matches the pattern of all of the other view paths, from "view descriptors" (like program/docs) to the specific item (ghop2008/rules).

"""
gather_profile_stats.py /path/to/dir/of/profiles

Note that the aggregated profiles must be read with pstats.Stats, not
hotshot.stats (the formats are incompatible)
"""

from hotshot import stats
import pstats
import sys, os

def gather_stats(p):
    profiles = {}
    for f in os.listdir(p):
        if f.endswith('.agg.prof'):
            path = f[:-9]
            prof = pstats.Stats(os.path.join(p, f))
        elif f.endswith('.prof'):
            bits = f.split('.')
            path = ".".join(bits[:-3])
            prof = stats.load(os.path.join(p, f))
        else:
            continue
        print "Processing %s" % f
        if path in profiles:
            profiles[path].add(prof)
        else:
            profiles[path] = prof
        os.unlink(os.path.join(p, f))
    for (path, prof) in profiles.items():
        prof.dump_stats(os.path.join(p, "%s.agg.prof" % path))
    
if __name__ == '__main__':
    gather_stats(sys.argv[1])