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

annotate Lib/trace.py @ 39042:c02fa187e0f4 legacy-trunk

Part of SF patch #1484695. This removes dead code. The chksum was already verified in .frombuf() on the lines above. If there was a problem an exception is raised, so there was no way this condition could have been true.

author	Neal Norwitz <nnorwitz@gmail.com>
date	Mon, 10 Jul 2006 00:23:17 +0000
parents	ba6eebf7c719
children	207e487f425d

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
29595 8911ca88981a Enable tracing of multi-threaded applications. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28774 diff changeset	55 import threading
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
d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	58 import types
31892 081972ae2f84 Refactor common code out of globaltrace_trackcallers() and Skip Montanaro <skip@pobox.com> parents: 31881 diff changeset	59 import gc
28712 d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	60
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
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	67 def usage(outfile):
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	68 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	69
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	70 Meta-options:
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	71 --help Display this help then exit.
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	72 --version Output version information then exit.
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	73
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	74 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	75 -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	76 -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	77 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	78 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	79 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	80 -l, --listfuncs Keep track of which functions are executed at least
29620 77eda0fb0e14 fix typo Fred Drake <fdrake@acm.org> parents: 29619 diff changeset	81 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	82 program exits.
31881 f35b4acc594f Added --trackcalls command line arg to display crude caller/callee Skip Montanaro <skip@pobox.com> parents: 31551 diff changeset	83 -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	84 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	85 -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	86 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	87 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	88 with `--count --file=FILE'.
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 Modifiers:
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	91 -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	92 -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	93 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	94 -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	95 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	96 <dir>/<package>/<module>.cover.
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	97 -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	98 with '>>>>>> '.
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	99 -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	100 (Can only be used with --count or --report.)
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	101
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	102 Filters, may be repeated multiple times:
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	103 --ignore-module=<mod> Ignore the given module and its submodules
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	104 (if it is a package).
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	105 --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	106 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	107 """ % sys.argv[0])
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	108
28712 d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	109 PRAGMA_NOCOVER = "#pragma NO COVER"
d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	110
d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	111 # Simple rx to find lines with no code.
d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	112 rx_blank = re.compile(r'^\s(#.)?$')
d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	113
27933 44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	114 class Ignore:
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	115 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	116 self._mods = modules or []
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	117 self._dirs = dirs or []
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	118
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	119 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	120 self._ignore = { '<string>': 1 }
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	121
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	122 def names(self, filename, modulename):
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	123 if self._ignore.has_key(modulename):
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	124 return self._ignore[modulename]
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	125
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	126 # 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	127 # 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	128 # "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	129 # "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	130 for mod in self._mods:
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	131 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	132 self._ignore[modulename] = 1
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	133 return 1
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	134 # 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	135 # the ignore list
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	136 n = len(mod)
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	137 # (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	138 # 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	139 # 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	140 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	141 self._ignore[modulename] = 1
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	142 return 1
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	143
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	144 # 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	145 if filename is None:
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	146 # 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	147 self._ignore[modulename] = 1
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	148 return 1
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	149
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	150 # 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	151 for d in self._dirs:
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	152 # 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	153 # as compared to cases like:
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	154 # d = "/usr/local"
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	155 # filename = "/usr/local.py"
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	156 # or
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	157 # d = "/usr/local.py"
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	158 # filename = "/usr/local.py"
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	159 if filename.startswith(d + os.sep):
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 # 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	164 self._ignore[modulename] = 0
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	165 return 0
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	166
28712 d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	167 def modname(path):
d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	168 """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	169
28712 d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	170 base = os.path.basename(path)
d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	171 filename, ext = os.path.splitext(base)
d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	172 return filename
d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	173
28713 d08d15a70295 Add helper function to get module name taking packages into account. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28712 diff changeset	174 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	175 """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	176
d08d15a70295 Add helper function to get module name taking packages into account. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28712 diff changeset	177 # 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	178 # 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	179 # 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	180 # 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	181
d08d15a70295 Add helper function to get module name taking packages into account. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28712 diff changeset	182 longest = ""
d08d15a70295 Add helper function to get module name taking packages into account. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28712 diff changeset	183 for dir in sys.path:
d08d15a70295 Add helper function to get module name taking packages into account. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28712 diff changeset	184 if path.startswith(dir) and path[len(dir)] == os.path.sep:
d08d15a70295 Add helper function to get module name taking packages into account. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28712 diff changeset	185 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	186 longest = dir
d08d15a70295 Add helper function to get module name taking packages into account. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28712 diff changeset	187
30667 400674958025 The fullmodname() function chopped off the first character if the Guido van Rossum <guido@python.org> parents: 29864 diff changeset	188 if longest:
400674958025 The fullmodname() function chopped off the first character if the Guido van Rossum <guido@python.org> parents: 29864 diff changeset	189 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	190 else:
400674958025 The fullmodname() function chopped off the first character if the Guido van Rossum <guido@python.org> parents: 29864 diff changeset	191 base = path
31551 a157cc20f1b7 Fix two small bugs: (1) on Windows, pathname munging didn't work Guido van Rossum <guido@python.org> parents: 31515 diff changeset	192 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	193 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	194 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	195 filename, ext = os.path.splitext(base)
d08d15a70295 Add helper function to get module name taking packages into account. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28712 diff changeset	196 return filename
d08d15a70295 Add helper function to get module name taking packages into account. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28712 diff changeset	197
27933 44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	198 class CoverageResults:
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	199 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	200 callers=None, outfile=None):
27933 44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	201 self.counts = counts
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	202 if self.counts is None:
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	203 self.counts = {}
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	204 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	205 self.calledfuncs = calledfuncs
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	206 if self.calledfuncs is None:
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	207 self.calledfuncs = {}
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	208 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	209 self.callers = callers
f35b4acc594f Added --trackcalls command line arg to display crude caller/callee Skip Montanaro <skip@pobox.com> parents: 31551 diff changeset	210 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	211 self.callers = {}
f35b4acc594f Added --trackcalls command line arg to display crude caller/callee Skip Montanaro <skip@pobox.com> parents: 31551 diff changeset	212 self.callers = self.callers.copy()
27933 44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	213 self.infile = infile
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	214 self.outfile = outfile
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	215 if self.infile:
29767 7770b7881346 Patch from Zooko to remove an experimental feature. Jeremy Hylton <jeremy@alum.mit.edu> parents: 29620 diff changeset	216 # 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	217 try:
31881 f35b4acc594f Added --trackcalls command line arg to display crude caller/callee Skip Montanaro <skip@pobox.com> parents: 31551 diff changeset	218 counts, calledfuncs, callers = \
f35b4acc594f Added --trackcalls command line arg to display crude caller/callee Skip Montanaro <skip@pobox.com> parents: 31551 diff changeset	219 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	220 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	221 except (IOError, EOFError, ValueError), err:
28712 d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	222 print >> sys.stderr, ("Skipping counts file %r: %s"
d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	223 % (self.infile, err))
27933 44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	224
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	225 def update(self, other):
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	226 """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	227 counts = self.counts
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	228 calledfuncs = self.calledfuncs
31881 f35b4acc594f Added --trackcalls command line arg to display crude caller/callee Skip Montanaro <skip@pobox.com> parents: 31551 diff changeset	229 callers = self.callers
27933 44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	230 other_counts = other.counts
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	231 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	232 other_callers = other.callers
27933 44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	233
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	234 for key in other_counts.keys():
28712 d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	235 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	236
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	237 for key in other_calledfuncs.keys():
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	238 calledfuncs[key] = 1
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	239
31881 f35b4acc594f Added --trackcalls command line arg to display crude caller/callee Skip Montanaro <skip@pobox.com> parents: 31551 diff changeset	240 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	241 callers[key] = 1
f35b4acc594f Added --trackcalls command line arg to display crude caller/callee Skip Montanaro <skip@pobox.com> parents: 31551 diff changeset	242
28712 d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	243 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	244 """
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	245 @param coverdir
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	246 """
31881 f35b4acc594f Added --trackcalls command line arg to display crude caller/callee Skip Montanaro <skip@pobox.com> parents: 31551 diff changeset	247 if self.calledfuncs:
31892 081972ae2f84 Refactor common code out of globaltrace_trackcallers() and Skip Montanaro <skip@pobox.com> parents: 31881 diff changeset	248 print
31881 f35b4acc594f Added --trackcalls command line arg to display crude caller/callee Skip Montanaro <skip@pobox.com> parents: 31551 diff changeset	249 print "functions called:"
f35b4acc594f Added --trackcalls command line arg to display crude caller/callee Skip Montanaro <skip@pobox.com> parents: 31551 diff changeset	250 calls = self.calledfuncs.keys()
f35b4acc594f Added --trackcalls command line arg to display crude caller/callee Skip Montanaro <skip@pobox.com> parents: 31551 diff changeset	251 calls.sort()
f35b4acc594f Added --trackcalls command line arg to display crude caller/callee Skip Montanaro <skip@pobox.com> parents: 31551 diff changeset	252 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	253 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	254 % (filename, modulename, funcname))
f35b4acc594f Added --trackcalls command line arg to display crude caller/callee Skip Montanaro <skip@pobox.com> parents: 31551 diff changeset	255
f35b4acc594f Added --trackcalls command line arg to display crude caller/callee Skip Montanaro <skip@pobox.com> parents: 31551 diff changeset	256 if self.callers:
31892 081972ae2f84 Refactor common code out of globaltrace_trackcallers() and Skip Montanaro <skip@pobox.com> parents: 31881 diff changeset	257 print
31881 f35b4acc594f Added --trackcalls command line arg to display crude caller/callee Skip Montanaro <skip@pobox.com> parents: 31551 diff changeset	258 print "calling relationships:"
f35b4acc594f Added --trackcalls command line arg to display crude caller/callee Skip Montanaro <skip@pobox.com> parents: 31551 diff changeset	259 calls = self.callers.keys()
f35b4acc594f Added --trackcalls command line arg to display crude caller/callee Skip Montanaro <skip@pobox.com> parents: 31551 diff changeset	260 calls.sort()
f35b4acc594f Added --trackcalls command line arg to display crude caller/callee Skip Montanaro <skip@pobox.com> parents: 31551 diff changeset	261 lastfile = lastcfile = ""
f35b4acc594f Added --trackcalls command line arg to display crude caller/callee Skip Montanaro <skip@pobox.com> parents: 31551 diff changeset	262 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	263 if pfile != lastfile:
31892 081972ae2f84 Refactor common code out of globaltrace_trackcallers() and Skip Montanaro <skip@pobox.com> parents: 31881 diff changeset	264 print
31881 f35b4acc594f Added --trackcalls command line arg to display crude caller/callee Skip Montanaro <skip@pobox.com> parents: 31551 diff changeset	265 print "*", pfile, "*"
f35b4acc594f Added --trackcalls command line arg to display crude caller/callee Skip Montanaro <skip@pobox.com> parents: 31551 diff changeset	266 lastfile = pfile
f35b4acc594f Added --trackcalls command line arg to display crude caller/callee Skip Montanaro <skip@pobox.com> parents: 31551 diff changeset	267 lastcfile = ""
f35b4acc594f Added --trackcalls command line arg to display crude caller/callee Skip Montanaro <skip@pobox.com> parents: 31551 diff changeset	268 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	269 print " -->", cfile
f35b4acc594f Added --trackcalls command line arg to display crude caller/callee Skip Montanaro <skip@pobox.com> parents: 31551 diff changeset	270 lastcfile = cfile
f35b4acc594f Added --trackcalls command line arg to display crude caller/callee Skip Montanaro <skip@pobox.com> parents: 31551 diff changeset	271 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	272
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	273 # 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	274 # accessible on a per-file basis
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	275 per_file = {}
28712 d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	276 for filename, lineno in self.counts.keys():
d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	277 lines_hit = per_file[filename] = per_file.get(filename, {})
d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	278 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	279
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	280 # accumulate summary info, if needed
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	281 sums = {}
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	282
28712 d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	283 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	284 # 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	285 if filename == "<string>":
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	286 continue
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	287
38704 ba6eebf7c719 Make use of new str.startswith/endswith semantics. Georg Brandl <georg@python.org> parents: 37964 diff changeset	288 if filename.endswith((".pyc", ".pyo")):
27933 44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	289 filename = filename[:-1]
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	290
28712 d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	291 if coverdir is None:
d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	292 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	293 modulename = modname(filename)
27933 44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	294 else:
28712 d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	295 dir = coverdir
d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	296 if not os.path.exists(dir):
d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	297 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	298 modulename = fullmodname(filename)
27933 44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	299
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	300 # 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	301 # 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	302 if show_missing:
28712 d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	303 lnotab = find_executable_linenos(filename)
27933 44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	304 else:
28712 d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	305 lnotab = {}
27933 44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	306
28712 d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	307 source = linecache.getlines(filename)
d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	308 coverpath = os.path.join(dir, modulename + ".cover")
d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	309 n_hits, n_lines = self.write_results_file(coverpath, source,
d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	310 lnotab, count)
28774 c9bee11eaa8b Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 28727 diff changeset	311
27933 44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	312 if summary and n_lines:
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	313 percent = int(100 * n_hits / n_lines)
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	314 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	315
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	316 if summary and sums:
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	317 mods = sums.keys()
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	318 mods.sort()
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	319 print "lines cov% module (path)"
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	320 for m in mods:
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	321 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	322 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	323
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	324 if self.outfile:
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	325 # 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	326 try:
31881 f35b4acc594f Added --trackcalls command line arg to display crude caller/callee Skip Montanaro <skip@pobox.com> parents: 31551 diff changeset	327 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	328 open(self.outfile, 'wb'), 1)
27933 44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	329 except IOError, err:
28712 d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	330 print >> sys.stderr, "Can't save counts files because %s" % err
d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	331
d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	332 def write_results_file(self, path, lines, lnotab, lines_hit):
d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	333 """Return a coverage results file in path."""
d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	334
d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	335 try:
d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	336 outfile = open(path, "w")
d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	337 except IOError, err:
d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	338 print >> sys.stderr, ("trace: Could not open %r for writing: %s"
d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	339 "- 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	340 return 0, 0
27933 44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	341
28712 d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	342 n_lines = 0
d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	343 n_hits = 0
d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	344 for i, line in enumerate(lines):
d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	345 lineno = i + 1
d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	346 # do the blank/comment match to try to mark more lines
d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	347 # (help the reader find stuff that hasn't been covered)
d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	348 if lineno in lines_hit:
d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	349 outfile.write("%5d: " % lines_hit[lineno])
d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	350 n_hits += 1
d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	351 n_lines += 1
d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	352 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	353 outfile.write(" ")
28712 d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	354 else:
d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	355 # lines preceded by no marks weren't hit
d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	356 # Highlight them if so indicated, unless the line contains
d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	357 # #pragma: NO COVER
d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	358 if lineno in lnotab and not PRAGMA_NOCOVER in lines[i]:
d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	359 outfile.write(">>>>>> ")
29595 8911ca88981a Enable tracing of multi-threaded applications. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28774 diff changeset	360 n_lines += 1
28712 d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	361 else:
d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	362 outfile.write(" ")
d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	363 outfile.write(lines[i].expandtabs(8))
d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	364 outfile.close()
d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	365
d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	366 return n_hits, n_lines
d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	367
d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	368 def find_lines_from_code(code, strs):
d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	369 """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	370 linenos = {}
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	371
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	372 line_increments = [ord(c) for c in code.co_lnotab[1::2]]
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	373 table_length = len(line_increments)
28712 d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	374 docstring = False
27933 44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	375
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	376 lineno = code.co_firstlineno
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	377 for li in line_increments:
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	378 lineno += li
28712 d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	379 if lineno not in strs:
d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	380 linenos[lineno] = 1
27933 44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	381
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	382 return linenos
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	383
28712 d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	384 def find_lines(code, strs):
d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	385 """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	386 # 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	387 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	388
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	389 # 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	390 for c in code.co_consts:
28712 d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	391 if isinstance(c, types.CodeType):
27933 44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	392 # find another code object, so recurse into it
28712 d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	393 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	394 return linenos
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	395
28712 d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	396 def find_strings(filename):
d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	397 """Return a dict of possible docstring positions.
d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	398
d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	399 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	400 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	401 string.
d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	402 """
d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	403 d = {}
d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	404 # 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	405 # 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	406 prev_ttype = token.INDENT
d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	407 f = open(filename)
d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	408 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	409 if ttype == token.STRING:
d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	410 if prev_ttype == token.INDENT:
d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	411 sline, scol = start
d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	412 eline, ecol = end
d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	413 for i in range(sline, eline + 1):
d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	414 d[i] = 1
d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	415 prev_ttype = ttype
d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	416 f.close()
d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	417 return d
d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	418
27933 44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	419 def find_executable_linenos(filename):
28712 d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	420 """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	421 try:
31919 39c36526d1b0 Open file in universal newline mode when passing to compile(). Solution Skip Montanaro <skip@pobox.com> parents: 31892 diff changeset	422 prog = open(filename, "rU").read()
28712 d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	423 except IOError, err:
d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	424 print >> sys.stderr, ("Not printing coverage data for %r: %s"
d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	425 % (filename, err))
d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	426 return {}
d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	427 code = compile(prog, filename, "exec")
d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	428 strs = find_strings(filename)
d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	429 return find_lines(code, strs)
27933 44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	430
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	431 class Trace:
31881 f35b4acc594f Added --trackcalls command line arg to display crude caller/callee Skip Montanaro <skip@pobox.com> parents: 31551 diff changeset	432 def __init__(self, count=1, trace=1, countfuncs=0, countcallers=0,
f35b4acc594f Added --trackcalls command line arg to display crude caller/callee Skip Montanaro <skip@pobox.com> parents: 31551 diff changeset	433 ignoremods=(), ignoredirs=(), infile=None, outfile=None):
27933 44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	434 """
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	435 @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	436 line is executed
27933 44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	437 @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	438 being counted
27933 44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	439 @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	440 (filename, modulename, funcname,) for functions
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	441 that were called at least once; This overrides
27963 226463445697 Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 27933 diff changeset	442 `count' and `trace'
27933 44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	443 @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	444 @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	445 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	446 @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	447 added into the results
27933 44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	448 @param outfile file in which to write the results
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 self.infile = infile
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	451 self.outfile = outfile
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	452 self.ignore = Ignore(ignoremods, ignoredirs)
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	453 self.counts = {} # keys are (filename, linenumber)
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	454 self.blabbed = {} # for debugging
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	455 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	456 self.donothing = 0
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	457 self.trace = trace
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	458 self._calledfuncs = {}
31881 f35b4acc594f Added --trackcalls command line arg to display crude caller/callee Skip Montanaro <skip@pobox.com> parents: 31551 diff changeset	459 self._callers = {}
31892 081972ae2f84 Refactor common code out of globaltrace_trackcallers() and Skip Montanaro <skip@pobox.com> parents: 31881 diff changeset	460 self._caller_cache = {}
31881 f35b4acc594f Added --trackcalls command line arg to display crude caller/callee Skip Montanaro <skip@pobox.com> parents: 31551 diff changeset	461 if countcallers:
f35b4acc594f Added --trackcalls command line arg to display crude caller/callee Skip Montanaro <skip@pobox.com> parents: 31551 diff changeset	462 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	463 elif countfuncs:
27933 44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	464 self.globaltrace = self.globaltrace_countfuncs
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	465 elif trace and count:
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	466 self.globaltrace = self.globaltrace_lt
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	467 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	468 elif trace:
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	469 self.globaltrace = self.globaltrace_lt
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	470 self.localtrace = self.localtrace_trace
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	471 elif count:
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	472 self.globaltrace = self.globaltrace_lt
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	473 self.localtrace = self.localtrace_count
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	474 else:
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	475 # Ahem -- do nothing? Okay.
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	476 self.donothing = 1
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	477
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	478 def run(self, cmd):
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	479 import __main__
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	480 dict = __main__.__dict__
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	481 if not self.donothing:
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	482 sys.settrace(self.globaltrace)
29595 8911ca88981a Enable tracing of multi-threaded applications. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28774 diff changeset	483 threading.settrace(self.globaltrace)
27933 44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	484 try:
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	485 exec cmd in dict, dict
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	486 finally:
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	487 if not self.donothing:
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	488 sys.settrace(None)
29595 8911ca88981a Enable tracing of multi-threaded applications. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28774 diff changeset	489 threading.settrace(None)
27933 44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	490
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	491 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	492 if globals is None: globals = {}
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	493 if locals is None: locals = {}
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	494 if not self.donothing:
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	495 sys.settrace(self.globaltrace)
29595 8911ca88981a Enable tracing of multi-threaded applications. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28774 diff changeset	496 threading.settrace(self.globaltrace)
27933 44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	497 try:
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	498 exec cmd in globals, locals
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	499 finally:
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	500 if not self.donothing:
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	501 sys.settrace(None)
29595 8911ca88981a Enable tracing of multi-threaded applications. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28774 diff changeset	502 threading.settrace(None)
27933 44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	503
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	504 def runfunc(self, func, args, *kw):
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	505 result = None
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	506 if not self.donothing:
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	507 sys.settrace(self.globaltrace)
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	508 try:
28089 14f65d661084 Get rid of many apply() calls. Guido van Rossum <guido@python.org> parents: 27963 diff changeset	509 result = func(args, *kw)
27933 44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	510 finally:
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	511 if not self.donothing:
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	512 sys.settrace(None)
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	513 return result
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	514
31892 081972ae2f84 Refactor common code out of globaltrace_trackcallers() and Skip Montanaro <skip@pobox.com> parents: 31881 diff changeset	515 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	516 code = frame.f_code
081972ae2f84 Refactor common code out of globaltrace_trackcallers() and Skip Montanaro <skip@pobox.com> parents: 31881 diff changeset	517 filename = code.co_filename
081972ae2f84 Refactor common code out of globaltrace_trackcallers() and Skip Montanaro <skip@pobox.com> parents: 31881 diff changeset	518 if filename:
081972ae2f84 Refactor common code out of globaltrace_trackcallers() and Skip Montanaro <skip@pobox.com> parents: 31881 diff changeset	519 modulename = modname(filename)
081972ae2f84 Refactor common code out of globaltrace_trackcallers() and Skip Montanaro <skip@pobox.com> parents: 31881 diff changeset	520 else:
081972ae2f84 Refactor common code out of globaltrace_trackcallers() and Skip Montanaro <skip@pobox.com> parents: 31881 diff changeset	521 modulename = None
081972ae2f84 Refactor common code out of globaltrace_trackcallers() and Skip Montanaro <skip@pobox.com> parents: 31881 diff changeset	522
081972ae2f84 Refactor common code out of globaltrace_trackcallers() and Skip Montanaro <skip@pobox.com> parents: 31881 diff changeset	523 funcname = code.co_name
081972ae2f84 Refactor common code out of globaltrace_trackcallers() and Skip Montanaro <skip@pobox.com> parents: 31881 diff changeset	524 clsname = None
081972ae2f84 Refactor common code out of globaltrace_trackcallers() and Skip Montanaro <skip@pobox.com> parents: 31881 diff changeset	525 if code in self._caller_cache:
081972ae2f84 Refactor common code out of globaltrace_trackcallers() and Skip Montanaro <skip@pobox.com> parents: 31881 diff changeset	526 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	527 clsname = self._caller_cache[code]
081972ae2f84 Refactor common code out of globaltrace_trackcallers() and Skip Montanaro <skip@pobox.com> parents: 31881 diff changeset	528 else:
081972ae2f84 Refactor common code out of globaltrace_trackcallers() and Skip Montanaro <skip@pobox.com> parents: 31881 diff changeset	529 self._caller_cache[code] = None
081972ae2f84 Refactor common code out of globaltrace_trackcallers() and Skip Montanaro <skip@pobox.com> parents: 31881 diff changeset	530 ## 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	531 # 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	532 funcs = [f for f in gc.get_referrers(code)
081972ae2f84 Refactor common code out of globaltrace_trackcallers() and Skip Montanaro <skip@pobox.com> parents: 31881 diff changeset	533 if hasattr(f, "func_doc")]
081972ae2f84 Refactor common code out of globaltrace_trackcallers() and Skip Montanaro <skip@pobox.com> parents: 31881 diff changeset	534 # 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	535 # 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	536 # temptation to guess."
081972ae2f84 Refactor common code out of globaltrace_trackcallers() and Skip Montanaro <skip@pobox.com> parents: 31881 diff changeset	537 if len(funcs) == 1:
081972ae2f84 Refactor common code out of globaltrace_trackcallers() and Skip Montanaro <skip@pobox.com> parents: 31881 diff changeset	538 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	539 if isinstance(d, dict)]
081972ae2f84 Refactor common code out of globaltrace_trackcallers() and Skip Montanaro <skip@pobox.com> parents: 31881 diff changeset	540 if len(dicts) == 1:
081972ae2f84 Refactor common code out of globaltrace_trackcallers() and Skip Montanaro <skip@pobox.com> parents: 31881 diff changeset	541 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	542 if hasattr(c, "__bases__")]
081972ae2f84 Refactor common code out of globaltrace_trackcallers() and Skip Montanaro <skip@pobox.com> parents: 31881 diff changeset	543 if len(classes) == 1:
081972ae2f84 Refactor common code out of globaltrace_trackcallers() and Skip Montanaro <skip@pobox.com> parents: 31881 diff changeset	544 # ditto for new.classobj()
081972ae2f84 Refactor common code out of globaltrace_trackcallers() and Skip Montanaro <skip@pobox.com> parents: 31881 diff changeset	545 clsname = str(classes[0])
081972ae2f84 Refactor common code out of globaltrace_trackcallers() and Skip Montanaro <skip@pobox.com> parents: 31881 diff changeset	546 # 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	547 # 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	548 # _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	549 # the new module get called.
081972ae2f84 Refactor common code out of globaltrace_trackcallers() and Skip Montanaro <skip@pobox.com> parents: 31881 diff changeset	550 self._caller_cache[code] = clsname
081972ae2f84 Refactor common code out of globaltrace_trackcallers() and Skip Montanaro <skip@pobox.com> parents: 31881 diff changeset	551 if clsname is not None:
081972ae2f84 Refactor common code out of globaltrace_trackcallers() and Skip Montanaro <skip@pobox.com> parents: 31881 diff changeset	552 # final hack - module name shows up in str(cls), but we've already
081972ae2f84 Refactor common code out of globaltrace_trackcallers() and Skip Montanaro <skip@pobox.com> parents: 31881 diff changeset	553 # computed module name, so remove it
081972ae2f84 Refactor common code out of globaltrace_trackcallers() and Skip Montanaro <skip@pobox.com> parents: 31881 diff changeset	554 clsname = clsname.split(".")[1:]
081972ae2f84 Refactor common code out of globaltrace_trackcallers() and Skip Montanaro <skip@pobox.com> parents: 31881 diff changeset	555 clsname = ".".join(clsname)
081972ae2f84 Refactor common code out of globaltrace_trackcallers() and Skip Montanaro <skip@pobox.com> parents: 31881 diff changeset	556 funcname = "%s.%s" % (clsname, funcname)
081972ae2f84 Refactor common code out of globaltrace_trackcallers() and Skip Montanaro <skip@pobox.com> parents: 31881 diff changeset	557
081972ae2f84 Refactor common code out of globaltrace_trackcallers() and Skip Montanaro <skip@pobox.com> parents: 31881 diff changeset	558 return filename, modulename, funcname
081972ae2f84 Refactor common code out of globaltrace_trackcallers() and Skip Montanaro <skip@pobox.com> parents: 31881 diff changeset	559
31881 f35b4acc594f Added --trackcalls command line arg to display crude caller/callee Skip Montanaro <skip@pobox.com> parents: 31551 diff changeset	560 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	561 """Handler for call events.
f35b4acc594f Added --trackcalls command line arg to display crude caller/callee Skip Montanaro <skip@pobox.com> parents: 31551 diff changeset	562
f35b4acc594f Added --trackcalls command line arg to display crude caller/callee Skip Montanaro <skip@pobox.com> parents: 31551 diff changeset	563 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	564 """
f35b4acc594f Added --trackcalls command line arg to display crude caller/callee Skip Montanaro <skip@pobox.com> parents: 31551 diff changeset	565 if why == 'call':
f35b4acc594f Added --trackcalls command line arg to display crude caller/callee Skip Montanaro <skip@pobox.com> parents: 31551 diff changeset	566 # 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	567 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	568 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	569 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	570
27933 44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	571 def globaltrace_countfuncs(self, frame, why, arg):
28712 d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	572 """Handler for call events.
28774 c9bee11eaa8b Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 28727 diff changeset	573
28712 d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	574 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	575 """
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	576 if why == 'call':
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 self._calledfuncs[this_func] = 1
27933 44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	579
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	580 def globaltrace_lt(self, frame, why, arg):
28712 d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	581 """Handler for call events.
d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	582
d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	583 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	584 else returns self.localtrace.
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':
28712 d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	587 code = frame.f_code
d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	588 filename = code.co_filename
27933 44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	589 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	590 # 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	591 # the ignore support won't work right for packages
28712 d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	592 modulename = modname(filename)
27933 44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	593 if modulename is not None:
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	594 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	595 if not ignore_it:
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	596 if self.trace:
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	597 print (" --- modulename: %s, funcname: %s"
28712 d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	598 % (modulename, code.co_name))
27933 44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	599 return self.localtrace
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	600 else:
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	601 return None
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	602
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	603 def localtrace_trace_and_count(self, frame, why, arg):
28712 d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	604 if why == "line":
27933 44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	605 # record the file name and line number of every trace
28712 d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	606 filename = frame.f_code.co_filename
d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	607 lineno = frame.f_lineno
27933 44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	608 key = filename, lineno
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	609 self.counts[key] = self.counts.get(key, 0) + 1
27963 226463445697 Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 27933 diff changeset	610
28712 d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	611 bname = os.path.basename(filename)
d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	612 print "%s(%d): %s" % (bname, lineno,
d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	613 linecache.getline(filename, lineno)),
27933 44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	614 return self.localtrace
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	615
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	616 def localtrace_trace(self, frame, why, arg):
28712 d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	617 if why == "line":
d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	618 # record the file name and line number of every trace
d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	619 filename = frame.f_code.co_filename
d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	620 lineno = frame.f_lineno
27963 226463445697 Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 27933 diff changeset	621
28712 d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	622 bname = os.path.basename(filename)
d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	623 print "%s(%d): %s" % (bname, lineno,
d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	624 linecache.getline(filename, lineno)),
27933 44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	625 return self.localtrace
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	626
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	627 def localtrace_count(self, frame, why, arg):
28712 d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	628 if why == "line":
27933 44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	629 filename = frame.f_code.co_filename
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	630 lineno = frame.f_lineno
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	631 key = filename, lineno
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	632 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	633 return self.localtrace
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	634
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	635 def results(self):
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	636 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	637 outfile=self.outfile,
31881 f35b4acc594f Added --trackcalls command line arg to display crude caller/callee Skip Montanaro <skip@pobox.com> parents: 31551 diff changeset	638 calledfuncs=self._calledfuncs,
f35b4acc594f Added --trackcalls command line arg to display crude caller/callee Skip Montanaro <skip@pobox.com> parents: 31551 diff changeset	639 callers=self._callers)
27933 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 _err_exit(msg):
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	642 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	643 sys.exit(1)
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	644
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	645 def main(argv=None):
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	646 import getopt
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	647
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	648 if argv is None:
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	649 argv = sys.argv
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	650 try:
31881 f35b4acc594f Added --trackcalls command line arg to display crude caller/callee Skip Montanaro <skip@pobox.com> parents: 31551 diff changeset	651 opts, prog_argv = getopt.getopt(argv[1:], "tcrRf:d:msC:lT",
27933 44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	652 ["help", "version", "trace", "count",
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	653 "report", "no-report", "summary",
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	654 "file=", "missing",
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	655 "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	656 "coverdir=", "listfuncs",
f35b4acc594f Added --trackcalls command line arg to display crude caller/callee Skip Montanaro <skip@pobox.com> parents: 31551 diff changeset	657 "trackcalls"])
27933 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 except getopt.error, msg:
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	660 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	661 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	662 % sys.argv[0])
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	663 sys.exit(1)
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	664
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	665 trace = 0
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	666 count = 0
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	667 report = 0
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	668 no_report = 0
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	669 counts_file = None
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	670 missing = 0
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	671 ignore_modules = []
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	672 ignore_dirs = []
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	673 coverdir = None
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	674 summary = 0
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	675 listfuncs = False
31881 f35b4acc594f Added --trackcalls command line arg to display crude caller/callee Skip Montanaro <skip@pobox.com> parents: 31551 diff changeset	676 countcallers = False
27933 44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	677
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	678 for opt, val in opts:
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	679 if opt == "--help":
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	680 usage(sys.stdout)
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	681 sys.exit(0)
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	682
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	683 if opt == "--version":
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	684 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	685 sys.exit(0)
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	686
31881 f35b4acc594f Added --trackcalls command line arg to display crude caller/callee Skip Montanaro <skip@pobox.com> parents: 31551 diff changeset	687 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	688 countcallers = True
f35b4acc594f Added --trackcalls command line arg to display crude caller/callee Skip Montanaro <skip@pobox.com> parents: 31551 diff changeset	689 continue
f35b4acc594f Added --trackcalls command line arg to display crude caller/callee Skip Montanaro <skip@pobox.com> parents: 31551 diff changeset	690
27933 44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	691 if opt == "-l" or opt == "--listfuncs":
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	692 listfuncs = True
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	693 continue
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	694
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	695 if opt == "-t" or opt == "--trace":
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	696 trace = 1
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	697 continue
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	698
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	699 if opt == "-c" or opt == "--count":
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	700 count = 1
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	701 continue
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	702
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	703 if opt == "-r" or opt == "--report":
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	704 report = 1
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	705 continue
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	706
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	707 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	708 no_report = 1
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	709 continue
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	710
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	711 if opt == "-f" or opt == "--file":
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	712 counts_file = val
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	713 continue
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	714
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	715 if opt == "-m" or opt == "--missing":
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	716 missing = 1
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	717 continue
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	718
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	719 if opt == "-C" or opt == "--coverdir":
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	720 coverdir = val
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	721 continue
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	722
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	723 if opt == "-s" or opt == "--summary":
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	724 summary = 1
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	725 continue
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	726
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	727 if opt == "--ignore-module":
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	728 ignore_modules.append(val)
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	729 continue
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	730
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	731 if opt == "--ignore-dir":
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	732 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	733 s = os.path.expandvars(s)
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	734 # 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	735
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	736 s = s.replace("$prefix",
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	737 os.path.join(sys.prefix, "lib",
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	738 "python" + sys.version[:3]))
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	739 s = s.replace("$exec_prefix",
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	740 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	741 "python" + sys.version[:3]))
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	742 s = os.path.normpath(s)
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	743 ignore_dirs.append(s)
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 assert 0, "Should never get here"
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	747
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	748 if listfuncs and (count or trace):
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	749 _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	750
31881 f35b4acc594f Added --trackcalls command line arg to display crude caller/callee Skip Montanaro <skip@pobox.com> parents: 31551 diff changeset	751 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	752 _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	753 "--listfuncs, or --trackcalls")
27933 44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	754
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	755 if report and no_report:
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	756 _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	757
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	758 if report and not counts_file:
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	759 _err_exit("--report requires a --file")
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	760
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	761 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	762 _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	763
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	764 # everything is ready
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	765 if report:
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	766 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	767 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	768 else:
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	769 sys.argv = prog_argv
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	770 progname = prog_argv[0]
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	771 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	772
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	773 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	774 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	775 ignoredirs=ignore_dirs, infile=counts_file,
f35b4acc594f Added --trackcalls command line arg to display crude caller/callee Skip Montanaro <skip@pobox.com> parents: 31551 diff changeset	776 outfile=counts_file)
27933 44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	777 try:
31515 71a294bd7950 Replace backticks with repr() or "%r" Walter Dörwald <walter@livinglogic.de> parents: 30838 diff changeset	778 t.run('execfile(%r)' % (progname,))
27933 44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	779 except IOError, err:
28712 d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	780 _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	781 except SystemExit:
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	782 pass
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 results = t.results()
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	785
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	786 if not no_report:
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
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	789 if __name__=='__main__':
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	790 main()

Mercurial > cpython-lru_cache-weakref

annotate Lib/trace.py @ 39042:c02fa187e0f4 legacy-trunk