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

annotate Lib/trace.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	bb7b14dd5ded
children

rev	line source
27933 44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	1 #!/usr/bin/env python
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	2
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	3 # portions copyright 2001, Autonomous Zones Industries, Inc., all rights...
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	4 # err... reserved and offered to the public under the terms of the
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	5 # Python 2.2 license.
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	6 # Author: Zooko O'Whielacronx
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	7 # http://zooko.com/
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	8 # mailto:zooko@zooko.com
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	9 #
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	10 # Copyright 2000, Mojam Media, Inc., all rights reserved.
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	11 # Author: Skip Montanaro
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	12 #
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	13 # Copyright 1999, Bioreason, Inc., all rights reserved.
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	14 # Author: Andrew Dalke
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	15 #
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	16 # Copyright 1995-1997, Automatrix, Inc., all rights reserved.
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	17 # Author: Skip Montanaro
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	18 #
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	19 # Copyright 1991-1995, Stichting Mathematisch Centrum, all rights reserved.
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	20 #
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	21 #
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	22 # Permission to use, copy, modify, and distribute this Python software and
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	23 # its associated documentation for any purpose without fee is hereby
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	24 # granted, provided that the above copyright notice appears in all copies,
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	25 # and that both that copyright notice and this permission notice appear in
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	26 # supporting documentation, and that the name of neither Automatrix,
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	27 # Bioreason or Mojam Media be used in advertising or publicity pertaining to
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	28 # distribution of the software without specific, written prior permission.
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	29 #
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	30 """program/module to trace Python program or function execution
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	31
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	32 Sample use, command line:
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	33 trace.py -c -f counts --ignore-dir '$prefix' spam.py eggs
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	34 trace.py -t --ignore-dir '$prefix' spam.py eggs
31892 081972ae2f84 Refactor common code out of globaltrace_trackcallers() and Skip Montanaro <skip@pobox.com> parents: 31881 diff changeset	35 trace.py --trackcalls spam.py eggs
27933 44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	36
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	37 Sample use, programmatically
37947 1b192037b62f correct example Skip Montanaro <skip@pobox.com> parents: 35806 diff changeset	38 import sys
1b192037b62f correct example Skip Montanaro <skip@pobox.com> parents: 35806 diff changeset	39
1b192037b62f correct example Skip Montanaro <skip@pobox.com> parents: 35806 diff changeset	40 # create a Trace object, telling it what to ignore, and whether to
1b192037b62f correct example Skip Montanaro <skip@pobox.com> parents: 35806 diff changeset	41 # do tracing or line-counting or both.
1b192037b62f correct example Skip Montanaro <skip@pobox.com> parents: 35806 diff changeset	42 tracer = trace.Trace(ignoredirs=[sys.prefix, sys.exec_prefix,], trace=0,
37964 acd5e5559e3c Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 37947 diff changeset	43 count=1)
37947 1b192037b62f correct example Skip Montanaro <skip@pobox.com> parents: 35806 diff changeset	44 # run the new command using the given tracer
1b192037b62f correct example Skip Montanaro <skip@pobox.com> parents: 35806 diff changeset	45 tracer.run('main()')
1b192037b62f correct example Skip Montanaro <skip@pobox.com> parents: 35806 diff changeset	46 # make a report, placing output in /tmp
1b192037b62f correct example Skip Montanaro <skip@pobox.com> parents: 35806 diff changeset	47 r = tracer.results()
1b192037b62f correct example Skip Montanaro <skip@pobox.com> parents: 35806 diff changeset	48 r.write_results(show_missing=True, coverdir="/tmp")
27933 44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	49 """
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	50
28712 d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	51 import linecache
d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	52 import os
d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	53 import re
d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	54 import sys
45535 c57cb2f8c350 Add a timing flag to Trace so you can see where slowness occurs Neal Norwitz <nnorwitz@gmail.com> parents: 44775 diff changeset	55 import time
28712 d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	56 import token
d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	57 import tokenize
64711 0f3b8f928ec1 Issue #9315: Fix for the trace module to record correct class name Alexander Belopolsky <alexander.belopolsky@gmail.com> parents: 63605 diff changeset	58 import inspect
31892 081972ae2f84 Refactor common code out of globaltrace_trackcallers() and Skip Montanaro <skip@pobox.com> parents: 31881 diff changeset	59 import gc
64926 12e0dffef7bd Merged revisions 84994 via svnmerge from Alexander Belopolsky <alexander.belopolsky@gmail.com> parents: 64711 diff changeset	60 import dis
27933 44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	61 try:
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	62 import cPickle
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	63 pickle = cPickle
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	64 except ImportError:
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	65 import pickle
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	66
66003 acb0a3cc1e18 Merged revisions 86229 via svnmerge from Alexander Belopolsky <alexander.belopolsky@gmail.com> parents: 64926 diff changeset	67 try:
acb0a3cc1e18 Merged revisions 86229 via svnmerge from Alexander Belopolsky <alexander.belopolsky@gmail.com> parents: 64926 diff changeset	68 import threading
acb0a3cc1e18 Merged revisions 86229 via svnmerge from Alexander Belopolsky <alexander.belopolsky@gmail.com> parents: 64926 diff changeset	69 except ImportError:
acb0a3cc1e18 Merged revisions 86229 via svnmerge from Alexander Belopolsky <alexander.belopolsky@gmail.com> parents: 64926 diff changeset	70 _settrace = sys.settrace
acb0a3cc1e18 Merged revisions 86229 via svnmerge from Alexander Belopolsky <alexander.belopolsky@gmail.com> parents: 64926 diff changeset	71
acb0a3cc1e18 Merged revisions 86229 via svnmerge from Alexander Belopolsky <alexander.belopolsky@gmail.com> parents: 64926 diff changeset	72 def _unsettrace():
acb0a3cc1e18 Merged revisions 86229 via svnmerge from Alexander Belopolsky <alexander.belopolsky@gmail.com> parents: 64926 diff changeset	73 sys.settrace(None)
acb0a3cc1e18 Merged revisions 86229 via svnmerge from Alexander Belopolsky <alexander.belopolsky@gmail.com> parents: 64926 diff changeset	74 else:
acb0a3cc1e18 Merged revisions 86229 via svnmerge from Alexander Belopolsky <alexander.belopolsky@gmail.com> parents: 64926 diff changeset	75 def _settrace(func):
acb0a3cc1e18 Merged revisions 86229 via svnmerge from Alexander Belopolsky <alexander.belopolsky@gmail.com> parents: 64926 diff changeset	76 threading.settrace(func)
acb0a3cc1e18 Merged revisions 86229 via svnmerge from Alexander Belopolsky <alexander.belopolsky@gmail.com> parents: 64926 diff changeset	77 sys.settrace(func)
acb0a3cc1e18 Merged revisions 86229 via svnmerge from Alexander Belopolsky <alexander.belopolsky@gmail.com> parents: 64926 diff changeset	78
acb0a3cc1e18 Merged revisions 86229 via svnmerge from Alexander Belopolsky <alexander.belopolsky@gmail.com> parents: 64926 diff changeset	79 def _unsettrace():
acb0a3cc1e18 Merged revisions 86229 via svnmerge from Alexander Belopolsky <alexander.belopolsky@gmail.com> parents: 64926 diff changeset	80 sys.settrace(None)
acb0a3cc1e18 Merged revisions 86229 via svnmerge from Alexander Belopolsky <alexander.belopolsky@gmail.com> parents: 64926 diff changeset	81 threading.settrace(None)
acb0a3cc1e18 Merged revisions 86229 via svnmerge from Alexander Belopolsky <alexander.belopolsky@gmail.com> parents: 64926 diff changeset	82
27933 44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	83 def usage(outfile):
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	84 outfile.write("""Usage: %s [OPTIONS] <file> [ARGS]
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	85
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	86 Meta-options:
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	87 --help Display this help then exit.
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	88 --version Output version information then exit.
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	89
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	90 Otherwise, exactly one of the following three options must be given:
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	91 -t, --trace Print each line to sys.stdout before it is executed.
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	92 -c, --count Count the number of times each line is executed
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	93 and write the counts to <module>.cover for each
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	94 module executed, in the module's directory.
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	95 See also `--coverdir', `--file', `--no-report' below.
29619 f8069f7edd4a add mention of -l, --listfuncs to usage() small part of 542562 Skip Montanaro <skip@pobox.com> parents: 29595 diff changeset	96 -l, --listfuncs Keep track of which functions are executed at least
29620 77eda0fb0e14 fix typo Fred Drake <fdrake@acm.org> parents: 29619 diff changeset	97 once and write the results to sys.stdout after the
29619 f8069f7edd4a add mention of -l, --listfuncs to usage() small part of 542562 Skip Montanaro <skip@pobox.com> parents: 29595 diff changeset	98 program exits.
31881 f35b4acc594f Added --trackcalls command line arg to display crude caller/callee Skip Montanaro <skip@pobox.com> parents: 31551 diff changeset	99 -T, --trackcalls Keep track of caller/called pairs and write the
f35b4acc594f Added --trackcalls command line arg to display crude caller/callee Skip Montanaro <skip@pobox.com> parents: 31551 diff changeset	100 results to sys.stdout after the program exits.
27933 44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	101 -r, --report Generate a report from a counts file; do not execute
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	102 any code. `--file' must specify the results file to
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	103 read, which must have been created in a previous run
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	104 with `--count --file=FILE'.
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	105
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	106 Modifiers:
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	107 -f, --file=<file> File to accumulate counts over several runs.
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	108 -R, --no-report Do not generate the coverage report files.
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	109 Useful if you want to accumulate over several runs.
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	110 -C, --coverdir=<dir> Directory where the report files. The coverage
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	111 report for <package>.<module> is written to file
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	112 <dir>/<package>/<module>.cover.
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	113 -m, --missing Annotate executable lines that were not executed
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	114 with '>>>>>> '.
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	115 -s, --summary Write a brief summary on stdout for each file.
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	116 (Can only be used with --count or --report.)
45535 c57cb2f8c350 Add a timing flag to Trace so you can see where slowness occurs Neal Norwitz <nnorwitz@gmail.com> parents: 44775 diff changeset	117 -g, --timing Prefix each line with the time since the program started.
c57cb2f8c350 Add a timing flag to Trace so you can see where slowness occurs Neal Norwitz <nnorwitz@gmail.com> parents: 44775 diff changeset	118 Only used while tracing.
27933 44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	119
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	120 Filters, may be repeated multiple times:
44775 9c0a913adcd4 Fix #1693149. Now you can pass several modules separated by Facundo Batista <facundobatista@gmail.com> parents: 44014 diff changeset	121 --ignore-module=<mod> Ignore the given module(s) and its submodules
9c0a913adcd4 Fix #1693149. Now you can pass several modules separated by Facundo Batista <facundobatista@gmail.com> parents: 44014 diff changeset	122 (if it is a package). Accepts comma separated
9c0a913adcd4 Fix #1693149. Now you can pass several modules separated by Facundo Batista <facundobatista@gmail.com> parents: 44014 diff changeset	123 list of module names
27933 44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	124 --ignore-dir=<dir> Ignore files in the given directory (multiple
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	125 directories can be joined by os.pathsep).
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	126 """ % sys.argv[0])
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	127
28712 d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	128 PRAGMA_NOCOVER = "#pragma NO COVER"
d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	129
d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	130 # Simple rx to find lines with no code.
d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	131 rx_blank = re.compile(r'^\s(#.)?$')
d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	132
27933 44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	133 class Ignore:
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	134 def __init__(self, modules = None, dirs = None):
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	135 self._mods = modules or []
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	136 self._dirs = dirs or []
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	137
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	138 self._dirs = map(os.path.normpath, self._dirs)
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	139 self._ignore = { '<string>': 1 }
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	140
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	141 def names(self, filename, modulename):
56436 684960fb1cbf replace has_key with 'in' operator Benjamin Peterson <benjamin@python.org> parents: 45535 diff changeset	142 if modulename in self._ignore:
27933 44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	143 return self._ignore[modulename]
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	144
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	145 # haven't seen this one before, so see if the module name is
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	146 # on the ignore list. Need to take some care since ignoring
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	147 # "cmp" musn't mean ignoring "cmpcache" but ignoring
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	148 # "Spam" must also mean ignoring "Spam.Eggs".
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	149 for mod in self._mods:
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	150 if mod == modulename: # Identical names, so ignore
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	151 self._ignore[modulename] = 1
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	152 return 1
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	153 # check if the module is a proper submodule of something on
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	154 # the ignore list
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	155 n = len(mod)
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	156 # (will not overflow since if the first n characters are the
35806 44fe157255bc fix typos, mostly in comments Fred Drake <fdrake@acm.org> parents: 33941 diff changeset	157 # same and the name has not already occurred, then the size
27933 44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	158 # of "name" is greater than that of "mod")
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	159 if mod == modulename[:n] and modulename[n] == '.':
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	160 self._ignore[modulename] = 1
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	161 return 1
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	162
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	163 # Now check that __file__ isn't in one of the directories
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	164 if filename is None:
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	165 # must be a built-in, so we must ignore
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	166 self._ignore[modulename] = 1
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	167 return 1
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	168
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	169 # Ignore a file when it contains one of the ignorable paths
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	170 for d in self._dirs:
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	171 # The '+ os.sep' is to ensure that d is a parent directory,
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	172 # as compared to cases like:
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	173 # d = "/usr/local"
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	174 # filename = "/usr/local.py"
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	175 # or
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	176 # d = "/usr/local.py"
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	177 # filename = "/usr/local.py"
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	178 if filename.startswith(d + os.sep):
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	179 self._ignore[modulename] = 1
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	180 return 1
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	181
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	182 # Tried the different ways, so we don't ignore this module
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	183 self._ignore[modulename] = 0
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	184 return 0
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	185
28712 d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	186 def modname(path):
d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	187 """Return a plausible module name for the patch."""
28713 d08d15a70295 Add helper function to get module name taking packages into account. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28712 diff changeset	188
28712 d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	189 base = os.path.basename(path)
d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	190 filename, ext = os.path.splitext(base)
d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	191 return filename
d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	192
28713 d08d15a70295 Add helper function to get module name taking packages into account. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28712 diff changeset	193 def fullmodname(path):
28727 051a4f12cc46 Only produce a dotted module name when writing output to a separate Jeremy Hylton <jeremy@alum.mit.edu> parents: 28713 diff changeset	194 """Return a plausible module name for the path."""
28713 d08d15a70295 Add helper function to get module name taking packages into account. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28712 diff changeset	195
d08d15a70295 Add helper function to get module name taking packages into account. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28712 diff changeset	196 # If the file 'path' is part of a package, then the filename isn't
d08d15a70295 Add helper function to get module name taking packages into account. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28712 diff changeset	197 # enough to uniquely identify it. Try to do the right thing by
d08d15a70295 Add helper function to get module name taking packages into account. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28712 diff changeset	198 # looking in sys.path for the longest matching prefix. We'll
d08d15a70295 Add helper function to get module name taking packages into account. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28712 diff changeset	199 # assume that the rest is the package name.
d08d15a70295 Add helper function to get module name taking packages into account. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28712 diff changeset	200
39500 207e487f425d Patch #1536071: trace.py should now find the full module name of a Georg Brandl <georg@python.org> parents: 38704 diff changeset	201 comparepath = os.path.normcase(path)
28713 d08d15a70295 Add helper function to get module name taking packages into account. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28712 diff changeset	202 longest = ""
d08d15a70295 Add helper function to get module name taking packages into account. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28712 diff changeset	203 for dir in sys.path:
39500 207e487f425d Patch #1536071: trace.py should now find the full module name of a Georg Brandl <georg@python.org> parents: 38704 diff changeset	204 dir = os.path.normcase(dir)
207e487f425d Patch #1536071: trace.py should now find the full module name of a Georg Brandl <georg@python.org> parents: 38704 diff changeset	205 if comparepath.startswith(dir) and comparepath[len(dir)] == os.sep:
28713 d08d15a70295 Add helper function to get module name taking packages into account. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28712 diff changeset	206 if len(dir) > len(longest):
d08d15a70295 Add helper function to get module name taking packages into account. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28712 diff changeset	207 longest = dir
d08d15a70295 Add helper function to get module name taking packages into account. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28712 diff changeset	208
30667 400674958025 The fullmodname() function chopped off the first character if the Guido van Rossum <guido@python.org> parents: 29864 diff changeset	209 if longest:
400674958025 The fullmodname() function chopped off the first character if the Guido van Rossum <guido@python.org> parents: 29864 diff changeset	210 base = path[len(longest) + 1:]
400674958025 The fullmodname() function chopped off the first character if the Guido van Rossum <guido@python.org> parents: 29864 diff changeset	211 else:
400674958025 The fullmodname() function chopped off the first character if the Guido van Rossum <guido@python.org> parents: 29864 diff changeset	212 base = path
63511 61ad81a774c5 Merged revisions 83421,83424 via svnmerge from Georg Brandl <georg@python.org> parents: 45535 diff changeset	213 # the drive letter is never part of the module name
61ad81a774c5 Merged revisions 83421,83424 via svnmerge from Georg Brandl <georg@python.org> parents: 45535 diff changeset	214 drive, base = os.path.splitdrive(base)
31551 a157cc20f1b7 Fix two small bugs: (1) on Windows, pathname munging didn't work Guido van Rossum <guido@python.org> parents: 31515 diff changeset	215 base = base.replace(os.sep, ".")
a157cc20f1b7 Fix two small bugs: (1) on Windows, pathname munging didn't work Guido van Rossum <guido@python.org> parents: 31515 diff changeset	216 if os.altsep:
a157cc20f1b7 Fix two small bugs: (1) on Windows, pathname munging didn't work Guido van Rossum <guido@python.org> parents: 31515 diff changeset	217 base = base.replace(os.altsep, ".")
28713 d08d15a70295 Add helper function to get module name taking packages into account. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28712 diff changeset	218 filename, ext = os.path.splitext(base)
63511 61ad81a774c5 Merged revisions 83421,83424 via svnmerge from Georg Brandl <georg@python.org> parents: 45535 diff changeset	219 return filename.lstrip(".")
28713 d08d15a70295 Add helper function to get module name taking packages into account. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28712 diff changeset	220
27933 44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	221 class CoverageResults:
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	222 def __init__(self, counts=None, calledfuncs=None, infile=None,
31881 f35b4acc594f Added --trackcalls command line arg to display crude caller/callee Skip Montanaro <skip@pobox.com> parents: 31551 diff changeset	223 callers=None, outfile=None):
27933 44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	224 self.counts = counts
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	225 if self.counts is None:
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	226 self.counts = {}
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	227 self.counter = self.counts.copy() # map (filename, lineno) to count
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	228 self.calledfuncs = calledfuncs
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	229 if self.calledfuncs is None:
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	230 self.calledfuncs = {}
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	231 self.calledfuncs = self.calledfuncs.copy()
31881 f35b4acc594f Added --trackcalls command line arg to display crude caller/callee Skip Montanaro <skip@pobox.com> parents: 31551 diff changeset	232 self.callers = callers
f35b4acc594f Added --trackcalls command line arg to display crude caller/callee Skip Montanaro <skip@pobox.com> parents: 31551 diff changeset	233 if self.callers is None:
f35b4acc594f Added --trackcalls command line arg to display crude caller/callee Skip Montanaro <skip@pobox.com> parents: 31551 diff changeset	234 self.callers = {}
f35b4acc594f Added --trackcalls command line arg to display crude caller/callee Skip Montanaro <skip@pobox.com> parents: 31551 diff changeset	235 self.callers = self.callers.copy()
27933 44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	236 self.infile = infile
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	237 self.outfile = outfile
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	238 if self.infile:
29767 7770b7881346 Patch from Zooko to remove an experimental feature. Jeremy Hylton <jeremy@alum.mit.edu> parents: 29620 diff changeset	239 # Try to merge existing counts file.
27933 44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	240 try:
31881 f35b4acc594f Added --trackcalls command line arg to display crude caller/callee Skip Montanaro <skip@pobox.com> parents: 31551 diff changeset	241 counts, calledfuncs, callers = \
f35b4acc594f Added --trackcalls command line arg to display crude caller/callee Skip Montanaro <skip@pobox.com> parents: 31551 diff changeset	242 pickle.load(open(self.infile, 'rb'))
f35b4acc594f Added --trackcalls command line arg to display crude caller/callee Skip Montanaro <skip@pobox.com> parents: 31551 diff changeset	243 self.update(self.__class__(counts, calledfuncs, callers))
29767 7770b7881346 Patch from Zooko to remove an experimental feature. Jeremy Hylton <jeremy@alum.mit.edu> parents: 29620 diff changeset	244 except (IOError, EOFError, ValueError), err:
28712 d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	245 print >> sys.stderr, ("Skipping counts file %r: %s"
d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	246 % (self.infile, err))
27933 44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	247
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	248 def update(self, other):
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	249 """Merge in the data from another CoverageResults"""
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	250 counts = self.counts
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	251 calledfuncs = self.calledfuncs
31881 f35b4acc594f Added --trackcalls command line arg to display crude caller/callee Skip Montanaro <skip@pobox.com> parents: 31551 diff changeset	252 callers = self.callers
27933 44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	253 other_counts = other.counts
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	254 other_calledfuncs = other.calledfuncs
31881 f35b4acc594f Added --trackcalls command line arg to display crude caller/callee Skip Montanaro <skip@pobox.com> parents: 31551 diff changeset	255 other_callers = other.callers
27933 44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	256
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	257 for key in other_counts.keys():
28712 d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	258 counts[key] = counts.get(key, 0) + other_counts[key]
27933 44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	259
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	260 for key in other_calledfuncs.keys():
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	261 calledfuncs[key] = 1
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	262
31881 f35b4acc594f Added --trackcalls command line arg to display crude caller/callee Skip Montanaro <skip@pobox.com> parents: 31551 diff changeset	263 for key in other_callers.keys():
f35b4acc594f Added --trackcalls command line arg to display crude caller/callee Skip Montanaro <skip@pobox.com> parents: 31551 diff changeset	264 callers[key] = 1
f35b4acc594f Added --trackcalls command line arg to display crude caller/callee Skip Montanaro <skip@pobox.com> parents: 31551 diff changeset	265
28712 d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	266 def write_results(self, show_missing=True, summary=False, coverdir=None):
27933 44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	267 """
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	268 @param coverdir
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	269 """
31881 f35b4acc594f Added --trackcalls command line arg to display crude caller/callee Skip Montanaro <skip@pobox.com> parents: 31551 diff changeset	270 if self.calledfuncs:
31892 081972ae2f84 Refactor common code out of globaltrace_trackcallers() and Skip Montanaro <skip@pobox.com> parents: 31881 diff changeset	271 print
31881 f35b4acc594f Added --trackcalls command line arg to display crude caller/callee Skip Montanaro <skip@pobox.com> parents: 31551 diff changeset	272 print "functions called:"
f35b4acc594f Added --trackcalls command line arg to display crude caller/callee Skip Montanaro <skip@pobox.com> parents: 31551 diff changeset	273 calls = self.calledfuncs.keys()
f35b4acc594f Added --trackcalls command line arg to display crude caller/callee Skip Montanaro <skip@pobox.com> parents: 31551 diff changeset	274 calls.sort()
f35b4acc594f Added --trackcalls command line arg to display crude caller/callee Skip Montanaro <skip@pobox.com> parents: 31551 diff changeset	275 for filename, modulename, funcname in calls:
f35b4acc594f Added --trackcalls command line arg to display crude caller/callee Skip Montanaro <skip@pobox.com> parents: 31551 diff changeset	276 print ("filename: %s, modulename: %s, funcname: %s"
f35b4acc594f Added --trackcalls command line arg to display crude caller/callee Skip Montanaro <skip@pobox.com> parents: 31551 diff changeset	277 % (filename, modulename, funcname))
f35b4acc594f Added --trackcalls command line arg to display crude caller/callee Skip Montanaro <skip@pobox.com> parents: 31551 diff changeset	278
f35b4acc594f Added --trackcalls command line arg to display crude caller/callee Skip Montanaro <skip@pobox.com> parents: 31551 diff changeset	279 if self.callers:
31892 081972ae2f84 Refactor common code out of globaltrace_trackcallers() and Skip Montanaro <skip@pobox.com> parents: 31881 diff changeset	280 print
31881 f35b4acc594f Added --trackcalls command line arg to display crude caller/callee Skip Montanaro <skip@pobox.com> parents: 31551 diff changeset	281 print "calling relationships:"
f35b4acc594f Added --trackcalls command line arg to display crude caller/callee Skip Montanaro <skip@pobox.com> parents: 31551 diff changeset	282 calls = self.callers.keys()
f35b4acc594f Added --trackcalls command line arg to display crude caller/callee Skip Montanaro <skip@pobox.com> parents: 31551 diff changeset	283 calls.sort()
f35b4acc594f Added --trackcalls command line arg to display crude caller/callee Skip Montanaro <skip@pobox.com> parents: 31551 diff changeset	284 lastfile = lastcfile = ""
f35b4acc594f Added --trackcalls command line arg to display crude caller/callee Skip Montanaro <skip@pobox.com> parents: 31551 diff changeset	285 for ((pfile, pmod, pfunc), (cfile, cmod, cfunc)) in calls:
f35b4acc594f Added --trackcalls command line arg to display crude caller/callee Skip Montanaro <skip@pobox.com> parents: 31551 diff changeset	286 if pfile != lastfile:
31892 081972ae2f84 Refactor common code out of globaltrace_trackcallers() and Skip Montanaro <skip@pobox.com> parents: 31881 diff changeset	287 print
31881 f35b4acc594f Added --trackcalls command line arg to display crude caller/callee Skip Montanaro <skip@pobox.com> parents: 31551 diff changeset	288 print "*", pfile, "*"
f35b4acc594f Added --trackcalls command line arg to display crude caller/callee Skip Montanaro <skip@pobox.com> parents: 31551 diff changeset	289 lastfile = pfile
f35b4acc594f Added --trackcalls command line arg to display crude caller/callee Skip Montanaro <skip@pobox.com> parents: 31551 diff changeset	290 lastcfile = ""
f35b4acc594f Added --trackcalls command line arg to display crude caller/callee Skip Montanaro <skip@pobox.com> parents: 31551 diff changeset	291 if cfile != pfile and lastcfile != cfile:
f35b4acc594f Added --trackcalls command line arg to display crude caller/callee Skip Montanaro <skip@pobox.com> parents: 31551 diff changeset	292 print " -->", cfile
f35b4acc594f Added --trackcalls command line arg to display crude caller/callee Skip Montanaro <skip@pobox.com> parents: 31551 diff changeset	293 lastcfile = cfile
f35b4acc594f Added --trackcalls command line arg to display crude caller/callee Skip Montanaro <skip@pobox.com> parents: 31551 diff changeset	294 print " %s.%s -> %s.%s" % (pmod, pfunc, cmod, cfunc)
27933 44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	295
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	296 # turn the counts data ("(filename, lineno) = count") into something
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	297 # accessible on a per-file basis
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	298 per_file = {}
28712 d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	299 for filename, lineno in self.counts.keys():
d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	300 lines_hit = per_file[filename] = per_file.get(filename, {})
d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	301 lines_hit[lineno] = self.counts[(filename, lineno)]
27933 44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	302
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	303 # accumulate summary info, if needed
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	304 sums = {}
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	305
28712 d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	306 for filename, count in per_file.iteritems():
27933 44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	307 # skip some "files" we don't care about...
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	308 if filename == "<string>":
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	309 continue
44014 e0465ba0e95d back in these go - thanks to Titus Brown for the fix Skip Montanaro <skip@pobox.com> parents: 44001 diff changeset	310 if filename.startswith("<doctest "):
e0465ba0e95d back in these go - thanks to Titus Brown for the fix Skip Montanaro <skip@pobox.com> parents: 44001 diff changeset	311 continue
27933 44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	312
38704 ba6eebf7c719 Make use of new str.startswith/endswith semantics. Georg Brandl <georg@python.org> parents: 37964 diff changeset	313 if filename.endswith((".pyc", ".pyo")):
27933 44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	314 filename = filename[:-1]
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	315
28712 d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	316 if coverdir is None:
d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	317 dir = os.path.dirname(os.path.abspath(filename))
28727 051a4f12cc46 Only produce a dotted module name when writing output to a separate Jeremy Hylton <jeremy@alum.mit.edu> parents: 28713 diff changeset	318 modulename = modname(filename)
27933 44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	319 else:
28712 d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	320 dir = coverdir
d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	321 if not os.path.exists(dir):
d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	322 os.makedirs(dir)
28727 051a4f12cc46 Only produce a dotted module name when writing output to a separate Jeremy Hylton <jeremy@alum.mit.edu> parents: 28713 diff changeset	323 modulename = fullmodname(filename)
27933 44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	324
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	325 # If desired, get a list of the line numbers which represent
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	326 # executable content (returned as a dict for better lookup speed)
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	327 if show_missing:
28712 d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	328 lnotab = find_executable_linenos(filename)
27933 44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	329 else:
28712 d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	330 lnotab = {}
27933 44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	331
28712 d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	332 source = linecache.getlines(filename)
d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	333 coverpath = os.path.join(dir, modulename + ".cover")
d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	334 n_hits, n_lines = self.write_results_file(coverpath, source,
d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	335 lnotab, count)
28774 c9bee11eaa8b Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 28727 diff changeset	336
27933 44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	337 if summary and n_lines:
69264 7b71872fb66a Use floor division operator instead of deprecated division operator. Raymond Hettinger <python@rcn.com> parents: 68256 diff changeset	338 percent = 100 * n_hits // n_lines
27933 44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	339 sums[modulename] = n_lines, percent, modulename, filename
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	340
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	341 if summary and sums:
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	342 mods = sums.keys()
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	343 mods.sort()
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	344 print "lines cov% module (path)"
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	345 for m in mods:
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	346 n_lines, percent, modulename, filename = sums[m]
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	347 print "%5d %3d%% %s (%s)" % sums[m]
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	348
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	349 if self.outfile:
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	350 # try and store counts and module info into self.outfile
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	351 try:
31881 f35b4acc594f Added --trackcalls command line arg to display crude caller/callee Skip Montanaro <skip@pobox.com> parents: 31551 diff changeset	352 pickle.dump((self.counts, self.calledfuncs, self.callers),
30692 a3e86665f245 Open results files, which contain binary pickles, in binary mode. Jeremy Hylton <jeremy@alum.mit.edu> parents: 30668 diff changeset	353 open(self.outfile, 'wb'), 1)
27933 44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	354 except IOError, err:
28712 d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	355 print >> sys.stderr, "Can't save counts files because %s" % err
d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	356
d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	357 def write_results_file(self, path, lines, lnotab, lines_hit):
d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	358 """Return a coverage results file in path."""
d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	359
d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	360 try:
d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	361 outfile = open(path, "w")
d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	362 except IOError, err:
d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	363 print >> sys.stderr, ("trace: Could not open %r for writing: %s"
d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	364 "- skipping" % (path, err))
31551 a157cc20f1b7 Fix two small bugs: (1) on Windows, pathname munging didn't work Guido van Rossum <guido@python.org> parents: 31515 diff changeset	365 return 0, 0
27933 44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	366
28712 d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	367 n_lines = 0
d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	368 n_hits = 0
d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	369 for i, line in enumerate(lines):
d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	370 lineno = i + 1
d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	371 # do the blank/comment match to try to mark more lines
d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	372 # (help the reader find stuff that hasn't been covered)
d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	373 if lineno in lines_hit:
d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	374 outfile.write("%5d: " % lines_hit[lineno])
d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	375 n_hits += 1
d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	376 n_lines += 1
d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	377 elif rx_blank.match(line):
29864 84bccbeab5fa Make the prefix 7 spaces long, so the lines are properly aligned. Walter Dörwald <walter@livinglogic.de> parents: 29767 diff changeset	378 outfile.write(" ")
28712 d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	379 else:
d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	380 # lines preceded by no marks weren't hit
d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	381 # Highlight them if so indicated, unless the line contains
d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	382 # #pragma: NO COVER
d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	383 if lineno in lnotab and not PRAGMA_NOCOVER in lines[i]:
d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	384 outfile.write(">>>>>> ")
29595 8911ca88981a Enable tracing of multi-threaded applications. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28774 diff changeset	385 n_lines += 1
28712 d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	386 else:
d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	387 outfile.write(" ")
d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	388 outfile.write(lines[i].expandtabs(8))
d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	389 outfile.close()
d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	390
d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	391 return n_hits, n_lines
d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	392
d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	393 def find_lines_from_code(code, strs):
d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	394 """Return dict where keys are lines in the line number table."""
27933 44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	395 linenos = {}
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	396
64926 12e0dffef7bd Merged revisions 84994 via svnmerge from Alexander Belopolsky <alexander.belopolsky@gmail.com> parents: 64711 diff changeset	397 for _, lineno in dis.findlinestarts(code):
28712 d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	398 if lineno not in strs:
d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	399 linenos[lineno] = 1
27933 44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	400
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	401 return linenos
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	402
28712 d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	403 def find_lines(code, strs):
d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	404 """Return lineno dict for all code objects reachable from code."""
27933 44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	405 # get all of the lineno information from the code of this scope level
28712 d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	406 linenos = find_lines_from_code(code, strs)
27933 44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	407
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	408 # and check the constants for references to other code objects
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	409 for c in code.co_consts:
64711 0f3b8f928ec1 Issue #9315: Fix for the trace module to record correct class name Alexander Belopolsky <alexander.belopolsky@gmail.com> parents: 63605 diff changeset	410 if inspect.iscode(c):
27933 44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	411 # find another code object, so recurse into it
28712 d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	412 linenos.update(find_lines(c, strs))
27933 44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	413 return linenos
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	414
28712 d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	415 def find_strings(filename):
d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	416 """Return a dict of possible docstring positions.
d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	417
d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	418 The dict maps line numbers to strings. There is an entry for
d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	419 line that contains only a string or a part of a triple-quoted
d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	420 string.
d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	421 """
d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	422 d = {}
d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	423 # If the first token is a string, then it's the module docstring.
d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	424 # Add this special case so that the test in the loop passes.
d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	425 prev_ttype = token.INDENT
d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	426 f = open(filename)
d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	427 for ttype, tstr, start, end, line in tokenize.generate_tokens(f.readline):
d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	428 if ttype == token.STRING:
d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	429 if prev_ttype == token.INDENT:
d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	430 sline, scol = start
d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	431 eline, ecol = end
d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	432 for i in range(sline, eline + 1):
d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	433 d[i] = 1
d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	434 prev_ttype = ttype
d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	435 f.close()
d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	436 return d
d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	437
27933 44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	438 def find_executable_linenos(filename):
28712 d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	439 """Return dict where keys are line numbers in the line number table."""
d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	440 try:
31919 39c36526d1b0 Open file in universal newline mode when passing to compile(). Solution Skip Montanaro <skip@pobox.com> parents: 31892 diff changeset	441 prog = open(filename, "rU").read()
28712 d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	442 except IOError, err:
d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	443 print >> sys.stderr, ("Not printing coverage data for %r: %s"
d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	444 % (filename, err))
d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	445 return {}
d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	446 code = compile(prog, filename, "exec")
d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	447 strs = find_strings(filename)
d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	448 return find_lines(code, strs)
27933 44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	449
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	450 class Trace:
31881 f35b4acc594f Added --trackcalls command line arg to display crude caller/callee Skip Montanaro <skip@pobox.com> parents: 31551 diff changeset	451 def __init__(self, count=1, trace=1, countfuncs=0, countcallers=0,
45535 c57cb2f8c350 Add a timing flag to Trace so you can see where slowness occurs Neal Norwitz <nnorwitz@gmail.com> parents: 44775 diff changeset	452 ignoremods=(), ignoredirs=(), infile=None, outfile=None,
c57cb2f8c350 Add a timing flag to Trace so you can see where slowness occurs Neal Norwitz <nnorwitz@gmail.com> parents: 44775 diff changeset	453 timing=False):
27933 44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	454 """
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	455 @param count true iff it should count number of times each
27963 226463445697 Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 27933 diff changeset	456 line is executed
27933 44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	457 @param trace true iff it should print out each line that is
27963 226463445697 Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 27933 diff changeset	458 being counted
27933 44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	459 @param countfuncs true iff it should just output a list of
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	460 (filename, modulename, funcname,) for functions
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	461 that were called at least once; This overrides
27963 226463445697 Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 27933 diff changeset	462 `count' and `trace'
27933 44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	463 @param ignoremods a list of the names of modules to ignore
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	464 @param ignoredirs a list of the names of directories to ignore
27963 226463445697 Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 27933 diff changeset	465 all of the (recursive) contents of
27933 44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	466 @param infile file from which to read stored counts to be
27963 226463445697 Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 27933 diff changeset	467 added into the results
27933 44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	468 @param outfile file in which to write the results
45535 c57cb2f8c350 Add a timing flag to Trace so you can see where slowness occurs Neal Norwitz <nnorwitz@gmail.com> parents: 44775 diff changeset	469 @param timing true iff timing information be displayed
27933 44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	470 """
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	471 self.infile = infile
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	472 self.outfile = outfile
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	473 self.ignore = Ignore(ignoremods, ignoredirs)
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	474 self.counts = {} # keys are (filename, linenumber)
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	475 self.blabbed = {} # for debugging
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	476 self.pathtobasename = {} # for memoizing os.path.basename
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	477 self.donothing = 0
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	478 self.trace = trace
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	479 self._calledfuncs = {}
31881 f35b4acc594f Added --trackcalls command line arg to display crude caller/callee Skip Montanaro <skip@pobox.com> parents: 31551 diff changeset	480 self._callers = {}
31892 081972ae2f84 Refactor common code out of globaltrace_trackcallers() and Skip Montanaro <skip@pobox.com> parents: 31881 diff changeset	481 self._caller_cache = {}
45535 c57cb2f8c350 Add a timing flag to Trace so you can see where slowness occurs Neal Norwitz <nnorwitz@gmail.com> parents: 44775 diff changeset	482 self.start_time = None
c57cb2f8c350 Add a timing flag to Trace so you can see where slowness occurs Neal Norwitz <nnorwitz@gmail.com> parents: 44775 diff changeset	483 if timing:
c57cb2f8c350 Add a timing flag to Trace so you can see where slowness occurs Neal Norwitz <nnorwitz@gmail.com> parents: 44775 diff changeset	484 self.start_time = time.time()
31881 f35b4acc594f Added --trackcalls command line arg to display crude caller/callee Skip Montanaro <skip@pobox.com> parents: 31551 diff changeset	485 if countcallers:
f35b4acc594f Added --trackcalls command line arg to display crude caller/callee Skip Montanaro <skip@pobox.com> parents: 31551 diff changeset	486 self.globaltrace = self.globaltrace_trackcallers
f35b4acc594f Added --trackcalls command line arg to display crude caller/callee Skip Montanaro <skip@pobox.com> parents: 31551 diff changeset	487 elif countfuncs:
27933 44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	488 self.globaltrace = self.globaltrace_countfuncs
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	489 elif trace and count:
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	490 self.globaltrace = self.globaltrace_lt
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	491 self.localtrace = self.localtrace_trace_and_count
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	492 elif trace:
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	493 self.globaltrace = self.globaltrace_lt
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	494 self.localtrace = self.localtrace_trace
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	495 elif count:
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	496 self.globaltrace = self.globaltrace_lt
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	497 self.localtrace = self.localtrace_count
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	498 else:
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	499 # Ahem -- do nothing? Okay.
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	500 self.donothing = 1
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	501
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	502 def run(self, cmd):
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	503 import __main__
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	504 dict = __main__.__dict__
72118 bb7b14dd5ded Fix test failures when Python is compiled without thread support. Nadeem Vawda <nadeem.vawda@gmail.com> parents: 69264 diff changeset	505 self.runctx(cmd, dict, dict)
27933 44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	506
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	507 def runctx(self, cmd, globals=None, locals=None):
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	508 if globals is None: globals = {}
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	509 if locals is None: locals = {}
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	510 if not self.donothing:
66003 acb0a3cc1e18 Merged revisions 86229 via svnmerge from Alexander Belopolsky <alexander.belopolsky@gmail.com> parents: 64926 diff changeset	511 _settrace(self.globaltrace)
27933 44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	512 try:
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	513 exec cmd in globals, locals
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	514 finally:
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	515 if not self.donothing:
66003 acb0a3cc1e18 Merged revisions 86229 via svnmerge from Alexander Belopolsky <alexander.belopolsky@gmail.com> parents: 64926 diff changeset	516 _unsettrace()
27933 44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	517
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	518 def runfunc(self, func, args, *kw):
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	519 result = None
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	520 if not self.donothing:
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	521 sys.settrace(self.globaltrace)
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	522 try:
28089 14f65d661084 Get rid of many apply() calls. Guido van Rossum <guido@python.org> parents: 27963 diff changeset	523 result = func(args, *kw)
27933 44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	524 finally:
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	525 if not self.donothing:
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	526 sys.settrace(None)
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	527 return result
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	528
31892 081972ae2f84 Refactor common code out of globaltrace_trackcallers() and Skip Montanaro <skip@pobox.com> parents: 31881 diff changeset	529 def file_module_function_of(self, frame):
081972ae2f84 Refactor common code out of globaltrace_trackcallers() and Skip Montanaro <skip@pobox.com> parents: 31881 diff changeset	530 code = frame.f_code
081972ae2f84 Refactor common code out of globaltrace_trackcallers() and Skip Montanaro <skip@pobox.com> parents: 31881 diff changeset	531 filename = code.co_filename
081972ae2f84 Refactor common code out of globaltrace_trackcallers() and Skip Montanaro <skip@pobox.com> parents: 31881 diff changeset	532 if filename:
081972ae2f84 Refactor common code out of globaltrace_trackcallers() and Skip Montanaro <skip@pobox.com> parents: 31881 diff changeset	533 modulename = modname(filename)
081972ae2f84 Refactor common code out of globaltrace_trackcallers() and Skip Montanaro <skip@pobox.com> parents: 31881 diff changeset	534 else:
081972ae2f84 Refactor common code out of globaltrace_trackcallers() and Skip Montanaro <skip@pobox.com> parents: 31881 diff changeset	535 modulename = None
081972ae2f84 Refactor common code out of globaltrace_trackcallers() and Skip Montanaro <skip@pobox.com> parents: 31881 diff changeset	536
081972ae2f84 Refactor common code out of globaltrace_trackcallers() and Skip Montanaro <skip@pobox.com> parents: 31881 diff changeset	537 funcname = code.co_name
081972ae2f84 Refactor common code out of globaltrace_trackcallers() and Skip Montanaro <skip@pobox.com> parents: 31881 diff changeset	538 clsname = None
081972ae2f84 Refactor common code out of globaltrace_trackcallers() and Skip Montanaro <skip@pobox.com> parents: 31881 diff changeset	539 if code in self._caller_cache:
081972ae2f84 Refactor common code out of globaltrace_trackcallers() and Skip Montanaro <skip@pobox.com> parents: 31881 diff changeset	540 if self._caller_cache[code] is not None:
081972ae2f84 Refactor common code out of globaltrace_trackcallers() and Skip Montanaro <skip@pobox.com> parents: 31881 diff changeset	541 clsname = self._caller_cache[code]
081972ae2f84 Refactor common code out of globaltrace_trackcallers() and Skip Montanaro <skip@pobox.com> parents: 31881 diff changeset	542 else:
081972ae2f84 Refactor common code out of globaltrace_trackcallers() and Skip Montanaro <skip@pobox.com> parents: 31881 diff changeset	543 self._caller_cache[code] = None
081972ae2f84 Refactor common code out of globaltrace_trackcallers() and Skip Montanaro <skip@pobox.com> parents: 31881 diff changeset	544 ## use of gc.get_referrers() was suggested by Michael Hudson
081972ae2f84 Refactor common code out of globaltrace_trackcallers() and Skip Montanaro <skip@pobox.com> parents: 31881 diff changeset	545 # all functions which refer to this code object
081972ae2f84 Refactor common code out of globaltrace_trackcallers() and Skip Montanaro <skip@pobox.com> parents: 31881 diff changeset	546 funcs = [f for f in gc.get_referrers(code)
64711 0f3b8f928ec1 Issue #9315: Fix for the trace module to record correct class name Alexander Belopolsky <alexander.belopolsky@gmail.com> parents: 63605 diff changeset	547 if inspect.isfunction(f)]
31892 081972ae2f84 Refactor common code out of globaltrace_trackcallers() and Skip Montanaro <skip@pobox.com> parents: 31881 diff changeset	548 # require len(func) == 1 to avoid ambiguity caused by calls to
081972ae2f84 Refactor common code out of globaltrace_trackcallers() and Skip Montanaro <skip@pobox.com> parents: 31881 diff changeset	549 # new.function(): "In the face of ambiguity, refuse the
081972ae2f84 Refactor common code out of globaltrace_trackcallers() and Skip Montanaro <skip@pobox.com> parents: 31881 diff changeset	550 # temptation to guess."
081972ae2f84 Refactor common code out of globaltrace_trackcallers() and Skip Montanaro <skip@pobox.com> parents: 31881 diff changeset	551 if len(funcs) == 1:
081972ae2f84 Refactor common code out of globaltrace_trackcallers() and Skip Montanaro <skip@pobox.com> parents: 31881 diff changeset	552 dicts = [d for d in gc.get_referrers(funcs[0])
081972ae2f84 Refactor common code out of globaltrace_trackcallers() and Skip Montanaro <skip@pobox.com> parents: 31881 diff changeset	553 if isinstance(d, dict)]
081972ae2f84 Refactor common code out of globaltrace_trackcallers() and Skip Montanaro <skip@pobox.com> parents: 31881 diff changeset	554 if len(dicts) == 1:
081972ae2f84 Refactor common code out of globaltrace_trackcallers() and Skip Montanaro <skip@pobox.com> parents: 31881 diff changeset	555 classes = [c for c in gc.get_referrers(dicts[0])
081972ae2f84 Refactor common code out of globaltrace_trackcallers() and Skip Montanaro <skip@pobox.com> parents: 31881 diff changeset	556 if hasattr(c, "__bases__")]
081972ae2f84 Refactor common code out of globaltrace_trackcallers() and Skip Montanaro <skip@pobox.com> parents: 31881 diff changeset	557 if len(classes) == 1:
081972ae2f84 Refactor common code out of globaltrace_trackcallers() and Skip Montanaro <skip@pobox.com> parents: 31881 diff changeset	558 # ditto for new.classobj()
64711 0f3b8f928ec1 Issue #9315: Fix for the trace module to record correct class name Alexander Belopolsky <alexander.belopolsky@gmail.com> parents: 63605 diff changeset	559 clsname = classes[0].__name__
31892 081972ae2f84 Refactor common code out of globaltrace_trackcallers() and Skip Montanaro <skip@pobox.com> parents: 31881 diff changeset	560 # cache the result - assumption is that new.* is
081972ae2f84 Refactor common code out of globaltrace_trackcallers() and Skip Montanaro <skip@pobox.com> parents: 31881 diff changeset	561 # not called later to disturb this relationship
081972ae2f84 Refactor common code out of globaltrace_trackcallers() and Skip Montanaro <skip@pobox.com> parents: 31881 diff changeset	562 # _caller_cache could be flushed if functions in
081972ae2f84 Refactor common code out of globaltrace_trackcallers() and Skip Montanaro <skip@pobox.com> parents: 31881 diff changeset	563 # the new module get called.
081972ae2f84 Refactor common code out of globaltrace_trackcallers() and Skip Montanaro <skip@pobox.com> parents: 31881 diff changeset	564 self._caller_cache[code] = clsname
081972ae2f84 Refactor common code out of globaltrace_trackcallers() and Skip Montanaro <skip@pobox.com> parents: 31881 diff changeset	565 if clsname is not None:
081972ae2f84 Refactor common code out of globaltrace_trackcallers() and Skip Montanaro <skip@pobox.com> parents: 31881 diff changeset	566 funcname = "%s.%s" % (clsname, funcname)
081972ae2f84 Refactor common code out of globaltrace_trackcallers() and Skip Montanaro <skip@pobox.com> parents: 31881 diff changeset	567
081972ae2f84 Refactor common code out of globaltrace_trackcallers() and Skip Montanaro <skip@pobox.com> parents: 31881 diff changeset	568 return filename, modulename, funcname
081972ae2f84 Refactor common code out of globaltrace_trackcallers() and Skip Montanaro <skip@pobox.com> parents: 31881 diff changeset	569
31881 f35b4acc594f Added --trackcalls command line arg to display crude caller/callee Skip Montanaro <skip@pobox.com> parents: 31551 diff changeset	570 def globaltrace_trackcallers(self, frame, why, arg):
f35b4acc594f Added --trackcalls command line arg to display crude caller/callee Skip Montanaro <skip@pobox.com> parents: 31551 diff changeset	571 """Handler for call events.
f35b4acc594f Added --trackcalls command line arg to display crude caller/callee Skip Montanaro <skip@pobox.com> parents: 31551 diff changeset	572
f35b4acc594f Added --trackcalls command line arg to display crude caller/callee Skip Montanaro <skip@pobox.com> parents: 31551 diff changeset	573 Adds information about who called who to the self._callers dict.
f35b4acc594f Added --trackcalls command line arg to display crude caller/callee Skip Montanaro <skip@pobox.com> parents: 31551 diff changeset	574 """
f35b4acc594f Added --trackcalls command line arg to display crude caller/callee Skip Montanaro <skip@pobox.com> parents: 31551 diff changeset	575 if why == 'call':
f35b4acc594f Added --trackcalls command line arg to display crude caller/callee Skip Montanaro <skip@pobox.com> parents: 31551 diff changeset	576 # XXX Should do a better job of identifying methods
31892 081972ae2f84 Refactor common code out of globaltrace_trackcallers() and Skip Montanaro <skip@pobox.com> parents: 31881 diff changeset	577 this_func = self.file_module_function_of(frame)
081972ae2f84 Refactor common code out of globaltrace_trackcallers() and Skip Montanaro <skip@pobox.com> parents: 31881 diff changeset	578 parent_func = self.file_module_function_of(frame.f_back)
31881 f35b4acc594f Added --trackcalls command line arg to display crude caller/callee Skip Montanaro <skip@pobox.com> parents: 31551 diff changeset	579 self._callers[(parent_func, this_func)] = 1
f35b4acc594f Added --trackcalls command line arg to display crude caller/callee Skip Montanaro <skip@pobox.com> parents: 31551 diff changeset	580
27933 44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	581 def globaltrace_countfuncs(self, frame, why, arg):
28712 d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	582 """Handler for call events.
28774 c9bee11eaa8b Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 28727 diff changeset	583
28712 d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	584 Adds (filename, modulename, funcname) to the self._calledfuncs dict.
27933 44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	585 """
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	586 if why == 'call':
31892 081972ae2f84 Refactor common code out of globaltrace_trackcallers() and Skip Montanaro <skip@pobox.com> parents: 31881 diff changeset	587 this_func = self.file_module_function_of(frame)
081972ae2f84 Refactor common code out of globaltrace_trackcallers() and Skip Montanaro <skip@pobox.com> parents: 31881 diff changeset	588 self._calledfuncs[this_func] = 1
27933 44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	589
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	590 def globaltrace_lt(self, frame, why, arg):
28712 d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	591 """Handler for call events.
d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	592
d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	593 If the code block being entered is to be ignored, returns `None',
d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	594 else returns self.localtrace.
27933 44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	595 """
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	596 if why == 'call':
28712 d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	597 code = frame.f_code
40811 5dbce2ea133c fix trace.py --ignore-dir Skip Montanaro <skip@pobox.com> parents: 39500 diff changeset	598 filename = frame.f_globals.get('__file__', None)
27933 44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	599 if filename:
28713 d08d15a70295 Add helper function to get module name taking packages into account. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28712 diff changeset	600 # XXX modname() doesn't work right for packages, so
d08d15a70295 Add helper function to get module name taking packages into account. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28712 diff changeset	601 # the ignore support won't work right for packages
28712 d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	602 modulename = modname(filename)
27933 44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	603 if modulename is not None:
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	604 ignore_it = self.ignore.names(filename, modulename)
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	605 if not ignore_it:
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	606 if self.trace:
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	607 print (" --- modulename: %s, funcname: %s"
28712 d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	608 % (modulename, code.co_name))
27933 44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	609 return self.localtrace
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	610 else:
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	611 return None
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	612
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	613 def localtrace_trace_and_count(self, frame, why, arg):
28712 d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	614 if why == "line":
27933 44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	615 # record the file name and line number of every trace
28712 d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	616 filename = frame.f_code.co_filename
d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	617 lineno = frame.f_lineno
27933 44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	618 key = filename, lineno
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	619 self.counts[key] = self.counts.get(key, 0) + 1
27963 226463445697 Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 27933 diff changeset	620
45535 c57cb2f8c350 Add a timing flag to Trace so you can see where slowness occurs Neal Norwitz <nnorwitz@gmail.com> parents: 44775 diff changeset	621 if self.start_time:
c57cb2f8c350 Add a timing flag to Trace so you can see where slowness occurs Neal Norwitz <nnorwitz@gmail.com> parents: 44775 diff changeset	622 print '%.2f' % (time.time() - self.start_time),
28712 d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	623 bname = os.path.basename(filename)
d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	624 print "%s(%d): %s" % (bname, lineno,
d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	625 linecache.getline(filename, lineno)),
27933 44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	626 return self.localtrace
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	627
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	628 def localtrace_trace(self, frame, why, arg):
28712 d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	629 if why == "line":
d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	630 # record the file name and line number of every trace
d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	631 filename = frame.f_code.co_filename
d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	632 lineno = frame.f_lineno
27963 226463445697 Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 27933 diff changeset	633
45535 c57cb2f8c350 Add a timing flag to Trace so you can see where slowness occurs Neal Norwitz <nnorwitz@gmail.com> parents: 44775 diff changeset	634 if self.start_time:
c57cb2f8c350 Add a timing flag to Trace so you can see where slowness occurs Neal Norwitz <nnorwitz@gmail.com> parents: 44775 diff changeset	635 print '%.2f' % (time.time() - self.start_time),
28712 d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	636 bname = os.path.basename(filename)
d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	637 print "%s(%d): %s" % (bname, lineno,
d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	638 linecache.getline(filename, lineno)),
27933 44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	639 return self.localtrace
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	640
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	641 def localtrace_count(self, frame, why, arg):
28712 d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	642 if why == "line":
27933 44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	643 filename = frame.f_code.co_filename
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	644 lineno = frame.f_lineno
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	645 key = filename, lineno
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	646 self.counts[key] = self.counts.get(key, 0) + 1
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	647 return self.localtrace
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	648
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	649 def results(self):
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	650 return CoverageResults(self.counts, infile=self.infile,
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	651 outfile=self.outfile,
31881 f35b4acc594f Added --trackcalls command line arg to display crude caller/callee Skip Montanaro <skip@pobox.com> parents: 31551 diff changeset	652 calledfuncs=self._calledfuncs,
f35b4acc594f Added --trackcalls command line arg to display crude caller/callee Skip Montanaro <skip@pobox.com> parents: 31551 diff changeset	653 callers=self._callers)
27933 44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	654
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	655 def _err_exit(msg):
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	656 sys.stderr.write("%s: %s\n" % (sys.argv[0], msg))
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	657 sys.exit(1)
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	658
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	659 def main(argv=None):
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	660 import getopt
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	661
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	662 if argv is None:
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	663 argv = sys.argv
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	664 try:
45535 c57cb2f8c350 Add a timing flag to Trace so you can see where slowness occurs Neal Norwitz <nnorwitz@gmail.com> parents: 44775 diff changeset	665 opts, prog_argv = getopt.getopt(argv[1:], "tcrRf:d:msC:lTg",
27933 44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	666 ["help", "version", "trace", "count",
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	667 "report", "no-report", "summary",
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	668 "file=", "missing",
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	669 "ignore-module=", "ignore-dir=",
31881 f35b4acc594f Added --trackcalls command line arg to display crude caller/callee Skip Montanaro <skip@pobox.com> parents: 31551 diff changeset	670 "coverdir=", "listfuncs",
45535 c57cb2f8c350 Add a timing flag to Trace so you can see where slowness occurs Neal Norwitz <nnorwitz@gmail.com> parents: 44775 diff changeset	671 "trackcalls", "timing"])
27933 44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	672
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	673 except getopt.error, msg:
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	674 sys.stderr.write("%s: %s\n" % (sys.argv[0], msg))
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	675 sys.stderr.write("Try `%s --help' for more information\n"
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	676 % sys.argv[0])
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	677 sys.exit(1)
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	678
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	679 trace = 0
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	680 count = 0
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	681 report = 0
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	682 no_report = 0
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	683 counts_file = None
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	684 missing = 0
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	685 ignore_modules = []
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	686 ignore_dirs = []
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	687 coverdir = None
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	688 summary = 0
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	689 listfuncs = False
31881 f35b4acc594f Added --trackcalls command line arg to display crude caller/callee Skip Montanaro <skip@pobox.com> parents: 31551 diff changeset	690 countcallers = False
45535 c57cb2f8c350 Add a timing flag to Trace so you can see where slowness occurs Neal Norwitz <nnorwitz@gmail.com> parents: 44775 diff changeset	691 timing = False
27933 44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	692
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	693 for opt, val in opts:
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	694 if opt == "--help":
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	695 usage(sys.stdout)
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	696 sys.exit(0)
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	697
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	698 if opt == "--version":
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	699 sys.stdout.write("trace 2.0\n")
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	700 sys.exit(0)
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	701
31881 f35b4acc594f Added --trackcalls command line arg to display crude caller/callee Skip Montanaro <skip@pobox.com> parents: 31551 diff changeset	702 if opt == "-T" or opt == "--trackcalls":
f35b4acc594f Added --trackcalls command line arg to display crude caller/callee Skip Montanaro <skip@pobox.com> parents: 31551 diff changeset	703 countcallers = True
f35b4acc594f Added --trackcalls command line arg to display crude caller/callee Skip Montanaro <skip@pobox.com> parents: 31551 diff changeset	704 continue
f35b4acc594f Added --trackcalls command line arg to display crude caller/callee Skip Montanaro <skip@pobox.com> parents: 31551 diff changeset	705
27933 44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	706 if opt == "-l" or opt == "--listfuncs":
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	707 listfuncs = True
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	708 continue
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	709
45535 c57cb2f8c350 Add a timing flag to Trace so you can see where slowness occurs Neal Norwitz <nnorwitz@gmail.com> parents: 44775 diff changeset	710 if opt == "-g" or opt == "--timing":
c57cb2f8c350 Add a timing flag to Trace so you can see where slowness occurs Neal Norwitz <nnorwitz@gmail.com> parents: 44775 diff changeset	711 timing = True
c57cb2f8c350 Add a timing flag to Trace so you can see where slowness occurs Neal Norwitz <nnorwitz@gmail.com> parents: 44775 diff changeset	712 continue
c57cb2f8c350 Add a timing flag to Trace so you can see where slowness occurs Neal Norwitz <nnorwitz@gmail.com> parents: 44775 diff changeset	713
27933 44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	714 if opt == "-t" or opt == "--trace":
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	715 trace = 1
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	716 continue
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	717
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	718 if opt == "-c" or opt == "--count":
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	719 count = 1
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	720 continue
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	721
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	722 if opt == "-r" or opt == "--report":
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	723 report = 1
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	724 continue
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	725
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	726 if opt == "-R" or opt == "--no-report":
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	727 no_report = 1
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	728 continue
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	729
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	730 if opt == "-f" or opt == "--file":
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	731 counts_file = val
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	732 continue
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	733
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	734 if opt == "-m" or opt == "--missing":
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	735 missing = 1
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	736 continue
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	737
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	738 if opt == "-C" or opt == "--coverdir":
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	739 coverdir = val
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	740 continue
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	741
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	742 if opt == "-s" or opt == "--summary":
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	743 summary = 1
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	744 continue
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	745
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	746 if opt == "--ignore-module":
44775 9c0a913adcd4 Fix #1693149. Now you can pass several modules separated by Facundo Batista <facundobatista@gmail.com> parents: 44014 diff changeset	747 for mod in val.split(","):
9c0a913adcd4 Fix #1693149. Now you can pass several modules separated by Facundo Batista <facundobatista@gmail.com> parents: 44014 diff changeset	748 ignore_modules.append(mod.strip())
27933 44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	749 continue
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	750
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	751 if opt == "--ignore-dir":
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	752 for s in val.split(os.pathsep):
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	753 s = os.path.expandvars(s)
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	754 # should I also call expanduser? (after all, could use $HOME)
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	755
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	756 s = s.replace("$prefix",
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	757 os.path.join(sys.prefix, "lib",
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	758 "python" + sys.version[:3]))
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	759 s = s.replace("$exec_prefix",
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	760 os.path.join(sys.exec_prefix, "lib",
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	761 "python" + sys.version[:3]))
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	762 s = os.path.normpath(s)
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	763 ignore_dirs.append(s)
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	764 continue
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	765
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	766 assert 0, "Should never get here"
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	767
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	768 if listfuncs and (count or trace):
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	769 _err_exit("cannot specify both --listfuncs and (--trace or --count)")
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	770
31881 f35b4acc594f Added --trackcalls command line arg to display crude caller/callee Skip Montanaro <skip@pobox.com> parents: 31551 diff changeset	771 if not (count or trace or report or listfuncs or countcallers):
f35b4acc594f Added --trackcalls command line arg to display crude caller/callee Skip Montanaro <skip@pobox.com> parents: 31551 diff changeset	772 _err_exit("must specify one of --trace, --count, --report, "
f35b4acc594f Added --trackcalls command line arg to display crude caller/callee Skip Montanaro <skip@pobox.com> parents: 31551 diff changeset	773 "--listfuncs, or --trackcalls")
27933 44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	774
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	775 if report and no_report:
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	776 _err_exit("cannot specify both --report and --no-report")
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	777
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	778 if report and not counts_file:
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	779 _err_exit("--report requires a --file")
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	780
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	781 if no_report and len(prog_argv) == 0:
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	782 _err_exit("missing name of file to run")
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	783
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	784 # everything is ready
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	785 if report:
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	786 results = CoverageResults(infile=counts_file, outfile=counts_file)
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	787 results.write_results(missing, summary=summary, coverdir=coverdir)
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	788 else:
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	789 sys.argv = prog_argv
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	790 progname = prog_argv[0]
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	791 sys.path[0] = os.path.split(progname)[0]
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	792
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	793 t = Trace(count, trace, countfuncs=listfuncs,
31881 f35b4acc594f Added --trackcalls command line arg to display crude caller/callee Skip Montanaro <skip@pobox.com> parents: 31551 diff changeset	794 countcallers=countcallers, ignoremods=ignore_modules,
f35b4acc594f Added --trackcalls command line arg to display crude caller/callee Skip Montanaro <skip@pobox.com> parents: 31551 diff changeset	795 ignoredirs=ignore_dirs, infile=counts_file,
45535 c57cb2f8c350 Add a timing flag to Trace so you can see where slowness occurs Neal Norwitz <nnorwitz@gmail.com> parents: 44775 diff changeset	796 outfile=counts_file, timing=timing)
27933 44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	797 try:
63511 61ad81a774c5 Merged revisions 83421,83424 via svnmerge from Georg Brandl <georg@python.org> parents: 45535 diff changeset	798 with open(progname) as fp:
61ad81a774c5 Merged revisions 83421,83424 via svnmerge from Georg Brandl <georg@python.org> parents: 45535 diff changeset	799 code = compile(fp.read(), progname, 'exec')
61ad81a774c5 Merged revisions 83421,83424 via svnmerge from Georg Brandl <georg@python.org> parents: 45535 diff changeset	800 # try to emulate __main__ namespace as much as possible
61ad81a774c5 Merged revisions 83421,83424 via svnmerge from Georg Brandl <georg@python.org> parents: 45535 diff changeset	801 globs = {
61ad81a774c5 Merged revisions 83421,83424 via svnmerge from Georg Brandl <georg@python.org> parents: 45535 diff changeset	802 '__file__': progname,
61ad81a774c5 Merged revisions 83421,83424 via svnmerge from Georg Brandl <georg@python.org> parents: 45535 diff changeset	803 '__name__': '__main__',
61ad81a774c5 Merged revisions 83421,83424 via svnmerge from Georg Brandl <georg@python.org> parents: 45535 diff changeset	804 '__package__': None,
61ad81a774c5 Merged revisions 83421,83424 via svnmerge from Georg Brandl <georg@python.org> parents: 45535 diff changeset	805 '__cached__': None,
61ad81a774c5 Merged revisions 83421,83424 via svnmerge from Georg Brandl <georg@python.org> parents: 45535 diff changeset	806 }
61ad81a774c5 Merged revisions 83421,83424 via svnmerge from Georg Brandl <georg@python.org> parents: 45535 diff changeset	807 t.runctx(code, globs, globs)
27933 44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	808 except IOError, err:
28712 d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	809 _err_exit("Cannot run file %r because: %s" % (sys.argv[0], err))
27933 44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	810 except SystemExit:
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	811 pass
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	812
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	813 results = t.results()
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	814
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	815 if not no_report:
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	816 results.write_results(missing, summary=summary, coverdir=coverdir)
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	817
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	818 if __name__=='__main__':
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	819 main()

Mercurial > cpython-lru_cache-weakref

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