1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
|
#!/usr/bin/env python
import sys
import pstats
prg = sys.argv[0]
try:
name = sys.argv[1]
except IndexError:
print "%s <filename> [<number> [<field>]] " % prg
print "%s -^" % (' ' * len(prg))
sys.exit(1)
try:
number = int(sys.argv[2])
except ValueError:
print "%s <filename> [<number> [<field>]] " % prg
print "%s -^" % (' ' * len(prg))
sys.exit(1)
except IndexError:
number = 0
try:
field = sys.argv[3]
except IndexError:
field = 'time'
options = {
'calls': 'call count',
'cumulative': 'cumulative time',
'file': 'file name',
'module': 'file name',
'pcalls': 'primitive call count',
'line': 'line number',
'name': 'function name',
'nfl': 'name/file/line',
'stdname': 'standard name',
'time': 'internal time',
}
if field not in options:
print 'invalid sorting field, valid enties are :\n%s' % '\n'.join(
" %-10s : %s" % (k, v) for (k, v) in options.iteritems()
)
sys.exit(1)
try:
stats = pstats.Stats(name)
except IOError:
print "can not open the file %s" % name
sys.exit(1)
if number:
print stats.strip_dirs().sort_stats(field).print_stats(number)
else:
print stats.print_stats()
|
