30 lines
814 B
Python
30 lines
814 B
Python
#!/usr/bin/env python
|
|
import sys
|
|
|
|
def prof_main(argv):
|
|
import hotshot, hotshot.stats
|
|
def usage():
|
|
print 'usage: %s module.function [args ...]' % argv[0]
|
|
return 100
|
|
args = argv[1:]
|
|
if len(args) < 1: return usage()
|
|
name = args.pop(0)
|
|
prof = name+'.prof'
|
|
i = name.rindex('.')
|
|
(modname, funcname) = (name[:i], name[i+1:])
|
|
module = __import__(modname, fromlist=1)
|
|
func = getattr(module, funcname)
|
|
if args:
|
|
args.insert(0, argv[0])
|
|
prof = hotshot.Profile(prof)
|
|
prof.runcall(lambda : func(args))
|
|
prof.close()
|
|
else:
|
|
stats = hotshot.stats.load(prof)
|
|
stats.strip_dirs()
|
|
stats.sort_stats('time', 'calls')
|
|
stats.print_stats(1000)
|
|
return
|
|
|
|
if __name__ == '__main__': sys.exit(prof_main(sys.argv))
|