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:]))