Tools/scripts/hotshotmain.py

   1 #!/usr/bin/env python
   2 # -*- coding: iso-8859-1 -*-
   3
   4 """
   5 Run a Python script under hotshot's control.
   6
   7 Adapted from a posting on python-dev by Walter Dörwald
   8
   9 usage %prog [ %prog args ] filename [ filename args ]
  10
  11 Any arguments after the filename are used as sys.argv for the filename.
  12 """
  13
  14 import sys
  15 import optparse
  16 import os
  17 import hotshot
  18 import hotshot.stats
  19
  20 PROFILE = "hotshot.prof"
  21
  22 def run_hotshot(filename, profile, args):
  23     prof = hotshot.Profile(profile)
  24     sys.path.insert(0, os.path.dirname(filename))
  25     sys.argv = [filename] + args
  26     prof.run("execfile(%r)" % filename)
  27     prof.close()
  28     stats = hotshot.stats.load(profile)
  29     stats.sort_stats("time", "calls")
  30
  31     # print_stats uses unadorned print statements, so the only way
  32     # to force output to stderr is to reassign sys.stdout temporarily
  33     save_stdout = sys.stdout
  34     sys.stdout = sys.stderr
  35     stats.print_stats()
  36     sys.stdout = save_stdout
  37
  38     return 0
  39
  40 def main(args):
  41     parser = optparse.OptionParser(__doc__)
  42     parser.disable_interspersed_args()
  43     parser.add_option("-p", "--profile", action="store", default=PROFILE,
  44                       dest="profile", help='Specify profile file to use')
  45     (options, args) = parser.parse_args(args)
  46
  47     if len(args) == 0:
  48         parser.print_help("missing script to execute")
  49         return 1
  50
  51     filename = args[0]
  52     return run_hotshot(filename, options.profile, args[1:])
  53
  54 if __name__ == "__main__":
  55     sys.exit(main(sys.argv[1:]))