cpython-lru_cache-weakref: Lib/pstats.py annotate

annotate Lib/pstats.py @ 80661:94de77bd0b4b 2.7

Issue #16194: document imp.load_dynamic problems when read several extension modules from the same binary object file.

author	Andrew Svetlov <andrew.svetlov@gmail.com>
date	Thu, 29 Nov 2012 19:51:28 +0200
parents	ad1cc596bedb
children

rev	line source
11706 b6f52c1ef6da More trivial comment -> docstring transformations by Ka-Ping Yee, Guido van Rossum <guido@python.org> parents: 10581 diff changeset	1 """Class for printing reports on profiled python code."""
b6f52c1ef6da More trivial comment -> docstring transformations by Ka-Ping Yee, Guido van Rossum <guido@python.org> parents: 10581 diff changeset	2
71039 02150e60636b update profile license (closes #12417) Benjamin Peterson <benjamin@python.org> parents: 63551 diff changeset	3 # Written by James Roskind
1679 5290428bad37 Companion module for new profile.py Guido van Rossum <guido@python.org> parents: diff changeset	4 # Based on prior profile module by Sjoerd Mullender...
5290428bad37 Companion module for new profile.py Guido van Rossum <guido@python.org> parents: diff changeset	5 # which was hacked somewhat by: Guido van Rossum
71039 02150e60636b update profile license (closes #12417) Benjamin Peterson <benjamin@python.org> parents: 63551 diff changeset	6
02150e60636b update profile license (closes #12417) Benjamin Peterson <benjamin@python.org> parents: 63551 diff changeset	7 # Copyright Disney Enterprises, Inc. All Rights Reserved.
02150e60636b update profile license (closes #12417) Benjamin Peterson <benjamin@python.org> parents: 63551 diff changeset	8 # Licensed to PSF under a Contributor Agreement
1679 5290428bad37 Companion module for new profile.py Guido van Rossum <guido@python.org> parents: diff changeset	9 #
71039 02150e60636b update profile license (closes #12417) Benjamin Peterson <benjamin@python.org> parents: 63551 diff changeset	10 # Licensed under the Apache License, Version 2.0 (the "License");
02150e60636b update profile license (closes #12417) Benjamin Peterson <benjamin@python.org> parents: 63551 diff changeset	11 # you may not use this file except in compliance with the License.
02150e60636b update profile license (closes #12417) Benjamin Peterson <benjamin@python.org> parents: 63551 diff changeset	12 # You may obtain a copy of the License at
15870 010d03853546 Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 13620 diff changeset	13 #
71039 02150e60636b update profile license (closes #12417) Benjamin Peterson <benjamin@python.org> parents: 63551 diff changeset	14 # http://www.apache.org/licenses/LICENSE-2.0
15870 010d03853546 Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 13620 diff changeset	15 #
71039 02150e60636b update profile license (closes #12417) Benjamin Peterson <benjamin@python.org> parents: 63551 diff changeset	16 # Unless required by applicable law or agreed to in writing, software
02150e60636b update profile license (closes #12417) Benjamin Peterson <benjamin@python.org> parents: 63551 diff changeset	17 # distributed under the License is distributed on an "AS IS" BASIS,
02150e60636b update profile license (closes #12417) Benjamin Peterson <benjamin@python.org> parents: 63551 diff changeset	18 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND,
02150e60636b update profile license (closes #12417) Benjamin Peterson <benjamin@python.org> parents: 63551 diff changeset	19 # either express or implied. See the License for the specific language
02150e60636b update profile license (closes #12417) Benjamin Peterson <benjamin@python.org> parents: 63551 diff changeset	20 # governing permissions and limitations under the License.
1679 5290428bad37 Companion module for new profile.py Guido van Rossum <guido@python.org> parents: diff changeset	21
5290428bad37 Companion module for new profile.py Guido van Rossum <guido@python.org> parents: diff changeset	22
37873 d35db84ca072 Allow pstats.Stats creator to specify an alternate to stdout. Skip Montanaro <skip@pobox.com> parents: 36459 diff changeset	23 import sys
1679 5290428bad37 Companion module for new profile.py Guido van Rossum <guido@python.org> parents: diff changeset	24 import os
5290428bad37 Companion module for new profile.py Guido van Rossum <guido@python.org> parents: diff changeset	25 import time
5290428bad37 Companion module for new profile.py Guido van Rossum <guido@python.org> parents: diff changeset	26 import marshal
6519 24da035ceb6e Convert all remaining simple cases of regex usage to re usage. Guido van Rossum <guido@python.org> parents: 6452 diff changeset	27 import re
60315 5dda3f32a2ce Add tests for cmp_to_key. Raymond Hettinger <python@rcn.com> parents: 60167 diff changeset	28 from functools import cmp_to_key
1679 5290428bad37 Companion module for new profile.py Guido van Rossum <guido@python.org> parents: diff changeset	29
16562 623456576d6a __all__ for several more modules Skip Montanaro <skip@pobox.com> parents: 16492 diff changeset	30 __all__ = ["Stats"]
623456576d6a __all__ for several more modules Skip Montanaro <skip@pobox.com> parents: 16492 diff changeset	31
1679 5290428bad37 Companion module for new profile.py Guido van Rossum <guido@python.org> parents: diff changeset	32 class Stats:
15870 010d03853546 Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 13620 diff changeset	33 """This class is used for creating reports from data generated by the
010d03853546 Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 13620 diff changeset	34 Profile class. It is a "friend" of that class, and imports data either
010d03853546 Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 13620 diff changeset	35 by direct access to members of Profile class, or by reading in a dictionary
010d03853546 Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 13620 diff changeset	36 that was emitted (via marshal) from the Profile class.
11706 b6f52c1ef6da More trivial comment -> docstring transformations by Ka-Ping Yee, Guido van Rossum <guido@python.org> parents: 10581 diff changeset	37
15870 010d03853546 Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 13620 diff changeset	38 The big change from the previous Profiler (in terms of raw functionality)
010d03853546 Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 13620 diff changeset	39 is that an "add()" method has been provided to combine Stats from
010d03853546 Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 13620 diff changeset	40 several distinct profile runs. Both the constructor and the add()
010d03853546 Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 13620 diff changeset	41 method now take arbitrarily many file names as arguments.
11706 b6f52c1ef6da More trivial comment -> docstring transformations by Ka-Ping Yee, Guido van Rossum <guido@python.org> parents: 10581 diff changeset	42
15870 010d03853546 Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 13620 diff changeset	43 All the print methods now take an argument that indicates how many lines
010d03853546 Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 13620 diff changeset	44 to print. If the arg is a floating point number between 0 and 1.0, then
010d03853546 Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 13620 diff changeset	45 it is taken as a decimal percentage of the available lines to be printed
010d03853546 Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 13620 diff changeset	46 (e.g., .1 means print 10% of all available lines). If it is an integer,
010d03853546 Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 13620 diff changeset	47 it is taken to mean the number of lines of data that you wish to have
010d03853546 Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 13620 diff changeset	48 printed.
11706 b6f52c1ef6da More trivial comment -> docstring transformations by Ka-Ping Yee, Guido van Rossum <guido@python.org> parents: 10581 diff changeset	49
15870 010d03853546 Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 13620 diff changeset	50 The sort_stats() method now processes some additional options (i.e., in
37873 d35db84ca072 Allow pstats.Stats creator to specify an alternate to stdout. Skip Montanaro <skip@pobox.com> parents: 36459 diff changeset	51 addition to the old -1, 0, 1, or 2). It takes an arbitrary number of
d35db84ca072 Allow pstats.Stats creator to specify an alternate to stdout. Skip Montanaro <skip@pobox.com> parents: 36459 diff changeset	52 quoted strings to select the sort order. For example sort_stats('time',
d35db84ca072 Allow pstats.Stats creator to specify an alternate to stdout. Skip Montanaro <skip@pobox.com> parents: 36459 diff changeset	53 'name') sorts on the major key of 'internal function time', and on the
d35db84ca072 Allow pstats.Stats creator to specify an alternate to stdout. Skip Montanaro <skip@pobox.com> parents: 36459 diff changeset	54 minor key of 'the name of the function'. Look at the two tables in
d35db84ca072 Allow pstats.Stats creator to specify an alternate to stdout. Skip Montanaro <skip@pobox.com> parents: 36459 diff changeset	55 sort_stats() and get_sort_arg_defs(self) for more examples.
15870 010d03853546 Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 13620 diff changeset	56
63602 e971c104a5c2 Merged revisions 83523 via svnmerge from Georg Brandl <georg@python.org> parents: 63549 diff changeset	57 All methods return self, so you can string together commands like:
15870 010d03853546 Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 13620 diff changeset	58 Stats('foo', 'goo').strip_dirs().sort_stats('calls').\
010d03853546 Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 13620 diff changeset	59 print_stats(5).print_callers(5)
010d03853546 Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 13620 diff changeset	60 """
11706 b6f52c1ef6da More trivial comment -> docstring transformations by Ka-Ping Yee, Guido van Rossum <guido@python.org> parents: 10581 diff changeset	61
37873 d35db84ca072 Allow pstats.Stats creator to specify an alternate to stdout. Skip Montanaro <skip@pobox.com> parents: 36459 diff changeset	62 def __init__(self, args, *kwds):
d35db84ca072 Allow pstats.Stats creator to specify an alternate to stdout. Skip Montanaro <skip@pobox.com> parents: 36459 diff changeset	63 # I can't figure out how to explictly specify a stream keyword arg
d35db84ca072 Allow pstats.Stats creator to specify an alternate to stdout. Skip Montanaro <skip@pobox.com> parents: 36459 diff changeset	64 # with *args:
d35db84ca072 Allow pstats.Stats creator to specify an alternate to stdout. Skip Montanaro <skip@pobox.com> parents: 36459 diff changeset	65 # def __init__(self, *args, stream=sys.stdout): ...
d35db84ca072 Allow pstats.Stats creator to specify an alternate to stdout. Skip Montanaro <skip@pobox.com> parents: 36459 diff changeset	66 # so I use **kwds and sqauwk if something unexpected is passed in.
d35db84ca072 Allow pstats.Stats creator to specify an alternate to stdout. Skip Montanaro <skip@pobox.com> parents: 36459 diff changeset	67 self.stream = sys.stdout
d35db84ca072 Allow pstats.Stats creator to specify an alternate to stdout. Skip Montanaro <skip@pobox.com> parents: 36459 diff changeset	68 if "stream" in kwds:
d35db84ca072 Allow pstats.Stats creator to specify an alternate to stdout. Skip Montanaro <skip@pobox.com> parents: 36459 diff changeset	69 self.stream = kwds["stream"]
d35db84ca072 Allow pstats.Stats creator to specify an alternate to stdout. Skip Montanaro <skip@pobox.com> parents: 36459 diff changeset	70 del kwds["stream"]
d35db84ca072 Allow pstats.Stats creator to specify an alternate to stdout. Skip Montanaro <skip@pobox.com> parents: 36459 diff changeset	71 if kwds:
d35db84ca072 Allow pstats.Stats creator to specify an alternate to stdout. Skip Montanaro <skip@pobox.com> parents: 36459 diff changeset	72 keys = kwds.keys()
d35db84ca072 Allow pstats.Stats creator to specify an alternate to stdout. Skip Montanaro <skip@pobox.com> parents: 36459 diff changeset	73 keys.sort()
d35db84ca072 Allow pstats.Stats creator to specify an alternate to stdout. Skip Montanaro <skip@pobox.com> parents: 36459 diff changeset	74 extras = ", ".join(["%s=%s" % (k, kwds[k]) for k in keys])
d35db84ca072 Allow pstats.Stats creator to specify an alternate to stdout. Skip Montanaro <skip@pobox.com> parents: 36459 diff changeset	75 raise ValueError, "unrecognized keyword args: %s" % extras
15870 010d03853546 Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 13620 diff changeset	76 if not len(args):
010d03853546 Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 13620 diff changeset	77 arg = None
010d03853546 Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 13620 diff changeset	78 else:
010d03853546 Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 13620 diff changeset	79 arg = args[0]
010d03853546 Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 13620 diff changeset	80 args = args[1:]
010d03853546 Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 13620 diff changeset	81 self.init(arg)
28089 14f65d661084 Get rid of many apply() calls. Guido van Rossum <guido@python.org> parents: 23634 diff changeset	82 self.add(*args)
15870 010d03853546 Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 13620 diff changeset	83
010d03853546 Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 13620 diff changeset	84 def init(self, arg):
010d03853546 Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 13620 diff changeset	85 self.all_callees = None # calc only if needed
010d03853546 Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 13620 diff changeset	86 self.files = []
010d03853546 Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 13620 diff changeset	87 self.fcn_list = None
010d03853546 Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 13620 diff changeset	88 self.total_tt = 0
010d03853546 Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 13620 diff changeset	89 self.total_calls = 0
010d03853546 Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 13620 diff changeset	90 self.prim_calls = 0
010d03853546 Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 13620 diff changeset	91 self.max_name_len = 0
010d03853546 Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 13620 diff changeset	92 self.top_level = {}
010d03853546 Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 13620 diff changeset	93 self.stats = {}
010d03853546 Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 13620 diff changeset	94 self.sort_arg_dict = {}
010d03853546 Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 13620 diff changeset	95 self.load_stats(arg)
010d03853546 Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 13620 diff changeset	96 trouble = 1
010d03853546 Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 13620 diff changeset	97 try:
010d03853546 Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 13620 diff changeset	98 self.get_top_level_stats()
010d03853546 Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 13620 diff changeset	99 trouble = 0
010d03853546 Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 13620 diff changeset	100 finally:
010d03853546 Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 13620 diff changeset	101 if trouble:
37873 d35db84ca072 Allow pstats.Stats creator to specify an alternate to stdout. Skip Montanaro <skip@pobox.com> parents: 36459 diff changeset	102 print >> self.stream, "Invalid timing data",
d35db84ca072 Allow pstats.Stats creator to specify an alternate to stdout. Skip Montanaro <skip@pobox.com> parents: 36459 diff changeset	103 if self.files: print >> self.stream, self.files[-1],
d35db84ca072 Allow pstats.Stats creator to specify an alternate to stdout. Skip Montanaro <skip@pobox.com> parents: 36459 diff changeset	104 print >> self.stream
1679 5290428bad37 Companion module for new profile.py Guido van Rossum <guido@python.org> parents: diff changeset	105
15870 010d03853546 Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 13620 diff changeset	106 def load_stats(self, arg):
010d03853546 Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 13620 diff changeset	107 if not arg: self.stats = {}
40397 454eff2706d3 Bug #1603321: make pstats.Stats accept Unicode file paths. Georg Brandl <georg@python.org> parents: 38742 diff changeset	108 elif isinstance(arg, basestring):
15870 010d03853546 Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 13620 diff changeset	109 f = open(arg, 'rb')
010d03853546 Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 13620 diff changeset	110 self.stats = marshal.load(f)
010d03853546 Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 13620 diff changeset	111 f.close()
010d03853546 Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 13620 diff changeset	112 try:
010d03853546 Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 13620 diff changeset	113 file_stats = os.stat(arg)
23625 65cf478950ff Replaced obsolete stat module constants with equivalent attributes Raymond Hettinger <python@rcn.com> parents: 23622 diff changeset	114 arg = time.ctime(file_stats.st_mtime) + " " + arg
15870 010d03853546 Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 13620 diff changeset	115 except: # in case this is not unix
010d03853546 Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 13620 diff changeset	116 pass
010d03853546 Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 13620 diff changeset	117 self.files = [ arg ]
010d03853546 Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 13620 diff changeset	118 elif hasattr(arg, 'create_stats'):
010d03853546 Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 13620 diff changeset	119 arg.create_stats()
010d03853546 Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 13620 diff changeset	120 self.stats = arg.stats
010d03853546 Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 13620 diff changeset	121 arg.stats = {}
010d03853546 Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 13620 diff changeset	122 if not self.stats:
79930 ad1cc596bedb #16303: remove extra quotes from exception and add (). Initial patch by Vladimir Rutsky. Ezio Melotti <ezio.melotti@gmail.com> parents: 79576 diff changeset	123 raise TypeError("Cannot create or construct a %r object from %r"
ad1cc596bedb #16303: remove extra quotes from exception and add (). Initial patch by Vladimir Rutsky. Ezio Melotti <ezio.melotti@gmail.com> parents: 79576 diff changeset	124 % (self.__class__, arg))
15870 010d03853546 Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 13620 diff changeset	125 return
1679 5290428bad37 Companion module for new profile.py Guido van Rossum <guido@python.org> parents: diff changeset	126
15870 010d03853546 Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 13620 diff changeset	127 def get_top_level_stats(self):
20308 ca2bc5cdaa8e Widespread random code cleanup. Tim Peters <tim.peters@gmail.com> parents: 19216 diff changeset	128 for func, (cc, nc, tt, ct, callers) in self.stats.items():
ca2bc5cdaa8e Widespread random code cleanup. Tim Peters <tim.peters@gmail.com> parents: 19216 diff changeset	129 self.total_calls += nc
ca2bc5cdaa8e Widespread random code cleanup. Tim Peters <tim.peters@gmail.com> parents: 19216 diff changeset	130 self.prim_calls += cc
ca2bc5cdaa8e Widespread random code cleanup. Tim Peters <tim.peters@gmail.com> parents: 19216 diff changeset	131 self.total_tt += tt
48146 1803de5ae0dc Silence -3 warnings in pstats: a dict.has_key() usage and backport solution to Brett Cannon <bcannon@gmail.com> parents: 44833 diff changeset	132 if ("jprofile", 0, "profiler") in callers:
15870 010d03853546 Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 13620 diff changeset	133 self.top_level[func] = None
010d03853546 Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 13620 diff changeset	134 if len(func_std_string(func)) > self.max_name_len:
010d03853546 Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 13620 diff changeset	135 self.max_name_len = len(func_std_string(func))
1679 5290428bad37 Companion module for new profile.py Guido van Rossum <guido@python.org> parents: diff changeset	136
15870 010d03853546 Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 13620 diff changeset	137 def add(self, *arg_list):
010d03853546 Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 13620 diff changeset	138 if not arg_list: return self
28089 14f65d661084 Get rid of many apply() calls. Guido van Rossum <guido@python.org> parents: 23634 diff changeset	139 if len(arg_list) > 1: self.add(*arg_list[1:])
15870 010d03853546 Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 13620 diff changeset	140 other = arg_list[0]
20308 ca2bc5cdaa8e Widespread random code cleanup. Tim Peters <tim.peters@gmail.com> parents: 19216 diff changeset	141 if type(self) != type(other) or self.__class__ != other.__class__:
15870 010d03853546 Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 13620 diff changeset	142 other = Stats(other)
20308 ca2bc5cdaa8e Widespread random code cleanup. Tim Peters <tim.peters@gmail.com> parents: 19216 diff changeset	143 self.files += other.files
ca2bc5cdaa8e Widespread random code cleanup. Tim Peters <tim.peters@gmail.com> parents: 19216 diff changeset	144 self.total_calls += other.total_calls
ca2bc5cdaa8e Widespread random code cleanup. Tim Peters <tim.peters@gmail.com> parents: 19216 diff changeset	145 self.prim_calls += other.prim_calls
ca2bc5cdaa8e Widespread random code cleanup. Tim Peters <tim.peters@gmail.com> parents: 19216 diff changeset	146 self.total_tt += other.total_tt
23634 dbee2f628a90 Replaced .keys() with dictionary iterators Raymond Hettinger <python@rcn.com> parents: 23625 diff changeset	147 for func in other.top_level:
15870 010d03853546 Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 13620 diff changeset	148 self.top_level[func] = None
1679 5290428bad37 Companion module for new profile.py Guido van Rossum <guido@python.org> parents: diff changeset	149
15870 010d03853546 Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 13620 diff changeset	150 if self.max_name_len < other.max_name_len:
010d03853546 Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 13620 diff changeset	151 self.max_name_len = other.max_name_len
010d03853546 Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 13620 diff changeset	152
010d03853546 Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 13620 diff changeset	153 self.fcn_list = None
1679 5290428bad37 Companion module for new profile.py Guido van Rossum <guido@python.org> parents: diff changeset	154
23634 dbee2f628a90 Replaced .keys() with dictionary iterators Raymond Hettinger <python@rcn.com> parents: 23625 diff changeset	155 for func, stat in other.stats.iteritems():
23620 a1de0bab5475 SF 563203. Replaced 'has_key()' with 'in'. Raymond Hettinger <python@rcn.com> parents: 20309 diff changeset	156 if func in self.stats:
15870 010d03853546 Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 13620 diff changeset	157 old_func_stat = self.stats[func]
010d03853546 Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 13620 diff changeset	158 else:
010d03853546 Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 13620 diff changeset	159 old_func_stat = (0, 0, 0, 0, {},)
23634 dbee2f628a90 Replaced .keys() with dictionary iterators Raymond Hettinger <python@rcn.com> parents: 23625 diff changeset	160 self.stats[func] = add_func_stats(old_func_stat, stat)
15870 010d03853546 Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 13620 diff changeset	161 return self
1679 5290428bad37 Companion module for new profile.py Guido van Rossum <guido@python.org> parents: diff changeset	162
29111 f4d406acf1ca - add a dump_stats() method similar to that of the profile.Profile class Fred Drake <fdrake@acm.org> parents: 28089 diff changeset	163 def dump_stats(self, filename):
f4d406acf1ca - add a dump_stats() method similar to that of the profile.Profile class Fred Drake <fdrake@acm.org> parents: 28089 diff changeset	164 """Write the profile data to a file we know how to load back."""
f4d406acf1ca - add a dump_stats() method similar to that of the profile.Profile class Fred Drake <fdrake@acm.org> parents: 28089 diff changeset	165 f = file(filename, 'wb')
f4d406acf1ca - add a dump_stats() method similar to that of the profile.Profile class Fred Drake <fdrake@acm.org> parents: 28089 diff changeset	166 try:
f4d406acf1ca - add a dump_stats() method similar to that of the profile.Profile class Fred Drake <fdrake@acm.org> parents: 28089 diff changeset	167 marshal.dump(self.stats, f)
f4d406acf1ca - add a dump_stats() method similar to that of the profile.Profile class Fred Drake <fdrake@acm.org> parents: 28089 diff changeset	168 finally:
f4d406acf1ca - add a dump_stats() method similar to that of the profile.Profile class Fred Drake <fdrake@acm.org> parents: 28089 diff changeset	169 f.close()
f4d406acf1ca - add a dump_stats() method similar to that of the profile.Profile class Fred Drake <fdrake@acm.org> parents: 28089 diff changeset	170
15870 010d03853546 Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 13620 diff changeset	171 # list the tuple indices and directions for sorting,
010d03853546 Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 13620 diff changeset	172 # along with some printable description
20308 ca2bc5cdaa8e Widespread random code cleanup. Tim Peters <tim.peters@gmail.com> parents: 19216 diff changeset	173 sort_arg_dict_default = {
ca2bc5cdaa8e Widespread random code cleanup. Tim Peters <tim.peters@gmail.com> parents: 19216 diff changeset	174 "calls" : (((1,-1), ), "call count"),
79576 69b310f7acfa Issue #14900: Add aliases for sorting params for pstat to follow column names from pstat output. Andrew Svetlov <andrew.svetlov@gmail.com> parents: 79570 diff changeset	175 "ncalls" : (((1,-1), ), "call count"),
69b310f7acfa Issue #14900: Add aliases for sorting params for pstat to follow column names from pstat output. Andrew Svetlov <andrew.svetlov@gmail.com> parents: 79570 diff changeset	176 "cumtime" : (((3,-1), ), "cumulative time"),
20308 ca2bc5cdaa8e Widespread random code cleanup. Tim Peters <tim.peters@gmail.com> parents: 19216 diff changeset	177 "cumulative": (((3,-1), ), "cumulative time"),
ca2bc5cdaa8e Widespread random code cleanup. Tim Peters <tim.peters@gmail.com> parents: 19216 diff changeset	178 "file" : (((4, 1), ), "file name"),
79576 69b310f7acfa Issue #14900: Add aliases for sorting params for pstat to follow column names from pstat output. Andrew Svetlov <andrew.svetlov@gmail.com> parents: 79570 diff changeset	179 "filename" : (((4, 1), ), "file name"),
20308 ca2bc5cdaa8e Widespread random code cleanup. Tim Peters <tim.peters@gmail.com> parents: 19216 diff changeset	180 "line" : (((5, 1), ), "line number"),
ca2bc5cdaa8e Widespread random code cleanup. Tim Peters <tim.peters@gmail.com> parents: 19216 diff changeset	181 "module" : (((4, 1), ), "file name"),
ca2bc5cdaa8e Widespread random code cleanup. Tim Peters <tim.peters@gmail.com> parents: 19216 diff changeset	182 "name" : (((6, 1), ), "function name"),
ca2bc5cdaa8e Widespread random code cleanup. Tim Peters <tim.peters@gmail.com> parents: 19216 diff changeset	183 "nfl" : (((6, 1),(4, 1),(5, 1),), "name/file/line"),
79570 cc3e27b7da41 Issue #14900: Distuguish call count and primitive call count in pstat output. Andrew Svetlov <andrew.svetlov@gmail.com> parents: 71580 diff changeset	184 "pcalls" : (((0,-1), ), "primitive call count"),
20308 ca2bc5cdaa8e Widespread random code cleanup. Tim Peters <tim.peters@gmail.com> parents: 19216 diff changeset	185 "stdname" : (((7, 1), ), "standard name"),
ca2bc5cdaa8e Widespread random code cleanup. Tim Peters <tim.peters@gmail.com> parents: 19216 diff changeset	186 "time" : (((2,-1), ), "internal time"),
79576 69b310f7acfa Issue #14900: Add aliases for sorting params for pstat to follow column names from pstat output. Andrew Svetlov <andrew.svetlov@gmail.com> parents: 79570 diff changeset	187 "tottime" : (((2,-1), ), "internal time"),
15870 010d03853546 Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 13620 diff changeset	188 }
1679 5290428bad37 Companion module for new profile.py Guido van Rossum <guido@python.org> parents: diff changeset	189
15870 010d03853546 Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 13620 diff changeset	190 def get_sort_arg_defs(self):
010d03853546 Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 13620 diff changeset	191 """Expand all abbreviations that are unique."""
010d03853546 Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 13620 diff changeset	192 if not self.sort_arg_dict:
010d03853546 Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 13620 diff changeset	193 self.sort_arg_dict = dict = {}
010d03853546 Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 13620 diff changeset	194 bad_list = {}
23634 dbee2f628a90 Replaced .keys() with dictionary iterators Raymond Hettinger <python@rcn.com> parents: 23625 diff changeset	195 for word, tup in self.sort_arg_dict_default.iteritems():
15870 010d03853546 Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 13620 diff changeset	196 fragment = word
010d03853546 Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 13620 diff changeset	197 while fragment:
010d03853546 Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 13620 diff changeset	198 if not fragment:
010d03853546 Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 13620 diff changeset	199 break
23620 a1de0bab5475 SF 563203. Replaced 'has_key()' with 'in'. Raymond Hettinger <python@rcn.com> parents: 20309 diff changeset	200 if fragment in dict:
15870 010d03853546 Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 13620 diff changeset	201 bad_list[fragment] = 0
010d03853546 Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 13620 diff changeset	202 break
23634 dbee2f628a90 Replaced .keys() with dictionary iterators Raymond Hettinger <python@rcn.com> parents: 23625 diff changeset	203 dict[fragment] = tup
15870 010d03853546 Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 13620 diff changeset	204 fragment = fragment[:-1]
23634 dbee2f628a90 Replaced .keys() with dictionary iterators Raymond Hettinger <python@rcn.com> parents: 23625 diff changeset	205 for word in bad_list:
15870 010d03853546 Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 13620 diff changeset	206 del dict[word]
010d03853546 Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 13620 diff changeset	207 return self.sort_arg_dict
1679 5290428bad37 Companion module for new profile.py Guido van Rossum <guido@python.org> parents: diff changeset	208
15870 010d03853546 Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 13620 diff changeset	209 def sort_stats(self, *field):
010d03853546 Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 13620 diff changeset	210 if not field:
010d03853546 Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 13620 diff changeset	211 self.fcn_list = 0
010d03853546 Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 13620 diff changeset	212 return self
63604 6f81d6f9a6d1 Fix merging glitches. Georg Brandl <georg@python.org> parents: 63602 diff changeset	213 if len(field) == 1 and isinstance(field[0], (int, long)):
15870 010d03853546 Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 13620 diff changeset	214 # Be compatible with old profiler
20308 ca2bc5cdaa8e Widespread random code cleanup. Tim Peters <tim.peters@gmail.com> parents: 19216 diff changeset	215 field = [ {-1: "stdname",
63602 e971c104a5c2 Merged revisions 83523 via svnmerge from Georg Brandl <georg@python.org> parents: 63549 diff changeset	216 0: "calls",
e971c104a5c2 Merged revisions 83523 via svnmerge from Georg Brandl <georg@python.org> parents: 63549 diff changeset	217 1: "time",
e971c104a5c2 Merged revisions 83523 via svnmerge from Georg Brandl <georg@python.org> parents: 63549 diff changeset	218 2: "cumulative"}[field[0]] ]
1679 5290428bad37 Companion module for new profile.py Guido van Rossum <guido@python.org> parents: diff changeset	219
15870 010d03853546 Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 13620 diff changeset	220 sort_arg_defs = self.get_sort_arg_defs()
010d03853546 Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 13620 diff changeset	221 sort_tuple = ()
010d03853546 Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 13620 diff changeset	222 self.sort_type = ""
010d03853546 Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 13620 diff changeset	223 connector = ""
010d03853546 Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 13620 diff changeset	224 for word in field:
010d03853546 Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 13620 diff changeset	225 sort_tuple = sort_tuple + sort_arg_defs[word][0]
20308 ca2bc5cdaa8e Widespread random code cleanup. Tim Peters <tim.peters@gmail.com> parents: 19216 diff changeset	226 self.sort_type += connector + sort_arg_defs[word][1]
15870 010d03853546 Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 13620 diff changeset	227 connector = ", "
010d03853546 Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 13620 diff changeset	228
010d03853546 Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 13620 diff changeset	229 stats_list = []
23634 dbee2f628a90 Replaced .keys() with dictionary iterators Raymond Hettinger <python@rcn.com> parents: 23625 diff changeset	230 for func, (cc, nc, tt, ct, callers) in self.stats.iteritems():
20308 ca2bc5cdaa8e Widespread random code cleanup. Tim Peters <tim.peters@gmail.com> parents: 19216 diff changeset	231 stats_list.append((cc, nc, tt, ct) + func +
ca2bc5cdaa8e Widespread random code cleanup. Tim Peters <tim.peters@gmail.com> parents: 19216 diff changeset	232 (func_std_string(func), func))
15870 010d03853546 Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 13620 diff changeset	233
60315 5dda3f32a2ce Add tests for cmp_to_key. Raymond Hettinger <python@rcn.com> parents: 60167 diff changeset	234 stats_list.sort(key=cmp_to_key(TupleComp(sort_tuple).compare))
15870 010d03853546 Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 13620 diff changeset	235
010d03853546 Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 13620 diff changeset	236 self.fcn_list = fcn_list = []
010d03853546 Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 13620 diff changeset	237 for tuple in stats_list:
010d03853546 Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 13620 diff changeset	238 fcn_list.append(tuple[-1])
010d03853546 Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 13620 diff changeset	239 return self
010d03853546 Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 13620 diff changeset	240
010d03853546 Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 13620 diff changeset	241 def reverse_order(self):
20308 ca2bc5cdaa8e Widespread random code cleanup. Tim Peters <tim.peters@gmail.com> parents: 19216 diff changeset	242 if self.fcn_list:
ca2bc5cdaa8e Widespread random code cleanup. Tim Peters <tim.peters@gmail.com> parents: 19216 diff changeset	243 self.fcn_list.reverse()
15870 010d03853546 Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 13620 diff changeset	244 return self
010d03853546 Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 13620 diff changeset	245
010d03853546 Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 13620 diff changeset	246 def strip_dirs(self):
010d03853546 Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 13620 diff changeset	247 oldstats = self.stats
010d03853546 Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 13620 diff changeset	248 self.stats = newstats = {}
010d03853546 Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 13620 diff changeset	249 max_name_len = 0
23634 dbee2f628a90 Replaced .keys() with dictionary iterators Raymond Hettinger <python@rcn.com> parents: 23625 diff changeset	250 for func, (cc, nc, tt, ct, callers) in oldstats.iteritems():
15870 010d03853546 Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 13620 diff changeset	251 newfunc = func_strip_path(func)
010d03853546 Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 13620 diff changeset	252 if len(func_std_string(newfunc)) > max_name_len:
010d03853546 Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 13620 diff changeset	253 max_name_len = len(func_std_string(newfunc))
010d03853546 Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 13620 diff changeset	254 newcallers = {}
23634 dbee2f628a90 Replaced .keys() with dictionary iterators Raymond Hettinger <python@rcn.com> parents: 23625 diff changeset	255 for func2, caller in callers.iteritems():
dbee2f628a90 Replaced .keys() with dictionary iterators Raymond Hettinger <python@rcn.com> parents: 23625 diff changeset	256 newcallers[func_strip_path(func2)] = caller
15870 010d03853546 Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 13620 diff changeset	257
23620 a1de0bab5475 SF 563203. Replaced 'has_key()' with 'in'. Raymond Hettinger <python@rcn.com> parents: 20309 diff changeset	258 if newfunc in newstats:
20308 ca2bc5cdaa8e Widespread random code cleanup. Tim Peters <tim.peters@gmail.com> parents: 19216 diff changeset	259 newstats[newfunc] = add_func_stats(
ca2bc5cdaa8e Widespread random code cleanup. Tim Peters <tim.peters@gmail.com> parents: 19216 diff changeset	260 newstats[newfunc],
ca2bc5cdaa8e Widespread random code cleanup. Tim Peters <tim.peters@gmail.com> parents: 19216 diff changeset	261 (cc, nc, tt, ct, newcallers))
15870 010d03853546 Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 13620 diff changeset	262 else:
010d03853546 Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 13620 diff changeset	263 newstats[newfunc] = (cc, nc, tt, ct, newcallers)
010d03853546 Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 13620 diff changeset	264 old_top = self.top_level
010d03853546 Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 13620 diff changeset	265 self.top_level = new_top = {}
23634 dbee2f628a90 Replaced .keys() with dictionary iterators Raymond Hettinger <python@rcn.com> parents: 23625 diff changeset	266 for func in old_top:
15870 010d03853546 Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 13620 diff changeset	267 new_top[func_strip_path(func)] = None
010d03853546 Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 13620 diff changeset	268
010d03853546 Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 13620 diff changeset	269 self.max_name_len = max_name_len
010d03853546 Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 13620 diff changeset	270
010d03853546 Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 13620 diff changeset	271 self.fcn_list = None
010d03853546 Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 13620 diff changeset	272 self.all_callees = None
010d03853546 Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 13620 diff changeset	273 return self
1679 5290428bad37 Companion module for new profile.py Guido van Rossum <guido@python.org> parents: diff changeset	274
15870 010d03853546 Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 13620 diff changeset	275 def calc_callees(self):
010d03853546 Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 13620 diff changeset	276 if self.all_callees: return
010d03853546 Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 13620 diff changeset	277 self.all_callees = all_callees = {}
23634 dbee2f628a90 Replaced .keys() with dictionary iterators Raymond Hettinger <python@rcn.com> parents: 23625 diff changeset	278 for func, (cc, nc, tt, ct, callers) in self.stats.iteritems():
23620 a1de0bab5475 SF 563203. Replaced 'has_key()' with 'in'. Raymond Hettinger <python@rcn.com> parents: 20309 diff changeset	279 if not func in all_callees:
15870 010d03853546 Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 13620 diff changeset	280 all_callees[func] = {}
23634 dbee2f628a90 Replaced .keys() with dictionary iterators Raymond Hettinger <python@rcn.com> parents: 23625 diff changeset	281 for func2, caller in callers.iteritems():
23620 a1de0bab5475 SF 563203. Replaced 'has_key()' with 'in'. Raymond Hettinger <python@rcn.com> parents: 20309 diff changeset	282 if not func2 in all_callees:
15870 010d03853546 Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 13620 diff changeset	283 all_callees[func2] = {}
23634 dbee2f628a90 Replaced .keys() with dictionary iterators Raymond Hettinger <python@rcn.com> parents: 23625 diff changeset	284 all_callees[func2][func] = caller
15870 010d03853546 Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 13620 diff changeset	285 return
1679 5290428bad37 Companion module for new profile.py Guido van Rossum <guido@python.org> parents: diff changeset	286
15870 010d03853546 Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 13620 diff changeset	287 #******************************************************************
010d03853546 Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 13620 diff changeset	288 # The following functions support actual printing of reports
010d03853546 Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 13620 diff changeset	289 #******************************************************************
010d03853546 Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 13620 diff changeset	290
010d03853546 Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 13620 diff changeset	291 # Optional "amount" is either a line count, or a percentage of lines.
1679 5290428bad37 Companion module for new profile.py Guido van Rossum <guido@python.org> parents: diff changeset	292
15870 010d03853546 Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 13620 diff changeset	293 def eval_print_amount(self, sel, list, msg):
010d03853546 Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 13620 diff changeset	294 new_list = list
63604 6f81d6f9a6d1 Fix merging glitches. Georg Brandl <georg@python.org> parents: 63602 diff changeset	295 if isinstance(sel, basestring):
63602 e971c104a5c2 Merged revisions 83523 via svnmerge from Georg Brandl <georg@python.org> parents: 63549 diff changeset	296 try:
e971c104a5c2 Merged revisions 83523 via svnmerge from Georg Brandl <georg@python.org> parents: 63549 diff changeset	297 rex = re.compile(sel)
e971c104a5c2 Merged revisions 83523 via svnmerge from Georg Brandl <georg@python.org> parents: 63549 diff changeset	298 except re.error:
e971c104a5c2 Merged revisions 83523 via svnmerge from Georg Brandl <georg@python.org> parents: 63549 diff changeset	299 msg += " <Invalid regular expression %r>\n" % sel
e971c104a5c2 Merged revisions 83523 via svnmerge from Georg Brandl <georg@python.org> parents: 63549 diff changeset	300 return new_list, msg
15870 010d03853546 Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 13620 diff changeset	301 new_list = []
010d03853546 Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 13620 diff changeset	302 for func in list:
63602 e971c104a5c2 Merged revisions 83523 via svnmerge from Georg Brandl <georg@python.org> parents: 63549 diff changeset	303 if rex.search(func_std_string(func)):
15870 010d03853546 Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 13620 diff changeset	304 new_list.append(func)
010d03853546 Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 13620 diff changeset	305 else:
010d03853546 Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 13620 diff changeset	306 count = len(list)
63602 e971c104a5c2 Merged revisions 83523 via svnmerge from Georg Brandl <georg@python.org> parents: 63549 diff changeset	307 if isinstance(sel, float) and 0.0 <= sel < 1.0:
20308 ca2bc5cdaa8e Widespread random code cleanup. Tim Peters <tim.peters@gmail.com> parents: 19216 diff changeset	308 count = int(count * sel + .5)
15870 010d03853546 Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 13620 diff changeset	309 new_list = list[:count]
63604 6f81d6f9a6d1 Fix merging glitches. Georg Brandl <georg@python.org> parents: 63602 diff changeset	310 elif isinstance(sel, (int, long)) and 0 <= sel < count:
15870 010d03853546 Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 13620 diff changeset	311 count = sel
010d03853546 Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 13620 diff changeset	312 new_list = list[:count]
010d03853546 Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 13620 diff changeset	313 if len(list) != len(new_list):
63602 e971c104a5c2 Merged revisions 83523 via svnmerge from Georg Brandl <georg@python.org> parents: 63549 diff changeset	314 msg += " List reduced from %r to %r due to restriction <%r>\n" % (
e971c104a5c2 Merged revisions 83523 via svnmerge from Georg Brandl <georg@python.org> parents: 63549 diff changeset	315 len(list), len(new_list), sel)
1679 5290428bad37 Companion module for new profile.py Guido van Rossum <guido@python.org> parents: diff changeset	316
15870 010d03853546 Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 13620 diff changeset	317 return new_list, msg
1679 5290428bad37 Companion module for new profile.py Guido van Rossum <guido@python.org> parents: diff changeset	318
15870 010d03853546 Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 13620 diff changeset	319 def get_print_list(self, sel_list):
010d03853546 Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 13620 diff changeset	320 width = self.max_name_len
010d03853546 Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 13620 diff changeset	321 if self.fcn_list:
63602 e971c104a5c2 Merged revisions 83523 via svnmerge from Georg Brandl <georg@python.org> parents: 63549 diff changeset	322 stat_list = self.fcn_list[:]
15870 010d03853546 Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 13620 diff changeset	323 msg = " Ordered by: " + self.sort_type + '\n'
010d03853546 Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 13620 diff changeset	324 else:
63604 6f81d6f9a6d1 Fix merging glitches. Georg Brandl <georg@python.org> parents: 63602 diff changeset	325 stat_list = self.stats.keys()
15870 010d03853546 Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 13620 diff changeset	326 msg = " Random listing order was used\n"
010d03853546 Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 13620 diff changeset	327
010d03853546 Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 13620 diff changeset	328 for selection in sel_list:
63602 e971c104a5c2 Merged revisions 83523 via svnmerge from Georg Brandl <georg@python.org> parents: 63549 diff changeset	329 stat_list, msg = self.eval_print_amount(selection, stat_list, msg)
15870 010d03853546 Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 13620 diff changeset	330
63602 e971c104a5c2 Merged revisions 83523 via svnmerge from Georg Brandl <georg@python.org> parents: 63549 diff changeset	331 count = len(stat_list)
15870 010d03853546 Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 13620 diff changeset	332
63602 e971c104a5c2 Merged revisions 83523 via svnmerge from Georg Brandl <georg@python.org> parents: 63549 diff changeset	333 if not stat_list:
e971c104a5c2 Merged revisions 83523 via svnmerge from Georg Brandl <georg@python.org> parents: 63549 diff changeset	334 return 0, stat_list
37873 d35db84ca072 Allow pstats.Stats creator to specify an alternate to stdout. Skip Montanaro <skip@pobox.com> parents: 36459 diff changeset	335 print >> self.stream, msg
15870 010d03853546 Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 13620 diff changeset	336 if count < len(self.stats):
010d03853546 Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 13620 diff changeset	337 width = 0
63602 e971c104a5c2 Merged revisions 83523 via svnmerge from Georg Brandl <georg@python.org> parents: 63549 diff changeset	338 for func in stat_list:
15870 010d03853546 Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 13620 diff changeset	339 if len(func_std_string(func)) > width:
010d03853546 Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 13620 diff changeset	340 width = len(func_std_string(func))
63602 e971c104a5c2 Merged revisions 83523 via svnmerge from Georg Brandl <georg@python.org> parents: 63549 diff changeset	341 return width+2, stat_list
15870 010d03853546 Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 13620 diff changeset	342
010d03853546 Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 13620 diff changeset	343 def print_stats(self, *amount):
010d03853546 Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 13620 diff changeset	344 for filename in self.files:
37873 d35db84ca072 Allow pstats.Stats creator to specify an alternate to stdout. Skip Montanaro <skip@pobox.com> parents: 36459 diff changeset	345 print >> self.stream, filename
d35db84ca072 Allow pstats.Stats creator to specify an alternate to stdout. Skip Montanaro <skip@pobox.com> parents: 36459 diff changeset	346 if self.files: print >> self.stream
20308 ca2bc5cdaa8e Widespread random code cleanup. Tim Peters <tim.peters@gmail.com> parents: 19216 diff changeset	347 indent = ' ' * 8
23634 dbee2f628a90 Replaced .keys() with dictionary iterators Raymond Hettinger <python@rcn.com> parents: 23625 diff changeset	348 for func in self.top_level:
37873 d35db84ca072 Allow pstats.Stats creator to specify an alternate to stdout. Skip Montanaro <skip@pobox.com> parents: 36459 diff changeset	349 print >> self.stream, indent, func_get_function_name(func)
15870 010d03853546 Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 13620 diff changeset	350
37873 d35db84ca072 Allow pstats.Stats creator to specify an alternate to stdout. Skip Montanaro <skip@pobox.com> parents: 36459 diff changeset	351 print >> self.stream, indent, self.total_calls, "function calls",
15870 010d03853546 Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 13620 diff changeset	352 if self.total_calls != self.prim_calls:
37873 d35db84ca072 Allow pstats.Stats creator to specify an alternate to stdout. Skip Montanaro <skip@pobox.com> parents: 36459 diff changeset	353 print >> self.stream, "(%d primitive calls)" % self.prim_calls,
66296 a7c22107aecd Merged revisions 86580 via svnmerge from Senthil Kumaran <orsenthil@gmail.com> parents: 63604 diff changeset	354 print >> self.stream, "in %.3f seconds" % self.total_tt
37873 d35db84ca072 Allow pstats.Stats creator to specify an alternate to stdout. Skip Montanaro <skip@pobox.com> parents: 36459 diff changeset	355 print >> self.stream
15870 010d03853546 Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 13620 diff changeset	356 width, list = self.get_print_list(amount)
010d03853546 Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 13620 diff changeset	357 if list:
010d03853546 Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 13620 diff changeset	358 self.print_title()
010d03853546 Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 13620 diff changeset	359 for func in list:
010d03853546 Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 13620 diff changeset	360 self.print_line(func)
37873 d35db84ca072 Allow pstats.Stats creator to specify an alternate to stdout. Skip Montanaro <skip@pobox.com> parents: 36459 diff changeset	361 print >> self.stream
d35db84ca072 Allow pstats.Stats creator to specify an alternate to stdout. Skip Montanaro <skip@pobox.com> parents: 36459 diff changeset	362 print >> self.stream
15870 010d03853546 Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 13620 diff changeset	363 return self
1679 5290428bad37 Companion module for new profile.py Guido van Rossum <guido@python.org> parents: diff changeset	364
15870 010d03853546 Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 13620 diff changeset	365 def print_callees(self, *amount):
010d03853546 Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 13620 diff changeset	366 width, list = self.get_print_list(amount)
010d03853546 Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 13620 diff changeset	367 if list:
010d03853546 Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 13620 diff changeset	368 self.calc_callees()
010d03853546 Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 13620 diff changeset	369
010d03853546 Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 13620 diff changeset	370 self.print_call_heading(width, "called...")
010d03853546 Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 13620 diff changeset	371 for func in list:
23620 a1de0bab5475 SF 563203. Replaced 'has_key()' with 'in'. Raymond Hettinger <python@rcn.com> parents: 20309 diff changeset	372 if func in self.all_callees:
20308 ca2bc5cdaa8e Widespread random code cleanup. Tim Peters <tim.peters@gmail.com> parents: 19216 diff changeset	373 self.print_call_line(width, func, self.all_callees[func])
15870 010d03853546 Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 13620 diff changeset	374 else:
010d03853546 Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 13620 diff changeset	375 self.print_call_line(width, func, {})
37873 d35db84ca072 Allow pstats.Stats creator to specify an alternate to stdout. Skip Montanaro <skip@pobox.com> parents: 36459 diff changeset	376 print >> self.stream
d35db84ca072 Allow pstats.Stats creator to specify an alternate to stdout. Skip Montanaro <skip@pobox.com> parents: 36459 diff changeset	377 print >> self.stream
15870 010d03853546 Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 13620 diff changeset	378 return self
010d03853546 Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 13620 diff changeset	379
010d03853546 Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 13620 diff changeset	380 def print_callers(self, *amount):
010d03853546 Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 13620 diff changeset	381 width, list = self.get_print_list(amount)
010d03853546 Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 13620 diff changeset	382 if list:
010d03853546 Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 13620 diff changeset	383 self.print_call_heading(width, "was called by...")
010d03853546 Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 13620 diff changeset	384 for func in list:
010d03853546 Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 13620 diff changeset	385 cc, nc, tt, ct, callers = self.stats[func]
36459 3abf33264146 Added the cProfile module. Armin Rigo <arigo@tunes.org> parents: 34097 diff changeset	386 self.print_call_line(width, func, callers, "<-")
37873 d35db84ca072 Allow pstats.Stats creator to specify an alternate to stdout. Skip Montanaro <skip@pobox.com> parents: 36459 diff changeset	387 print >> self.stream
d35db84ca072 Allow pstats.Stats creator to specify an alternate to stdout. Skip Montanaro <skip@pobox.com> parents: 36459 diff changeset	388 print >> self.stream
15870 010d03853546 Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 13620 diff changeset	389 return self
010d03853546 Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 13620 diff changeset	390
010d03853546 Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 13620 diff changeset	391 def print_call_heading(self, name_size, column_title):
37873 d35db84ca072 Allow pstats.Stats creator to specify an alternate to stdout. Skip Montanaro <skip@pobox.com> parents: 36459 diff changeset	392 print >> self.stream, "Function ".ljust(name_size) + column_title
36459 3abf33264146 Added the cProfile module. Armin Rigo <arigo@tunes.org> parents: 34097 diff changeset	393 # print sub-header only if we have new-style callers
3abf33264146 Added the cProfile module. Armin Rigo <arigo@tunes.org> parents: 34097 diff changeset	394 subheader = False
3abf33264146 Added the cProfile module. Armin Rigo <arigo@tunes.org> parents: 34097 diff changeset	395 for cc, nc, tt, ct, callers in self.stats.itervalues():
3abf33264146 Added the cProfile module. Armin Rigo <arigo@tunes.org> parents: 34097 diff changeset	396 if callers:
3abf33264146 Added the cProfile module. Armin Rigo <arigo@tunes.org> parents: 34097 diff changeset	397 value = callers.itervalues().next()
3abf33264146 Added the cProfile module. Armin Rigo <arigo@tunes.org> parents: 34097 diff changeset	398 subheader = isinstance(value, tuple)
3abf33264146 Added the cProfile module. Armin Rigo <arigo@tunes.org> parents: 34097 diff changeset	399 break
3abf33264146 Added the cProfile module. Armin Rigo <arigo@tunes.org> parents: 34097 diff changeset	400 if subheader:
37873 d35db84ca072 Allow pstats.Stats creator to specify an alternate to stdout. Skip Montanaro <skip@pobox.com> parents: 36459 diff changeset	401 print >> self.stream, " "*name_size + " ncalls tottime cumtime"
1679 5290428bad37 Companion module for new profile.py Guido van Rossum <guido@python.org> parents: diff changeset	402
36459 3abf33264146 Added the cProfile module. Armin Rigo <arigo@tunes.org> parents: 34097 diff changeset	403 def print_call_line(self, name_size, source, call_dict, arrow="->"):
37873 d35db84ca072 Allow pstats.Stats creator to specify an alternate to stdout. Skip Montanaro <skip@pobox.com> parents: 36459 diff changeset	404 print >> self.stream, func_std_string(source).ljust(name_size) + arrow,
15870 010d03853546 Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 13620 diff changeset	405 if not call_dict:
37873 d35db84ca072 Allow pstats.Stats creator to specify an alternate to stdout. Skip Montanaro <skip@pobox.com> parents: 36459 diff changeset	406 print >> self.stream
15870 010d03853546 Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 13620 diff changeset	407 return
010d03853546 Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 13620 diff changeset	408 clist = call_dict.keys()
010d03853546 Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 13620 diff changeset	409 clist.sort()
010d03853546 Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 13620 diff changeset	410 indent = ""
010d03853546 Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 13620 diff changeset	411 for func in clist:
010d03853546 Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 13620 diff changeset	412 name = func_std_string(func)
36459 3abf33264146 Added the cProfile module. Armin Rigo <arigo@tunes.org> parents: 34097 diff changeset	413 value = call_dict[func]
3abf33264146 Added the cProfile module. Armin Rigo <arigo@tunes.org> parents: 34097 diff changeset	414 if isinstance(value, tuple):
3abf33264146 Added the cProfile module. Armin Rigo <arigo@tunes.org> parents: 34097 diff changeset	415 nc, cc, tt, ct = value
3abf33264146 Added the cProfile module. Armin Rigo <arigo@tunes.org> parents: 34097 diff changeset	416 if nc != cc:
3abf33264146 Added the cProfile module. Armin Rigo <arigo@tunes.org> parents: 34097 diff changeset	417 substats = '%d/%d' % (nc, cc)
3abf33264146 Added the cProfile module. Armin Rigo <arigo@tunes.org> parents: 34097 diff changeset	418 else:
3abf33264146 Added the cProfile module. Armin Rigo <arigo@tunes.org> parents: 34097 diff changeset	419 substats = '%d' % (nc,)
3abf33264146 Added the cProfile module. Armin Rigo <arigo@tunes.org> parents: 34097 diff changeset	420 substats = '%s %s %s %s' % (substats.rjust(7+2*len(indent)),
3abf33264146 Added the cProfile module. Armin Rigo <arigo@tunes.org> parents: 34097 diff changeset	421 f8(tt), f8(ct), name)
3abf33264146 Added the cProfile module. Armin Rigo <arigo@tunes.org> parents: 34097 diff changeset	422 left_width = name_size + 1
3abf33264146 Added the cProfile module. Armin Rigo <arigo@tunes.org> parents: 34097 diff changeset	423 else:
3abf33264146 Added the cProfile module. Armin Rigo <arigo@tunes.org> parents: 34097 diff changeset	424 substats = '%s(%r) %s' % (name, value, f8(self.stats[func][3]))
3abf33264146 Added the cProfile module. Armin Rigo <arigo@tunes.org> parents: 34097 diff changeset	425 left_width = name_size + 3
37873 d35db84ca072 Allow pstats.Stats creator to specify an alternate to stdout. Skip Montanaro <skip@pobox.com> parents: 36459 diff changeset	426 print >> self.stream, indent*left_width + substats
15870 010d03853546 Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 13620 diff changeset	427 indent = " "
010d03853546 Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 13620 diff changeset	428
010d03853546 Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 13620 diff changeset	429 def print_title(self):
37873 d35db84ca072 Allow pstats.Stats creator to specify an alternate to stdout. Skip Montanaro <skip@pobox.com> parents: 36459 diff changeset	430 print >> self.stream, ' ncalls tottime percall cumtime percall',
d35db84ca072 Allow pstats.Stats creator to specify an alternate to stdout. Skip Montanaro <skip@pobox.com> parents: 36459 diff changeset	431 print >> self.stream, 'filename:lineno(function)'
15870 010d03853546 Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 13620 diff changeset	432
010d03853546 Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 13620 diff changeset	433 def print_line(self, func): # hack : should print percentages
010d03853546 Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 13620 diff changeset	434 cc, nc, tt, ct, callers = self.stats[func]
20308 ca2bc5cdaa8e Widespread random code cleanup. Tim Peters <tim.peters@gmail.com> parents: 19216 diff changeset	435 c = str(nc)
15870 010d03853546 Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 13620 diff changeset	436 if nc != cc:
20308 ca2bc5cdaa8e Widespread random code cleanup. Tim Peters <tim.peters@gmail.com> parents: 19216 diff changeset	437 c = c + '/' + str(cc)
37873 d35db84ca072 Allow pstats.Stats creator to specify an alternate to stdout. Skip Montanaro <skip@pobox.com> parents: 36459 diff changeset	438 print >> self.stream, c.rjust(9),
d35db84ca072 Allow pstats.Stats creator to specify an alternate to stdout. Skip Montanaro <skip@pobox.com> parents: 36459 diff changeset	439 print >> self.stream, f8(tt),
15870 010d03853546 Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 13620 diff changeset	440 if nc == 0:
37873 d35db84ca072 Allow pstats.Stats creator to specify an alternate to stdout. Skip Montanaro <skip@pobox.com> parents: 36459 diff changeset	441 print >> self.stream, ' '*8,
15870 010d03853546 Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 13620 diff changeset	442 else:
63528 751c6dd71b5e Merged revisions 77310-77311 via svnmerge from Ezio Melotti <ezio.melotti@gmail.com> parents: 63511 diff changeset	443 print >> self.stream, f8(float(tt)/nc),
37873 d35db84ca072 Allow pstats.Stats creator to specify an alternate to stdout. Skip Montanaro <skip@pobox.com> parents: 36459 diff changeset	444 print >> self.stream, f8(ct),
15870 010d03853546 Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 13620 diff changeset	445 if cc == 0:
37873 d35db84ca072 Allow pstats.Stats creator to specify an alternate to stdout. Skip Montanaro <skip@pobox.com> parents: 36459 diff changeset	446 print >> self.stream, ' '*8,
15870 010d03853546 Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 13620 diff changeset	447 else:
63528 751c6dd71b5e Merged revisions 77310-77311 via svnmerge from Ezio Melotti <ezio.melotti@gmail.com> parents: 63511 diff changeset	448 print >> self.stream, f8(float(ct)/cc),
37873 d35db84ca072 Allow pstats.Stats creator to specify an alternate to stdout. Skip Montanaro <skip@pobox.com> parents: 36459 diff changeset	449 print >> self.stream, func_std_string(func)
15870 010d03853546 Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 13620 diff changeset	450
1679 5290428bad37 Companion module for new profile.py Guido van Rossum <guido@python.org> parents: diff changeset	451 class TupleComp:
15870 010d03853546 Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 13620 diff changeset	452 """This class provides a generic function for comparing any two tuples.
010d03853546 Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 13620 diff changeset	453 Each instance records a list of tuple-indices (from most significant
010d03853546 Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 13620 diff changeset	454 to least significant), and sort direction (ascending or decending) for
010d03853546 Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 13620 diff changeset	455 each tuple-index. The compare functions can then be used as the function
010d03853546 Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 13620 diff changeset	456 argument to the system sort() function when a list of tuples need to be
010d03853546 Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 13620 diff changeset	457 sorted in the instances order."""
11706 b6f52c1ef6da More trivial comment -> docstring transformations by Ka-Ping Yee, Guido van Rossum <guido@python.org> parents: 10581 diff changeset	458
15870 010d03853546 Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 13620 diff changeset	459 def __init__(self, comp_select_list):
010d03853546 Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 13620 diff changeset	460 self.comp_select_list = comp_select_list
1679 5290428bad37 Companion module for new profile.py Guido van Rossum <guido@python.org> parents: diff changeset	461
15870 010d03853546 Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 13620 diff changeset	462 def compare (self, left, right):
010d03853546 Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 13620 diff changeset	463 for index, direction in self.comp_select_list:
010d03853546 Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 13620 diff changeset	464 l = left[index]
010d03853546 Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 13620 diff changeset	465 r = right[index]
010d03853546 Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 13620 diff changeset	466 if l < r:
010d03853546 Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 13620 diff changeset	467 return -direction
010d03853546 Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 13620 diff changeset	468 if l > r:
010d03853546 Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 13620 diff changeset	469 return direction
010d03853546 Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 13620 diff changeset	470 return 0
1679 5290428bad37 Companion module for new profile.py Guido van Rossum <guido@python.org> parents: diff changeset	471
5290428bad37 Companion module for new profile.py Guido van Rossum <guido@python.org> parents: diff changeset	472 #**************************************************************************
20308 ca2bc5cdaa8e Widespread random code cleanup. Tim Peters <tim.peters@gmail.com> parents: 19216 diff changeset	473 # func_name is a triple (file:string, line:int, name:string)
1679 5290428bad37 Companion module for new profile.py Guido van Rossum <guido@python.org> parents: diff changeset	474
5290428bad37 Companion module for new profile.py Guido van Rossum <guido@python.org> parents: diff changeset	475 def func_strip_path(func_name):
29111 f4d406acf1ca - add a dump_stats() method similar to that of the profile.Profile class Fred Drake <fdrake@acm.org> parents: 28089 diff changeset	476 filename, line, name = func_name
f4d406acf1ca - add a dump_stats() method similar to that of the profile.Profile class Fred Drake <fdrake@acm.org> parents: 28089 diff changeset	477 return os.path.basename(filename), line, name
1679 5290428bad37 Companion module for new profile.py Guido van Rossum <guido@python.org> parents: diff changeset	478
5290428bad37 Companion module for new profile.py Guido van Rossum <guido@python.org> parents: diff changeset	479 def func_get_function_name(func):
15870 010d03853546 Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 13620 diff changeset	480 return func[2]
1679 5290428bad37 Companion module for new profile.py Guido van Rossum <guido@python.org> parents: diff changeset	481
5290428bad37 Companion module for new profile.py Guido van Rossum <guido@python.org> parents: diff changeset	482 def func_std_string(func_name): # match what old profile produced
36459 3abf33264146 Added the cProfile module. Armin Rigo <arigo@tunes.org> parents: 34097 diff changeset	483 if func_name[:2] == ('~', 0):
3abf33264146 Added the cProfile module. Armin Rigo <arigo@tunes.org> parents: 34097 diff changeset	484 # special case for built-in functions
3abf33264146 Added the cProfile module. Armin Rigo <arigo@tunes.org> parents: 34097 diff changeset	485 name = func_name[2]
3abf33264146 Added the cProfile module. Armin Rigo <arigo@tunes.org> parents: 34097 diff changeset	486 if name.startswith('<') and name.endswith('>'):
3abf33264146 Added the cProfile module. Armin Rigo <arigo@tunes.org> parents: 34097 diff changeset	487 return '{%s}' % name[1:-1]
3abf33264146 Added the cProfile module. Armin Rigo <arigo@tunes.org> parents: 34097 diff changeset	488 else:
3abf33264146 Added the cProfile module. Armin Rigo <arigo@tunes.org> parents: 34097 diff changeset	489 return name
3abf33264146 Added the cProfile module. Armin Rigo <arigo@tunes.org> parents: 34097 diff changeset	490 else:
3abf33264146 Added the cProfile module. Armin Rigo <arigo@tunes.org> parents: 34097 diff changeset	491 return "%s:%d(%s)" % func_name
1679 5290428bad37 Companion module for new profile.py Guido van Rossum <guido@python.org> parents: diff changeset	492
5290428bad37 Companion module for new profile.py Guido van Rossum <guido@python.org> parents: diff changeset	493 #**************************************************************************
5290428bad37 Companion module for new profile.py Guido van Rossum <guido@python.org> parents: diff changeset	494 # The following functions combine statists for pairs functions.
5290428bad37 Companion module for new profile.py Guido van Rossum <guido@python.org> parents: diff changeset	495 # The bulk of the processing involves correctly handling "call" lists,
15870 010d03853546 Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 13620 diff changeset	496 # such as callers and callees.
1679 5290428bad37 Companion module for new profile.py Guido van Rossum <guido@python.org> parents: diff changeset	497 #**************************************************************************
5290428bad37 Companion module for new profile.py Guido van Rossum <guido@python.org> parents: diff changeset	498
11706 b6f52c1ef6da More trivial comment -> docstring transformations by Ka-Ping Yee, Guido van Rossum <guido@python.org> parents: 10581 diff changeset	499 def add_func_stats(target, source):
15870 010d03853546 Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 13620 diff changeset	500 """Add together all the stats for two profile entries."""
010d03853546 Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 13620 diff changeset	501 cc, nc, tt, ct, callers = source
010d03853546 Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 13620 diff changeset	502 t_cc, t_nc, t_tt, t_ct, t_callers = target
20308 ca2bc5cdaa8e Widespread random code cleanup. Tim Peters <tim.peters@gmail.com> parents: 19216 diff changeset	503 return (cc+t_cc, nc+t_nc, tt+t_tt, ct+t_ct,
15870 010d03853546 Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 13620 diff changeset	504 add_callers(t_callers, callers))
1679 5290428bad37 Companion module for new profile.py Guido van Rossum <guido@python.org> parents: diff changeset	505
5290428bad37 Companion module for new profile.py Guido van Rossum <guido@python.org> parents: diff changeset	506 def add_callers(target, source):
15870 010d03853546 Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 13620 diff changeset	507 """Combine two caller lists in a single list."""
010d03853546 Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 13620 diff changeset	508 new_callers = {}
23634 dbee2f628a90 Replaced .keys() with dictionary iterators Raymond Hettinger <python@rcn.com> parents: 23625 diff changeset	509 for func, caller in target.iteritems():
dbee2f628a90 Replaced .keys() with dictionary iterators Raymond Hettinger <python@rcn.com> parents: 23625 diff changeset	510 new_callers[func] = caller
dbee2f628a90 Replaced .keys() with dictionary iterators Raymond Hettinger <python@rcn.com> parents: 23625 diff changeset	511 for func, caller in source.iteritems():
23620 a1de0bab5475 SF 563203. Replaced 'has_key()' with 'in'. Raymond Hettinger <python@rcn.com> parents: 20309 diff changeset	512 if func in new_callers:
63551 3be2464c0c39 Merged revisions 83533 via svnmerge from Georg Brandl <georg@python.org> parents: 63528 diff changeset	513 if isinstance(caller, tuple):
3be2464c0c39 Merged revisions 83533 via svnmerge from Georg Brandl <georg@python.org> parents: 63528 diff changeset	514 # format used by cProfile
3be2464c0c39 Merged revisions 83533 via svnmerge from Georg Brandl <georg@python.org> parents: 63528 diff changeset	515 new_callers[func] = tuple([i[0] + i[1] for i in
3be2464c0c39 Merged revisions 83533 via svnmerge from Georg Brandl <georg@python.org> parents: 63528 diff changeset	516 zip(caller, new_callers[func])])
3be2464c0c39 Merged revisions 83533 via svnmerge from Georg Brandl <georg@python.org> parents: 63528 diff changeset	517 else:
3be2464c0c39 Merged revisions 83533 via svnmerge from Georg Brandl <georg@python.org> parents: 63528 diff changeset	518 # format used by profile
3be2464c0c39 Merged revisions 83533 via svnmerge from Georg Brandl <georg@python.org> parents: 63528 diff changeset	519 new_callers[func] += caller
15870 010d03853546 Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 13620 diff changeset	520 else:
23634 dbee2f628a90 Replaced .keys() with dictionary iterators Raymond Hettinger <python@rcn.com> parents: 23625 diff changeset	521 new_callers[func] = caller
15870 010d03853546 Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 13620 diff changeset	522 return new_callers
1679 5290428bad37 Companion module for new profile.py Guido van Rossum <guido@python.org> parents: diff changeset	523
5290428bad37 Companion module for new profile.py Guido van Rossum <guido@python.org> parents: diff changeset	524 def count_calls(callers):
15870 010d03853546 Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 13620 diff changeset	525 """Sum the caller statistics to get total number of calls received."""
010d03853546 Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 13620 diff changeset	526 nc = 0
23634 dbee2f628a90 Replaced .keys() with dictionary iterators Raymond Hettinger <python@rcn.com> parents: 23625 diff changeset	527 for calls in callers.itervalues():
dbee2f628a90 Replaced .keys() with dictionary iterators Raymond Hettinger <python@rcn.com> parents: 23625 diff changeset	528 nc += calls
15870 010d03853546 Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 13620 diff changeset	529 return nc
1679 5290428bad37 Companion module for new profile.py Guido van Rossum <guido@python.org> parents: diff changeset	530
5290428bad37 Companion module for new profile.py Guido van Rossum <guido@python.org> parents: diff changeset	531 #**************************************************************************
5290428bad37 Companion module for new profile.py Guido van Rossum <guido@python.org> parents: diff changeset	532 # The following functions support printing of reports
5290428bad37 Companion module for new profile.py Guido van Rossum <guido@python.org> parents: diff changeset	533 #**************************************************************************
5290428bad37 Companion module for new profile.py Guido van Rossum <guido@python.org> parents: diff changeset	534
5290428bad37 Companion module for new profile.py Guido van Rossum <guido@python.org> parents: diff changeset	535 def f8(x):
20308 ca2bc5cdaa8e Widespread random code cleanup. Tim Peters <tim.peters@gmail.com> parents: 19216 diff changeset	536 return "%8.3f" % x
17462 e8741dc36ad1 Added a test main to the pstats library that can help you browse profile dumps. Eric S. Raymond <esr@thyrsus.com> parents: 16562 diff changeset	537
e8741dc36ad1 Added a test main to the pstats library that can help you browse profile dumps. Eric S. Raymond <esr@thyrsus.com> parents: 16562 diff changeset	538 #**************************************************************************
e8741dc36ad1 Added a test main to the pstats library that can help you browse profile dumps. Eric S. Raymond <esr@thyrsus.com> parents: 16562 diff changeset	539 # Statistics browser added by ESR, April 2001
e8741dc36ad1 Added a test main to the pstats library that can help you browse profile dumps. Eric S. Raymond <esr@thyrsus.com> parents: 16562 diff changeset	540 #**************************************************************************
e8741dc36ad1 Added a test main to the pstats library that can help you browse profile dumps. Eric S. Raymond <esr@thyrsus.com> parents: 16562 diff changeset	541
e8741dc36ad1 Added a test main to the pstats library that can help you browse profile dumps. Eric S. Raymond <esr@thyrsus.com> parents: 16562 diff changeset	542 if __name__ == '__main__':
e8741dc36ad1 Added a test main to the pstats library that can help you browse profile dumps. Eric S. Raymond <esr@thyrsus.com> parents: 16562 diff changeset	543 import cmd
17510 fd0683cb7301 Import readline when possible to make the commaninterpreter UI nicer. Eric S. Raymond <esr@thyrsus.com> parents: 17467 diff changeset	544 try:
fd0683cb7301 Import readline when possible to make the commaninterpreter UI nicer. Eric S. Raymond <esr@thyrsus.com> parents: 17467 diff changeset	545 import readline
17800 c19980c03e2d When guarding an import, only catch ImportError. Fred Drake <fdrake@acm.org> parents: 17634 diff changeset	546 except ImportError:
17510 fd0683cb7301 Import readline when possible to make the commaninterpreter UI nicer. Eric S. Raymond <esr@thyrsus.com> parents: 17467 diff changeset	547 pass
17462 e8741dc36ad1 Added a test main to the pstats library that can help you browse profile dumps. Eric S. Raymond <esr@thyrsus.com> parents: 16562 diff changeset	548
e8741dc36ad1 Added a test main to the pstats library that can help you browse profile dumps. Eric S. Raymond <esr@thyrsus.com> parents: 16562 diff changeset	549 class ProfileBrowser(cmd.Cmd):
e8741dc36ad1 Added a test main to the pstats library that can help you browse profile dumps. Eric S. Raymond <esr@thyrsus.com> parents: 16562 diff changeset	550 def __init__(self, profile=None):
18855 6f301b3cf323 Patch #416224: add readline completion to cmd.Cmd. Martin v. Löwis <martin@v.loewis.de> parents: 18048 diff changeset	551 cmd.Cmd.__init__(self)
17462 e8741dc36ad1 Added a test main to the pstats library that can help you browse profile dumps. Eric S. Raymond <esr@thyrsus.com> parents: 16562 diff changeset	552 self.prompt = "% "
63602 e971c104a5c2 Merged revisions 83523 via svnmerge from Georg Brandl <georg@python.org> parents: 63549 diff changeset	553 self.stats = None
e971c104a5c2 Merged revisions 83523 via svnmerge from Georg Brandl <georg@python.org> parents: 63549 diff changeset	554 self.stream = sys.stdout
23622 5edd80726d96 Replace boolean test with is None. Raymond Hettinger <python@rcn.com> parents: 23620 diff changeset	555 if profile is not None:
63602 e971c104a5c2 Merged revisions 83523 via svnmerge from Georg Brandl <georg@python.org> parents: 63549 diff changeset	556 self.do_read(profile)
17462 e8741dc36ad1 Added a test main to the pstats library that can help you browse profile dumps. Eric S. Raymond <esr@thyrsus.com> parents: 16562 diff changeset	557
e8741dc36ad1 Added a test main to the pstats library that can help you browse profile dumps. Eric S. Raymond <esr@thyrsus.com> parents: 16562 diff changeset	558 def generic(self, fn, line):
e8741dc36ad1 Added a test main to the pstats library that can help you browse profile dumps. Eric S. Raymond <esr@thyrsus.com> parents: 16562 diff changeset	559 args = line.split()
e8741dc36ad1 Added a test main to the pstats library that can help you browse profile dumps. Eric S. Raymond <esr@thyrsus.com> parents: 16562 diff changeset	560 processed = []
e8741dc36ad1 Added a test main to the pstats library that can help you browse profile dumps. Eric S. Raymond <esr@thyrsus.com> parents: 16562 diff changeset	561 for term in args:
e8741dc36ad1 Added a test main to the pstats library that can help you browse profile dumps. Eric S. Raymond <esr@thyrsus.com> parents: 16562 diff changeset	562 try:
e8741dc36ad1 Added a test main to the pstats library that can help you browse profile dumps. Eric S. Raymond <esr@thyrsus.com> parents: 16562 diff changeset	563 processed.append(int(term))
e8741dc36ad1 Added a test main to the pstats library that can help you browse profile dumps. Eric S. Raymond <esr@thyrsus.com> parents: 16562 diff changeset	564 continue
e8741dc36ad1 Added a test main to the pstats library that can help you browse profile dumps. Eric S. Raymond <esr@thyrsus.com> parents: 16562 diff changeset	565 except ValueError:
e8741dc36ad1 Added a test main to the pstats library that can help you browse profile dumps. Eric S. Raymond <esr@thyrsus.com> parents: 16562 diff changeset	566 pass
e8741dc36ad1 Added a test main to the pstats library that can help you browse profile dumps. Eric S. Raymond <esr@thyrsus.com> parents: 16562 diff changeset	567 try:
e8741dc36ad1 Added a test main to the pstats library that can help you browse profile dumps. Eric S. Raymond <esr@thyrsus.com> parents: 16562 diff changeset	568 frac = float(term)
e8741dc36ad1 Added a test main to the pstats library that can help you browse profile dumps. Eric S. Raymond <esr@thyrsus.com> parents: 16562 diff changeset	569 if frac > 1 or frac < 0:
37873 d35db84ca072 Allow pstats.Stats creator to specify an alternate to stdout. Skip Montanaro <skip@pobox.com> parents: 36459 diff changeset	570 print >> self.stream, "Fraction argument must be in [0, 1]"
17462 e8741dc36ad1 Added a test main to the pstats library that can help you browse profile dumps. Eric S. Raymond <esr@thyrsus.com> parents: 16562 diff changeset	571 continue
e8741dc36ad1 Added a test main to the pstats library that can help you browse profile dumps. Eric S. Raymond <esr@thyrsus.com> parents: 16562 diff changeset	572 processed.append(frac)
e8741dc36ad1 Added a test main to the pstats library that can help you browse profile dumps. Eric S. Raymond <esr@thyrsus.com> parents: 16562 diff changeset	573 continue
e8741dc36ad1 Added a test main to the pstats library that can help you browse profile dumps. Eric S. Raymond <esr@thyrsus.com> parents: 16562 diff changeset	574 except ValueError:
e8741dc36ad1 Added a test main to the pstats library that can help you browse profile dumps. Eric S. Raymond <esr@thyrsus.com> parents: 16562 diff changeset	575 pass
e8741dc36ad1 Added a test main to the pstats library that can help you browse profile dumps. Eric S. Raymond <esr@thyrsus.com> parents: 16562 diff changeset	576 processed.append(term)
e8741dc36ad1 Added a test main to the pstats library that can help you browse profile dumps. Eric S. Raymond <esr@thyrsus.com> parents: 16562 diff changeset	577 if self.stats:
28089 14f65d661084 Get rid of many apply() calls. Guido van Rossum <guido@python.org> parents: 23634 diff changeset	578 getattr(self.stats, fn)(*processed)
17462 e8741dc36ad1 Added a test main to the pstats library that can help you browse profile dumps. Eric S. Raymond <esr@thyrsus.com> parents: 16562 diff changeset	579 else:
37873 d35db84ca072 Allow pstats.Stats creator to specify an alternate to stdout. Skip Montanaro <skip@pobox.com> parents: 36459 diff changeset	580 print >> self.stream, "No statistics object is loaded."
17462 e8741dc36ad1 Added a test main to the pstats library that can help you browse profile dumps. Eric S. Raymond <esr@thyrsus.com> parents: 16562 diff changeset	581 return 0
17634 0bc3377d0d72 Added more help, and recovery from misspelled sort key arguments. Eric S. Raymond <esr@thyrsus.com> parents: 17514 diff changeset	582 def generic_help(self):
37873 d35db84ca072 Allow pstats.Stats creator to specify an alternate to stdout. Skip Montanaro <skip@pobox.com> parents: 36459 diff changeset	583 print >> self.stream, "Arguments may be:"
d35db84ca072 Allow pstats.Stats creator to specify an alternate to stdout. Skip Montanaro <skip@pobox.com> parents: 36459 diff changeset	584 print >> self.stream, "* An integer maximum number of entries to print."
d35db84ca072 Allow pstats.Stats creator to specify an alternate to stdout. Skip Montanaro <skip@pobox.com> parents: 36459 diff changeset	585 print >> self.stream, "* A decimal fractional number between 0 and 1, controlling"
d35db84ca072 Allow pstats.Stats creator to specify an alternate to stdout. Skip Montanaro <skip@pobox.com> parents: 36459 diff changeset	586 print >> self.stream, " what fraction of selected entries to print."
d35db84ca072 Allow pstats.Stats creator to specify an alternate to stdout. Skip Montanaro <skip@pobox.com> parents: 36459 diff changeset	587 print >> self.stream, "* A regular expression; only entries with function names"
d35db84ca072 Allow pstats.Stats creator to specify an alternate to stdout. Skip Montanaro <skip@pobox.com> parents: 36459 diff changeset	588 print >> self.stream, " that match it are printed."
17462 e8741dc36ad1 Added a test main to the pstats library that can help you browse profile dumps. Eric S. Raymond <esr@thyrsus.com> parents: 16562 diff changeset	589
e8741dc36ad1 Added a test main to the pstats library that can help you browse profile dumps. Eric S. Raymond <esr@thyrsus.com> parents: 16562 diff changeset	590 def do_add(self, line):
63511 61ad81a774c5 Merged revisions 83421,83424 via svnmerge from Georg Brandl <georg@python.org> parents: 48146 diff changeset	591 if self.stats:
61ad81a774c5 Merged revisions 83421,83424 via svnmerge from Georg Brandl <georg@python.org> parents: 48146 diff changeset	592 self.stats.add(line)
61ad81a774c5 Merged revisions 83421,83424 via svnmerge from Georg Brandl <georg@python.org> parents: 48146 diff changeset	593 else:
61ad81a774c5 Merged revisions 83421,83424 via svnmerge from Georg Brandl <georg@python.org> parents: 48146 diff changeset	594 print >> self.stream, "No statistics object is loaded."
17462 e8741dc36ad1 Added a test main to the pstats library that can help you browse profile dumps. Eric S. Raymond <esr@thyrsus.com> parents: 16562 diff changeset	595 return 0
e8741dc36ad1 Added a test main to the pstats library that can help you browse profile dumps. Eric S. Raymond <esr@thyrsus.com> parents: 16562 diff changeset	596 def help_add(self):
37873 d35db84ca072 Allow pstats.Stats creator to specify an alternate to stdout. Skip Montanaro <skip@pobox.com> parents: 36459 diff changeset	597 print >> self.stream, "Add profile info from given file to current statistics object."
17462 e8741dc36ad1 Added a test main to the pstats library that can help you browse profile dumps. Eric S. Raymond <esr@thyrsus.com> parents: 16562 diff changeset	598
e8741dc36ad1 Added a test main to the pstats library that can help you browse profile dumps. Eric S. Raymond <esr@thyrsus.com> parents: 16562 diff changeset	599 def do_callees(self, line):
17514 773c64d38ac6 Should resolve [ #416039 ] pstats browser crashes. Eric S. Raymond <esr@thyrsus.com> parents: 17510 diff changeset	600 return self.generic('print_callees', line)
17462 e8741dc36ad1 Added a test main to the pstats library that can help you browse profile dumps. Eric S. Raymond <esr@thyrsus.com> parents: 16562 diff changeset	601 def help_callees(self):
37873 d35db84ca072 Allow pstats.Stats creator to specify an alternate to stdout. Skip Montanaro <skip@pobox.com> parents: 36459 diff changeset	602 print >> self.stream, "Print callees statistics from the current stat object."
17634 0bc3377d0d72 Added more help, and recovery from misspelled sort key arguments. Eric S. Raymond <esr@thyrsus.com> parents: 17514 diff changeset	603 self.generic_help()
17462 e8741dc36ad1 Added a test main to the pstats library that can help you browse profile dumps. Eric S. Raymond <esr@thyrsus.com> parents: 16562 diff changeset	604
e8741dc36ad1 Added a test main to the pstats library that can help you browse profile dumps. Eric S. Raymond <esr@thyrsus.com> parents: 16562 diff changeset	605 def do_callers(self, line):
17514 773c64d38ac6 Should resolve [ #416039 ] pstats browser crashes. Eric S. Raymond <esr@thyrsus.com> parents: 17510 diff changeset	606 return self.generic('print_callers', line)
17462 e8741dc36ad1 Added a test main to the pstats library that can help you browse profile dumps. Eric S. Raymond <esr@thyrsus.com> parents: 16562 diff changeset	607 def help_callers(self):
37873 d35db84ca072 Allow pstats.Stats creator to specify an alternate to stdout. Skip Montanaro <skip@pobox.com> parents: 36459 diff changeset	608 print >> self.stream, "Print callers statistics from the current stat object."
17634 0bc3377d0d72 Added more help, and recovery from misspelled sort key arguments. Eric S. Raymond <esr@thyrsus.com> parents: 17514 diff changeset	609 self.generic_help()
17462 e8741dc36ad1 Added a test main to the pstats library that can help you browse profile dumps. Eric S. Raymond <esr@thyrsus.com> parents: 16562 diff changeset	610
e8741dc36ad1 Added a test main to the pstats library that can help you browse profile dumps. Eric S. Raymond <esr@thyrsus.com> parents: 16562 diff changeset	611 def do_EOF(self, line):
37873 d35db84ca072 Allow pstats.Stats creator to specify an alternate to stdout. Skip Montanaro <skip@pobox.com> parents: 36459 diff changeset	612 print >> self.stream, ""
17462 e8741dc36ad1 Added a test main to the pstats library that can help you browse profile dumps. Eric S. Raymond <esr@thyrsus.com> parents: 16562 diff changeset	613 return 1
e8741dc36ad1 Added a test main to the pstats library that can help you browse profile dumps. Eric S. Raymond <esr@thyrsus.com> parents: 16562 diff changeset	614 def help_EOF(self):
37873 d35db84ca072 Allow pstats.Stats creator to specify an alternate to stdout. Skip Montanaro <skip@pobox.com> parents: 36459 diff changeset	615 print >> self.stream, "Leave the profile brower."
17462 e8741dc36ad1 Added a test main to the pstats library that can help you browse profile dumps. Eric S. Raymond <esr@thyrsus.com> parents: 16562 diff changeset	616
e8741dc36ad1 Added a test main to the pstats library that can help you browse profile dumps. Eric S. Raymond <esr@thyrsus.com> parents: 16562 diff changeset	617 def do_quit(self, line):
e8741dc36ad1 Added a test main to the pstats library that can help you browse profile dumps. Eric S. Raymond <esr@thyrsus.com> parents: 16562 diff changeset	618 return 1
e8741dc36ad1 Added a test main to the pstats library that can help you browse profile dumps. Eric S. Raymond <esr@thyrsus.com> parents: 16562 diff changeset	619 def help_quit(self):
37873 d35db84ca072 Allow pstats.Stats creator to specify an alternate to stdout. Skip Montanaro <skip@pobox.com> parents: 36459 diff changeset	620 print >> self.stream, "Leave the profile brower."
17462 e8741dc36ad1 Added a test main to the pstats library that can help you browse profile dumps. Eric S. Raymond <esr@thyrsus.com> parents: 16562 diff changeset	621
e8741dc36ad1 Added a test main to the pstats library that can help you browse profile dumps. Eric S. Raymond <esr@thyrsus.com> parents: 16562 diff changeset	622 def do_read(self, line):
e8741dc36ad1 Added a test main to the pstats library that can help you browse profile dumps. Eric S. Raymond <esr@thyrsus.com> parents: 16562 diff changeset	623 if line:
e8741dc36ad1 Added a test main to the pstats library that can help you browse profile dumps. Eric S. Raymond <esr@thyrsus.com> parents: 16562 diff changeset	624 try:
e8741dc36ad1 Added a test main to the pstats library that can help you browse profile dumps. Eric S. Raymond <esr@thyrsus.com> parents: 16562 diff changeset	625 self.stats = Stats(line)
e8741dc36ad1 Added a test main to the pstats library that can help you browse profile dumps. Eric S. Raymond <esr@thyrsus.com> parents: 16562 diff changeset	626 except IOError, args:
37873 d35db84ca072 Allow pstats.Stats creator to specify an alternate to stdout. Skip Montanaro <skip@pobox.com> parents: 36459 diff changeset	627 print >> self.stream, args[1]
17462 e8741dc36ad1 Added a test main to the pstats library that can help you browse profile dumps. Eric S. Raymond <esr@thyrsus.com> parents: 16562 diff changeset	628 return
63511 61ad81a774c5 Merged revisions 83421,83424 via svnmerge from Georg Brandl <georg@python.org> parents: 48146 diff changeset	629 except Exception as err:
61ad81a774c5 Merged revisions 83421,83424 via svnmerge from Georg Brandl <georg@python.org> parents: 48146 diff changeset	630 print >> self.stream, err.__class__.__name__ + ':', err
61ad81a774c5 Merged revisions 83421,83424 via svnmerge from Georg Brandl <georg@python.org> parents: 48146 diff changeset	631 return
17462 e8741dc36ad1 Added a test main to the pstats library that can help you browse profile dumps. Eric S. Raymond <esr@thyrsus.com> parents: 16562 diff changeset	632 self.prompt = line + "% "
18048 14dc75a5600a Patch #416220: Fix misplaced paren. Martin v. Löwis <martin@v.loewis.de> parents: 17800 diff changeset	633 elif len(self.prompt) > 2:
63511 61ad81a774c5 Merged revisions 83421,83424 via svnmerge from Georg Brandl <georg@python.org> parents: 48146 diff changeset	634 line = self.prompt[:-2]
61ad81a774c5 Merged revisions 83421,83424 via svnmerge from Georg Brandl <georg@python.org> parents: 48146 diff changeset	635 self.do_read(line)
17462 e8741dc36ad1 Added a test main to the pstats library that can help you browse profile dumps. Eric S. Raymond <esr@thyrsus.com> parents: 16562 diff changeset	636 else:
37873 d35db84ca072 Allow pstats.Stats creator to specify an alternate to stdout. Skip Montanaro <skip@pobox.com> parents: 36459 diff changeset	637 print >> self.stream, "No statistics object is current -- cannot reload."
17462 e8741dc36ad1 Added a test main to the pstats library that can help you browse profile dumps. Eric S. Raymond <esr@thyrsus.com> parents: 16562 diff changeset	638 return 0
e8741dc36ad1 Added a test main to the pstats library that can help you browse profile dumps. Eric S. Raymond <esr@thyrsus.com> parents: 16562 diff changeset	639 def help_read(self):
37873 d35db84ca072 Allow pstats.Stats creator to specify an alternate to stdout. Skip Montanaro <skip@pobox.com> parents: 36459 diff changeset	640 print >> self.stream, "Read in profile data from a specified file."
63511 61ad81a774c5 Merged revisions 83421,83424 via svnmerge from Georg Brandl <georg@python.org> parents: 48146 diff changeset	641 print >> self.stream, "Without argument, reload the current file."
17462 e8741dc36ad1 Added a test main to the pstats library that can help you browse profile dumps. Eric S. Raymond <esr@thyrsus.com> parents: 16562 diff changeset	642
e8741dc36ad1 Added a test main to the pstats library that can help you browse profile dumps. Eric S. Raymond <esr@thyrsus.com> parents: 16562 diff changeset	643 def do_reverse(self, line):
63511 61ad81a774c5 Merged revisions 83421,83424 via svnmerge from Georg Brandl <georg@python.org> parents: 48146 diff changeset	644 if self.stats:
61ad81a774c5 Merged revisions 83421,83424 via svnmerge from Georg Brandl <georg@python.org> parents: 48146 diff changeset	645 self.stats.reverse_order()
61ad81a774c5 Merged revisions 83421,83424 via svnmerge from Georg Brandl <georg@python.org> parents: 48146 diff changeset	646 else:
61ad81a774c5 Merged revisions 83421,83424 via svnmerge from Georg Brandl <georg@python.org> parents: 48146 diff changeset	647 print >> self.stream, "No statistics object is loaded."
17462 e8741dc36ad1 Added a test main to the pstats library that can help you browse profile dumps. Eric S. Raymond <esr@thyrsus.com> parents: 16562 diff changeset	648 return 0
e8741dc36ad1 Added a test main to the pstats library that can help you browse profile dumps. Eric S. Raymond <esr@thyrsus.com> parents: 16562 diff changeset	649 def help_reverse(self):
37873 d35db84ca072 Allow pstats.Stats creator to specify an alternate to stdout. Skip Montanaro <skip@pobox.com> parents: 36459 diff changeset	650 print >> self.stream, "Reverse the sort order of the profiling report."
17462 e8741dc36ad1 Added a test main to the pstats library that can help you browse profile dumps. Eric S. Raymond <esr@thyrsus.com> parents: 16562 diff changeset	651
e8741dc36ad1 Added a test main to the pstats library that can help you browse profile dumps. Eric S. Raymond <esr@thyrsus.com> parents: 16562 diff changeset	652 def do_sort(self, line):
63511 61ad81a774c5 Merged revisions 83421,83424 via svnmerge from Georg Brandl <georg@python.org> parents: 48146 diff changeset	653 if not self.stats:
61ad81a774c5 Merged revisions 83421,83424 via svnmerge from Georg Brandl <georg@python.org> parents: 48146 diff changeset	654 print >> self.stream, "No statistics object is loaded."
61ad81a774c5 Merged revisions 83421,83424 via svnmerge from Georg Brandl <georg@python.org> parents: 48146 diff changeset	655 return
23634 dbee2f628a90 Replaced .keys() with dictionary iterators Raymond Hettinger <python@rcn.com> parents: 23625 diff changeset	656 abbrevs = self.stats.get_sort_arg_defs()
60167 4ac05d1f60c1 #4440: modernize a use of filter(), making it compatible with 3.x Andrew M. Kuchling <amk@amk.ca> parents: 58159 diff changeset	657 if line and all((x in abbrevs) for x in line.split()):
28089 14f65d661084 Get rid of many apply() calls. Guido van Rossum <guido@python.org> parents: 23634 diff changeset	658 self.stats.sort_stats(*line.split())
17634 0bc3377d0d72 Added more help, and recovery from misspelled sort key arguments. Eric S. Raymond <esr@thyrsus.com> parents: 17514 diff changeset	659 else:
37873 d35db84ca072 Allow pstats.Stats creator to specify an alternate to stdout. Skip Montanaro <skip@pobox.com> parents: 36459 diff changeset	660 print >> self.stream, "Valid sort keys (unique prefixes are accepted):"
23634 dbee2f628a90 Replaced .keys() with dictionary iterators Raymond Hettinger <python@rcn.com> parents: 23625 diff changeset	661 for (key, value) in Stats.sort_arg_dict_default.iteritems():
37873 d35db84ca072 Allow pstats.Stats creator to specify an alternate to stdout. Skip Montanaro <skip@pobox.com> parents: 36459 diff changeset	662 print >> self.stream, "%s -- %s" % (key, value[1])
17462 e8741dc36ad1 Added a test main to the pstats library that can help you browse profile dumps. Eric S. Raymond <esr@thyrsus.com> parents: 16562 diff changeset	663 return 0
e8741dc36ad1 Added a test main to the pstats library that can help you browse profile dumps. Eric S. Raymond <esr@thyrsus.com> parents: 16562 diff changeset	664 def help_sort(self):
37873 d35db84ca072 Allow pstats.Stats creator to specify an alternate to stdout. Skip Montanaro <skip@pobox.com> parents: 36459 diff changeset	665 print >> self.stream, "Sort profile data according to specified keys."
d35db84ca072 Allow pstats.Stats creator to specify an alternate to stdout. Skip Montanaro <skip@pobox.com> parents: 36459 diff changeset	666 print >> self.stream, "(Typing `sort' without arguments lists valid keys.)"
18877 0e212e88d33d Patch #445538: add completion for pstats.py sort cmd. Martin v. Löwis <martin@v.loewis.de> parents: 18855 diff changeset	667 def complete_sort(self, text, *args):
23634 dbee2f628a90 Replaced .keys() with dictionary iterators Raymond Hettinger <python@rcn.com> parents: 23625 diff changeset	668 return [a for a in Stats.sort_arg_dict_default if a.startswith(text)]
17462 e8741dc36ad1 Added a test main to the pstats library that can help you browse profile dumps. Eric S. Raymond <esr@thyrsus.com> parents: 16562 diff changeset	669
e8741dc36ad1 Added a test main to the pstats library that can help you browse profile dumps. Eric S. Raymond <esr@thyrsus.com> parents: 16562 diff changeset	670 def do_stats(self, line):
e8741dc36ad1 Added a test main to the pstats library that can help you browse profile dumps. Eric S. Raymond <esr@thyrsus.com> parents: 16562 diff changeset	671 return self.generic('print_stats', line)
e8741dc36ad1 Added a test main to the pstats library that can help you browse profile dumps. Eric S. Raymond <esr@thyrsus.com> parents: 16562 diff changeset	672 def help_stats(self):
37873 d35db84ca072 Allow pstats.Stats creator to specify an alternate to stdout. Skip Montanaro <skip@pobox.com> parents: 36459 diff changeset	673 print >> self.stream, "Print statistics from the current stat object."
17634 0bc3377d0d72 Added more help, and recovery from misspelled sort key arguments. Eric S. Raymond <esr@thyrsus.com> parents: 17514 diff changeset	674 self.generic_help()
17462 e8741dc36ad1 Added a test main to the pstats library that can help you browse profile dumps. Eric S. Raymond <esr@thyrsus.com> parents: 16562 diff changeset	675
e8741dc36ad1 Added a test main to the pstats library that can help you browse profile dumps. Eric S. Raymond <esr@thyrsus.com> parents: 16562 diff changeset	676 def do_strip(self, line):
63511 61ad81a774c5 Merged revisions 83421,83424 via svnmerge from Georg Brandl <georg@python.org> parents: 48146 diff changeset	677 if self.stats:
61ad81a774c5 Merged revisions 83421,83424 via svnmerge from Georg Brandl <georg@python.org> parents: 48146 diff changeset	678 self.stats.strip_dirs()
61ad81a774c5 Merged revisions 83421,83424 via svnmerge from Georg Brandl <georg@python.org> parents: 48146 diff changeset	679 else:
61ad81a774c5 Merged revisions 83421,83424 via svnmerge from Georg Brandl <georg@python.org> parents: 48146 diff changeset	680 print >> self.stream, "No statistics object is loaded."
17462 e8741dc36ad1 Added a test main to the pstats library that can help you browse profile dumps. Eric S. Raymond <esr@thyrsus.com> parents: 16562 diff changeset	681 def help_strip(self):
37873 d35db84ca072 Allow pstats.Stats creator to specify an alternate to stdout. Skip Montanaro <skip@pobox.com> parents: 36459 diff changeset	682 print >> self.stream, "Strip leading path information from filenames in the report."
17462 e8741dc36ad1 Added a test main to the pstats library that can help you browse profile dumps. Eric S. Raymond <esr@thyrsus.com> parents: 16562 diff changeset	683
63511 61ad81a774c5 Merged revisions 83421,83424 via svnmerge from Georg Brandl <georg@python.org> parents: 48146 diff changeset	684 def help_help(self):
61ad81a774c5 Merged revisions 83421,83424 via svnmerge from Georg Brandl <georg@python.org> parents: 48146 diff changeset	685 print >> self.stream, "Show help for a given command."
61ad81a774c5 Merged revisions 83421,83424 via svnmerge from Georg Brandl <georg@python.org> parents: 48146 diff changeset	686
17462 e8741dc36ad1 Added a test main to the pstats library that can help you browse profile dumps. Eric S. Raymond <esr@thyrsus.com> parents: 16562 diff changeset	687 def postcmd(self, stop, line):
e8741dc36ad1 Added a test main to the pstats library that can help you browse profile dumps. Eric S. Raymond <esr@thyrsus.com> parents: 16562 diff changeset	688 if stop:
e8741dc36ad1 Added a test main to the pstats library that can help you browse profile dumps. Eric S. Raymond <esr@thyrsus.com> parents: 16562 diff changeset	689 return stop
e8741dc36ad1 Added a test main to the pstats library that can help you browse profile dumps. Eric S. Raymond <esr@thyrsus.com> parents: 16562 diff changeset	690 return None
e8741dc36ad1 Added a test main to the pstats library that can help you browse profile dumps. Eric S. Raymond <esr@thyrsus.com> parents: 16562 diff changeset	691
e8741dc36ad1 Added a test main to the pstats library that can help you browse profile dumps. Eric S. Raymond <esr@thyrsus.com> parents: 16562 diff changeset	692 import sys
e8741dc36ad1 Added a test main to the pstats library that can help you browse profile dumps. Eric S. Raymond <esr@thyrsus.com> parents: 16562 diff changeset	693 if len(sys.argv) > 1:
e8741dc36ad1 Added a test main to the pstats library that can help you browse profile dumps. Eric S. Raymond <esr@thyrsus.com> parents: 16562 diff changeset	694 initprofile = sys.argv[1]
e8741dc36ad1 Added a test main to the pstats library that can help you browse profile dumps. Eric S. Raymond <esr@thyrsus.com> parents: 16562 diff changeset	695 else:
e8741dc36ad1 Added a test main to the pstats library that can help you browse profile dumps. Eric S. Raymond <esr@thyrsus.com> parents: 16562 diff changeset	696 initprofile = None
e8741dc36ad1 Added a test main to the pstats library that can help you browse profile dumps. Eric S. Raymond <esr@thyrsus.com> parents: 16562 diff changeset	697 try:
38742 49026af9d1d7 Fix errors found by pychecker. Neal Norwitz <nnorwitz@gmail.com> parents: 37873 diff changeset	698 browser = ProfileBrowser(initprofile)
49026af9d1d7 Fix errors found by pychecker. Neal Norwitz <nnorwitz@gmail.com> parents: 37873 diff changeset	699 print >> browser.stream, "Welcome to the profile statistics browser."
49026af9d1d7 Fix errors found by pychecker. Neal Norwitz <nnorwitz@gmail.com> parents: 37873 diff changeset	700 browser.cmdloop()
49026af9d1d7 Fix errors found by pychecker. Neal Norwitz <nnorwitz@gmail.com> parents: 37873 diff changeset	701 print >> browser.stream, "Goodbye."
17462 e8741dc36ad1 Added a test main to the pstats library that can help you browse profile dumps. Eric S. Raymond <esr@thyrsus.com> parents: 16562 diff changeset	702 except KeyboardInterrupt:
e8741dc36ad1 Added a test main to the pstats library that can help you browse profile dumps. Eric S. Raymond <esr@thyrsus.com> parents: 16562 diff changeset	703 pass
e8741dc36ad1 Added a test main to the pstats library that can help you browse profile dumps. Eric S. Raymond <esr@thyrsus.com> parents: 16562 diff changeset	704
e8741dc36ad1 Added a test main to the pstats library that can help you browse profile dumps. Eric S. Raymond <esr@thyrsus.com> parents: 16562 diff changeset	705 # That's all, folks.

Mercurial > cpython-lru_cache-weakref

annotate Lib/pstats.py @ 80661:94de77bd0b4b 2.7