#!/usr/bin/env python | |
# -*- coding: iso-8859-1 -*- | |
""" | |
Run a Python script under hotshot's control. | |
Adapted from a posting on python-dev by Walter Dörwald | |
usage %prog [ %prog args ] filename [ filename args ] | |
Any arguments after the filename are used as sys.argv for the filename. | |
""" | |
import sys | |
import optparse | |
import os | |
import hotshot | |
import hotshot.stats | |
PROFILE = "hotshot.prof" | |
def run_hotshot(filename, profile, args): | |
prof = hotshot.Profile(profile) | |
sys.path.insert(0, os.path.dirname(filename)) | |
sys.argv = [filename] + args | |
prof.run("execfile(%r)" % filename) | |
prof.close() | |
stats = hotshot.stats.load(profile) | |
stats.sort_stats("time", "calls") | |
# print_stats uses unadorned print statements, so the only way | |
# to force output to stderr is to reassign sys.stdout temporarily | |
save_stdout = sys.stdout | |
sys.stdout = sys.stderr | |
stats.print_stats() | |
sys.stdout = save_stdout | |
return 0 | |
def main(args): | |
parser = optparse.OptionParser(__doc__) | |
parser.disable_interspersed_args() | |
parser.add_option("-p", "--profile", action="store", default=PROFILE, | |
dest="profile", help='Specify profile file to use') | |
(options, args) = parser.parse_args(args) | |
if len(args) == 0: | |
parser.print_help("missing script to execute") | |
return 1 | |
filename = args[0] | |
return run_hotshot(filename, options.profile, args[1:]) | |
if __name__ == "__main__": | |
sys.exit(main(sys.argv[1:])) |