script/show_test_time: attempt py3 compat
[garming/samba-autobuild/.git] / script / show_test_time
1 #!/usr/bin/python
2 from __future__ import print_function
3 import optparse
4 import os.path
5 import subprocess
6 import sys
7
8 parser = optparse.OptionParser()
9 parser.add_option("--limit", dest="limit", type=int,
10                   help="Limit to this number of output entries.", default=0)
11 (opts, args) = parser.parse_args()
12
13 third_party_path = os.path.join(os.path.dirname(sys.argv[0]), "..", "lib")
14
15 durations = {}
16
17 cmd = "subunit-1to2 | subunit-ls --times --no-passthrough"
18
19 p = subprocess.Popen(cmd, stdout=subprocess.PIPE, stdin=sys.stdin, shell=True)
20 for l in p.stdout:
21     l = l.strip()
22     (name, duration) = l.rsplit(" ", 1)
23     durations[name] = float(duration)
24
25 if opts.limit:
26     print("Top %d tests by run time:" % opts.limit)
27
28 for i, (name, length) in enumerate(sorted(
29         durations.items(), key=lambda x: x[1], reverse=True)):
30     if opts.limit and i == opts.limit:
31         break
32     print("%d: %s -> %ds" % (i+1, name, length))