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

annotate Lib/trace.py @ 65035:87f4b32905df 2.7

Cosmetic fix to use print function.

author	Hirokazu Yamamoto <ocean-city@m2.ccsnet.ne.jp>
date	Fri, 01 Oct 2010 10:48:47 +0000
parents	12e0dffef7bd
children	acb0a3cc1e18

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

Mercurial > cpython-lru_cache-weakref

annotate Lib/trace.py @ 65035:87f4b32905df 2.7