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

annotate Lib/trace.py @ 58765:2ba7677bfa95 legacy-trunk

testCopysign was defined twice in test_math; combine the definitions

author	Mark Dickinson <dickinsm@gmail.com>
date	Sat, 06 Feb 2010 23:11:25 +0000
parents	684960fb1cbf
children	84363c747c21

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
d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	59 import types
31892 081972ae2f84 Refactor common code out of globaltrace_trackcallers() and Skip Montanaro <skip@pobox.com> parents: 31881 diff changeset	60 import gc
28712 d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	61
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
31551 a157cc20f1b7 Fix two small bugs: (1) on Windows, pathname munging didn't work Guido van Rossum <guido@python.org> parents: 31515 diff changeset	198 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	199 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	200 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	201 filename, ext = os.path.splitext(base)
d08d15a70295 Add helper function to get module name taking packages into account. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28712 diff changeset	202 return filename
d08d15a70295 Add helper function to get module name taking packages into account. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28712 diff changeset	203
27933 44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	204 class CoverageResults:
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	205 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	206 callers=None, outfile=None):
27933 44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	207 self.counts = counts
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	208 if self.counts is None:
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	209 self.counts = {}
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	210 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	211 self.calledfuncs = calledfuncs
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	212 if self.calledfuncs is None:
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	213 self.calledfuncs = {}
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	214 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	215 self.callers = callers
f35b4acc594f Added --trackcalls command line arg to display crude caller/callee Skip Montanaro <skip@pobox.com> parents: 31551 diff changeset	216 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	217 self.callers = {}
f35b4acc594f Added --trackcalls command line arg to display crude caller/callee Skip Montanaro <skip@pobox.com> parents: 31551 diff changeset	218 self.callers = self.callers.copy()
27933 44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	219 self.infile = infile
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	220 self.outfile = outfile
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	221 if self.infile:
29767 7770b7881346 Patch from Zooko to remove an experimental feature. Jeremy Hylton <jeremy@alum.mit.edu> parents: 29620 diff changeset	222 # 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	223 try:
31881 f35b4acc594f Added --trackcalls command line arg to display crude caller/callee Skip Montanaro <skip@pobox.com> parents: 31551 diff changeset	224 counts, calledfuncs, callers = \
f35b4acc594f Added --trackcalls command line arg to display crude caller/callee Skip Montanaro <skip@pobox.com> parents: 31551 diff changeset	225 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	226 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	227 except (IOError, EOFError, ValueError), err:
28712 d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	228 print >> sys.stderr, ("Skipping counts file %r: %s"
d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	229 % (self.infile, err))
27933 44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	230
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	231 def update(self, other):
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	232 """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	233 counts = self.counts
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	234 calledfuncs = self.calledfuncs
31881 f35b4acc594f Added --trackcalls command line arg to display crude caller/callee Skip Montanaro <skip@pobox.com> parents: 31551 diff changeset	235 callers = self.callers
27933 44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	236 other_counts = other.counts
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	237 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	238 other_callers = other.callers
27933 44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	239
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	240 for key in other_counts.keys():
28712 d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	241 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	242
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	243 for key in other_calledfuncs.keys():
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	244 calledfuncs[key] = 1
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	245
31881 f35b4acc594f Added --trackcalls command line arg to display crude caller/callee Skip Montanaro <skip@pobox.com> parents: 31551 diff changeset	246 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	247 callers[key] = 1
f35b4acc594f Added --trackcalls command line arg to display crude caller/callee Skip Montanaro <skip@pobox.com> parents: 31551 diff changeset	248
28712 d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	249 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	250 """
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	251 @param coverdir
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	252 """
31881 f35b4acc594f Added --trackcalls command line arg to display crude caller/callee Skip Montanaro <skip@pobox.com> parents: 31551 diff changeset	253 if self.calledfuncs:
31892 081972ae2f84 Refactor common code out of globaltrace_trackcallers() and Skip Montanaro <skip@pobox.com> parents: 31881 diff changeset	254 print
31881 f35b4acc594f Added --trackcalls command line arg to display crude caller/callee Skip Montanaro <skip@pobox.com> parents: 31551 diff changeset	255 print "functions called:"
f35b4acc594f Added --trackcalls command line arg to display crude caller/callee Skip Montanaro <skip@pobox.com> parents: 31551 diff changeset	256 calls = self.calledfuncs.keys()
f35b4acc594f Added --trackcalls command line arg to display crude caller/callee Skip Montanaro <skip@pobox.com> parents: 31551 diff changeset	257 calls.sort()
f35b4acc594f Added --trackcalls command line arg to display crude caller/callee Skip Montanaro <skip@pobox.com> parents: 31551 diff changeset	258 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	259 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	260 % (filename, modulename, funcname))
f35b4acc594f Added --trackcalls command line arg to display crude caller/callee Skip Montanaro <skip@pobox.com> parents: 31551 diff changeset	261
f35b4acc594f Added --trackcalls command line arg to display crude caller/callee Skip Montanaro <skip@pobox.com> parents: 31551 diff changeset	262 if self.callers:
31892 081972ae2f84 Refactor common code out of globaltrace_trackcallers() and Skip Montanaro <skip@pobox.com> parents: 31881 diff changeset	263 print
31881 f35b4acc594f Added --trackcalls command line arg to display crude caller/callee Skip Montanaro <skip@pobox.com> parents: 31551 diff changeset	264 print "calling relationships:"
f35b4acc594f Added --trackcalls command line arg to display crude caller/callee Skip Montanaro <skip@pobox.com> parents: 31551 diff changeset	265 calls = self.callers.keys()
f35b4acc594f Added --trackcalls command line arg to display crude caller/callee Skip Montanaro <skip@pobox.com> parents: 31551 diff changeset	266 calls.sort()
f35b4acc594f Added --trackcalls command line arg to display crude caller/callee Skip Montanaro <skip@pobox.com> parents: 31551 diff changeset	267 lastfile = lastcfile = ""
f35b4acc594f Added --trackcalls command line arg to display crude caller/callee Skip Montanaro <skip@pobox.com> parents: 31551 diff changeset	268 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	269 if pfile != lastfile:
31892 081972ae2f84 Refactor common code out of globaltrace_trackcallers() and Skip Montanaro <skip@pobox.com> parents: 31881 diff changeset	270 print
31881 f35b4acc594f Added --trackcalls command line arg to display crude caller/callee Skip Montanaro <skip@pobox.com> parents: 31551 diff changeset	271 print "*", pfile, "*"
f35b4acc594f Added --trackcalls command line arg to display crude caller/callee Skip Montanaro <skip@pobox.com> parents: 31551 diff changeset	272 lastfile = pfile
f35b4acc594f Added --trackcalls command line arg to display crude caller/callee Skip Montanaro <skip@pobox.com> parents: 31551 diff changeset	273 lastcfile = ""
f35b4acc594f Added --trackcalls command line arg to display crude caller/callee Skip Montanaro <skip@pobox.com> parents: 31551 diff changeset	274 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	275 print " -->", cfile
f35b4acc594f Added --trackcalls command line arg to display crude caller/callee Skip Montanaro <skip@pobox.com> parents: 31551 diff changeset	276 lastcfile = cfile
f35b4acc594f Added --trackcalls command line arg to display crude caller/callee Skip Montanaro <skip@pobox.com> parents: 31551 diff changeset	277 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	278
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	279 # 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	280 # accessible on a per-file basis
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	281 per_file = {}
28712 d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	282 for filename, lineno in self.counts.keys():
d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	283 lines_hit = per_file[filename] = per_file.get(filename, {})
d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	284 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	285
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	286 # accumulate summary info, if needed
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	287 sums = {}
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	288
28712 d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	289 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	290 # 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	291 if filename == "<string>":
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	292 continue
44014 e0465ba0e95d back in these go - thanks to Titus Brown for the fix Skip Montanaro <skip@pobox.com> parents: 44001 diff changeset	293 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	294 continue
27933 44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	295
38704 ba6eebf7c719 Make use of new str.startswith/endswith semantics. Georg Brandl <georg@python.org> parents: 37964 diff changeset	296 if filename.endswith((".pyc", ".pyo")):
27933 44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	297 filename = filename[:-1]
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	298
28712 d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	299 if coverdir is None:
d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	300 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	301 modulename = modname(filename)
27933 44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	302 else:
28712 d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	303 dir = coverdir
d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	304 if not os.path.exists(dir):
d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	305 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	306 modulename = fullmodname(filename)
27933 44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	307
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	308 # 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	309 # 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	310 if show_missing:
28712 d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	311 lnotab = find_executable_linenos(filename)
27933 44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	312 else:
28712 d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	313 lnotab = {}
27933 44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	314
28712 d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	315 source = linecache.getlines(filename)
d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	316 coverpath = os.path.join(dir, modulename + ".cover")
d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	317 n_hits, n_lines = self.write_results_file(coverpath, source,
d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	318 lnotab, count)
28774 c9bee11eaa8b Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 28727 diff changeset	319
27933 44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	320 if summary and n_lines:
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	321 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	322 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	323
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	324 if summary and sums:
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	325 mods = sums.keys()
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	326 mods.sort()
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	327 print "lines cov% module (path)"
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	328 for m in mods:
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	329 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	330 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	331
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	332 if self.outfile:
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	333 # 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	334 try:
31881 f35b4acc594f Added --trackcalls command line arg to display crude caller/callee Skip Montanaro <skip@pobox.com> parents: 31551 diff changeset	335 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	336 open(self.outfile, 'wb'), 1)
27933 44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	337 except IOError, err:
28712 d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	338 print >> sys.stderr, "Can't save counts files because %s" % err
d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	339
d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	340 def write_results_file(self, path, lines, lnotab, lines_hit):
d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	341 """Return a coverage results file in path."""
d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	342
d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	343 try:
d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	344 outfile = open(path, "w")
d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	345 except IOError, err:
d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	346 print >> sys.stderr, ("trace: Could not open %r for writing: %s"
d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	347 "- 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	348 return 0, 0
27933 44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	349
28712 d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	350 n_lines = 0
d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	351 n_hits = 0
d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	352 for i, line in enumerate(lines):
d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	353 lineno = i + 1
d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	354 # do the blank/comment match to try to mark more lines
d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	355 # (help the reader find stuff that hasn't been covered)
d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	356 if lineno in lines_hit:
d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	357 outfile.write("%5d: " % lines_hit[lineno])
d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	358 n_hits += 1
d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	359 n_lines += 1
d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	360 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	361 outfile.write(" ")
28712 d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	362 else:
d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	363 # lines preceded by no marks weren't hit
d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	364 # Highlight them if so indicated, unless the line contains
d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	365 # #pragma: NO COVER
d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	366 if lineno in lnotab and not PRAGMA_NOCOVER in lines[i]:
d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	367 outfile.write(">>>>>> ")
29595 8911ca88981a Enable tracing of multi-threaded applications. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28774 diff changeset	368 n_lines += 1
28712 d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	369 else:
d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	370 outfile.write(" ")
d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	371 outfile.write(lines[i].expandtabs(8))
d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	372 outfile.close()
d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	373
d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	374 return n_hits, n_lines
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 def find_lines_from_code(code, strs):
d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	377 """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	378 linenos = {}
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	379
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	380 line_increments = [ord(c) for c in code.co_lnotab[1::2]]
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	381 table_length = len(line_increments)
28712 d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	382 docstring = False
27933 44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	383
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	384 lineno = code.co_firstlineno
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	385 for li in line_increments:
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	386 lineno += li
28712 d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	387 if lineno not in strs:
d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	388 linenos[lineno] = 1
27933 44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	389
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	390 return linenos
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	391
28712 d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	392 def find_lines(code, strs):
d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	393 """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	394 # 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	395 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	396
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	397 # 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	398 for c in code.co_consts:
28712 d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	399 if isinstance(c, types.CodeType):
27933 44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	400 # find another code object, so recurse into it
28712 d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	401 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	402 return linenos
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	403
28712 d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	404 def find_strings(filename):
d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	405 """Return a dict of possible docstring positions.
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 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	408 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	409 string.
d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	410 """
d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	411 d = {}
d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	412 # 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	413 # 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	414 prev_ttype = token.INDENT
d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	415 f = open(filename)
d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	416 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	417 if ttype == token.STRING:
d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	418 if prev_ttype == token.INDENT:
d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	419 sline, scol = start
d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	420 eline, ecol = end
d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	421 for i in range(sline, eline + 1):
d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	422 d[i] = 1
d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	423 prev_ttype = ttype
d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	424 f.close()
d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	425 return d
d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	426
27933 44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	427 def find_executable_linenos(filename):
28712 d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	428 """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	429 try:
31919 39c36526d1b0 Open file in universal newline mode when passing to compile(). Solution Skip Montanaro <skip@pobox.com> parents: 31892 diff changeset	430 prog = open(filename, "rU").read()
28712 d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	431 except IOError, err:
d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	432 print >> sys.stderr, ("Not printing coverage data for %r: %s"
d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	433 % (filename, err))
d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	434 return {}
d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	435 code = compile(prog, filename, "exec")
d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	436 strs = find_strings(filename)
d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	437 return find_lines(code, strs)
27933 44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	438
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	439 class Trace:
31881 f35b4acc594f Added --trackcalls command line arg to display crude caller/callee Skip Montanaro <skip@pobox.com> parents: 31551 diff changeset	440 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	441 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	442 timing=False):
27933 44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	443 """
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	444 @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	445 line is executed
27933 44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	446 @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	447 being counted
27933 44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	448 @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	449 (filename, modulename, funcname,) for functions
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	450 that were called at least once; This overrides
27963 226463445697 Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 27933 diff changeset	451 `count' and `trace'
27933 44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	452 @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	453 @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	454 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	455 @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	456 added into the results
27933 44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	457 @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	458 @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	459 """
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	460 self.infile = infile
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	461 self.outfile = outfile
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	462 self.ignore = Ignore(ignoremods, ignoredirs)
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	463 self.counts = {} # keys are (filename, linenumber)
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	464 self.blabbed = {} # for debugging
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	465 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	466 self.donothing = 0
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	467 self.trace = trace
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	468 self._calledfuncs = {}
31881 f35b4acc594f Added --trackcalls command line arg to display crude caller/callee Skip Montanaro <skip@pobox.com> parents: 31551 diff changeset	469 self._callers = {}
31892 081972ae2f84 Refactor common code out of globaltrace_trackcallers() and Skip Montanaro <skip@pobox.com> parents: 31881 diff changeset	470 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	471 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	472 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	473 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	474 if countcallers:
f35b4acc594f Added --trackcalls command line arg to display crude caller/callee Skip Montanaro <skip@pobox.com> parents: 31551 diff changeset	475 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	476 elif countfuncs:
27933 44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	477 self.globaltrace = self.globaltrace_countfuncs
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	478 elif trace and count:
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	479 self.globaltrace = self.globaltrace_lt
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	480 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	481 elif trace:
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	482 self.globaltrace = self.globaltrace_lt
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	483 self.localtrace = self.localtrace_trace
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	484 elif count:
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	485 self.globaltrace = self.globaltrace_lt
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	486 self.localtrace = self.localtrace_count
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	487 else:
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	488 # Ahem -- do nothing? Okay.
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	489 self.donothing = 1
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	490
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	491 def run(self, cmd):
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	492 import __main__
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	493 dict = __main__.__dict__
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	494 if not self.donothing:
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	495 sys.settrace(self.globaltrace)
29595 8911ca88981a Enable tracing of multi-threaded applications. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28774 diff changeset	496 threading.settrace(self.globaltrace)
27933 44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	497 try:
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	498 exec cmd in dict, dict
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	499 finally:
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	500 if not self.donothing:
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	501 sys.settrace(None)
29595 8911ca88981a Enable tracing of multi-threaded applications. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28774 diff changeset	502 threading.settrace(None)
27933 44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	503
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	504 def 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	505 if globals is None: globals = {}
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	506 if locals is None: locals = {}
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	507 if not self.donothing:
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	508 sys.settrace(self.globaltrace)
29595 8911ca88981a Enable tracing of multi-threaded applications. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28774 diff changeset	509 threading.settrace(self.globaltrace)
27933 44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	510 try:
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	511 exec cmd in globals, locals
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	512 finally:
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	513 if not self.donothing:
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	514 sys.settrace(None)
29595 8911ca88981a Enable tracing of multi-threaded applications. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28774 diff changeset	515 threading.settrace(None)
27933 44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	516
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	517 def runfunc(self, func, args, *kw):
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	518 result = None
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	519 if not self.donothing:
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	520 sys.settrace(self.globaltrace)
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	521 try:
28089 14f65d661084 Get rid of many apply() calls. Guido van Rossum <guido@python.org> parents: 27963 diff changeset	522 result = func(args, *kw)
27933 44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	523 finally:
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	524 if not self.donothing:
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	525 sys.settrace(None)
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	526 return result
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	527
31892 081972ae2f84 Refactor common code out of globaltrace_trackcallers() and Skip Montanaro <skip@pobox.com> parents: 31881 diff changeset	528 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	529 code = frame.f_code
081972ae2f84 Refactor common code out of globaltrace_trackcallers() and Skip Montanaro <skip@pobox.com> parents: 31881 diff changeset	530 filename = code.co_filename
081972ae2f84 Refactor common code out of globaltrace_trackcallers() and Skip Montanaro <skip@pobox.com> parents: 31881 diff changeset	531 if filename:
081972ae2f84 Refactor common code out of globaltrace_trackcallers() and Skip Montanaro <skip@pobox.com> parents: 31881 diff changeset	532 modulename = modname(filename)
081972ae2f84 Refactor common code out of globaltrace_trackcallers() and Skip Montanaro <skip@pobox.com> parents: 31881 diff changeset	533 else:
081972ae2f84 Refactor common code out of globaltrace_trackcallers() and Skip Montanaro <skip@pobox.com> parents: 31881 diff changeset	534 modulename = None
081972ae2f84 Refactor common code out of globaltrace_trackcallers() and Skip Montanaro <skip@pobox.com> parents: 31881 diff changeset	535
081972ae2f84 Refactor common code out of globaltrace_trackcallers() and Skip Montanaro <skip@pobox.com> parents: 31881 diff changeset	536 funcname = code.co_name
081972ae2f84 Refactor common code out of globaltrace_trackcallers() and Skip Montanaro <skip@pobox.com> parents: 31881 diff changeset	537 clsname = None
081972ae2f84 Refactor common code out of globaltrace_trackcallers() and Skip Montanaro <skip@pobox.com> parents: 31881 diff changeset	538 if code in self._caller_cache:
081972ae2f84 Refactor common code out of globaltrace_trackcallers() and Skip Montanaro <skip@pobox.com> parents: 31881 diff changeset	539 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	540 clsname = self._caller_cache[code]
081972ae2f84 Refactor common code out of globaltrace_trackcallers() and Skip Montanaro <skip@pobox.com> parents: 31881 diff changeset	541 else:
081972ae2f84 Refactor common code out of globaltrace_trackcallers() and Skip Montanaro <skip@pobox.com> parents: 31881 diff changeset	542 self._caller_cache[code] = None
081972ae2f84 Refactor common code out of globaltrace_trackcallers() and Skip Montanaro <skip@pobox.com> parents: 31881 diff changeset	543 ## 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	544 # 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	545 funcs = [f for f in gc.get_referrers(code)
081972ae2f84 Refactor common code out of globaltrace_trackcallers() and Skip Montanaro <skip@pobox.com> parents: 31881 diff changeset	546 if hasattr(f, "func_doc")]
081972ae2f84 Refactor common code out of globaltrace_trackcallers() and Skip Montanaro <skip@pobox.com> parents: 31881 diff changeset	547 # 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	548 # 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	549 # temptation to guess."
081972ae2f84 Refactor common code out of globaltrace_trackcallers() and Skip Montanaro <skip@pobox.com> parents: 31881 diff changeset	550 if len(funcs) == 1:
081972ae2f84 Refactor common code out of globaltrace_trackcallers() and Skip Montanaro <skip@pobox.com> parents: 31881 diff changeset	551 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	552 if isinstance(d, dict)]
081972ae2f84 Refactor common code out of globaltrace_trackcallers() and Skip Montanaro <skip@pobox.com> parents: 31881 diff changeset	553 if len(dicts) == 1:
081972ae2f84 Refactor common code out of globaltrace_trackcallers() and Skip Montanaro <skip@pobox.com> parents: 31881 diff changeset	554 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	555 if hasattr(c, "__bases__")]
081972ae2f84 Refactor common code out of globaltrace_trackcallers() and Skip Montanaro <skip@pobox.com> parents: 31881 diff changeset	556 if len(classes) == 1:
081972ae2f84 Refactor common code out of globaltrace_trackcallers() and Skip Montanaro <skip@pobox.com> parents: 31881 diff changeset	557 # ditto for new.classobj()
081972ae2f84 Refactor common code out of globaltrace_trackcallers() and Skip Montanaro <skip@pobox.com> parents: 31881 diff changeset	558 clsname = str(classes[0])
081972ae2f84 Refactor common code out of globaltrace_trackcallers() and Skip Montanaro <skip@pobox.com> parents: 31881 diff changeset	559 # 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	560 # 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	561 # _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	562 # the new module get called.
081972ae2f84 Refactor common code out of globaltrace_trackcallers() and Skip Montanaro <skip@pobox.com> parents: 31881 diff changeset	563 self._caller_cache[code] = clsname
081972ae2f84 Refactor common code out of globaltrace_trackcallers() and Skip Montanaro <skip@pobox.com> parents: 31881 diff changeset	564 if clsname is not None:
081972ae2f84 Refactor common code out of globaltrace_trackcallers() and Skip Montanaro <skip@pobox.com> parents: 31881 diff changeset	565 # final hack - module name shows up in str(cls), but we've already
081972ae2f84 Refactor common code out of globaltrace_trackcallers() and Skip Montanaro <skip@pobox.com> parents: 31881 diff changeset	566 # computed module name, so remove it
081972ae2f84 Refactor common code out of globaltrace_trackcallers() and Skip Montanaro <skip@pobox.com> parents: 31881 diff changeset	567 clsname = clsname.split(".")[1:]
081972ae2f84 Refactor common code out of globaltrace_trackcallers() and Skip Montanaro <skip@pobox.com> parents: 31881 diff changeset	568 clsname = ".".join(clsname)
081972ae2f84 Refactor common code out of globaltrace_trackcallers() and Skip Montanaro <skip@pobox.com> parents: 31881 diff changeset	569 funcname = "%s.%s" % (clsname, funcname)
081972ae2f84 Refactor common code out of globaltrace_trackcallers() and Skip Montanaro <skip@pobox.com> parents: 31881 diff changeset	570
081972ae2f84 Refactor common code out of globaltrace_trackcallers() and Skip Montanaro <skip@pobox.com> parents: 31881 diff changeset	571 return filename, modulename, funcname
081972ae2f84 Refactor common code out of globaltrace_trackcallers() and Skip Montanaro <skip@pobox.com> parents: 31881 diff changeset	572
31881 f35b4acc594f Added --trackcalls command line arg to display crude caller/callee Skip Montanaro <skip@pobox.com> parents: 31551 diff changeset	573 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	574 """Handler for call events.
f35b4acc594f Added --trackcalls command line arg to display crude caller/callee Skip Montanaro <skip@pobox.com> parents: 31551 diff changeset	575
f35b4acc594f Added --trackcalls command line arg to display crude caller/callee Skip Montanaro <skip@pobox.com> parents: 31551 diff changeset	576 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	577 """
f35b4acc594f Added --trackcalls command line arg to display crude caller/callee Skip Montanaro <skip@pobox.com> parents: 31551 diff changeset	578 if why == 'call':
f35b4acc594f Added --trackcalls command line arg to display crude caller/callee Skip Montanaro <skip@pobox.com> parents: 31551 diff changeset	579 # 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	580 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	581 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	582 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	583
27933 44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	584 def globaltrace_countfuncs(self, frame, why, arg):
28712 d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	585 """Handler for call events.
28774 c9bee11eaa8b Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 28727 diff changeset	586
28712 d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	587 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	588 """
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	589 if why == 'call':
31892 081972ae2f84 Refactor common code out of globaltrace_trackcallers() and Skip Montanaro <skip@pobox.com> parents: 31881 diff changeset	590 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	591 self._calledfuncs[this_func] = 1
27933 44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	592
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	593 def globaltrace_lt(self, frame, why, arg):
28712 d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	594 """Handler for call events.
d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	595
d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	596 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	597 else returns self.localtrace.
27933 44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	598 """
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	599 if why == 'call':
28712 d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	600 code = frame.f_code
40811 5dbce2ea133c fix trace.py --ignore-dir Skip Montanaro <skip@pobox.com> parents: 39500 diff changeset	601 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	602 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	603 # 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	604 # the ignore support won't work right for packages
28712 d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	605 modulename = modname(filename)
27933 44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	606 if modulename is not None:
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	607 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	608 if not ignore_it:
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	609 if self.trace:
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	610 print (" --- modulename: %s, funcname: %s"
28712 d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	611 % (modulename, code.co_name))
27933 44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	612 return self.localtrace
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	613 else:
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	614 return None
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	615
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	616 def localtrace_trace_and_count(self, frame, why, arg):
28712 d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	617 if why == "line":
27933 44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	618 # record the file name and line number of every trace
28712 d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	619 filename = frame.f_code.co_filename
d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	620 lineno = frame.f_lineno
27933 44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	621 key = filename, lineno
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	622 self.counts[key] = self.counts.get(key, 0) + 1
27963 226463445697 Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 27933 diff changeset	623
45535 c57cb2f8c350 Add a timing flag to Trace so you can see where slowness occurs Neal Norwitz <nnorwitz@gmail.com> parents: 44775 diff changeset	624 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	625 print '%.2f' % (time.time() - self.start_time),
28712 d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	626 bname = os.path.basename(filename)
d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	627 print "%s(%d): %s" % (bname, lineno,
d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	628 linecache.getline(filename, lineno)),
27933 44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	629 return self.localtrace
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	630
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	631 def localtrace_trace(self, frame, why, arg):
28712 d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	632 if why == "line":
d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	633 # record the file name and line number of every trace
d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	634 filename = frame.f_code.co_filename
d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	635 lineno = frame.f_lineno
27963 226463445697 Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 27933 diff changeset	636
45535 c57cb2f8c350 Add a timing flag to Trace so you can see where slowness occurs Neal Norwitz <nnorwitz@gmail.com> parents: 44775 diff changeset	637 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	638 print '%.2f' % (time.time() - self.start_time),
28712 d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	639 bname = os.path.basename(filename)
d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	640 print "%s(%d): %s" % (bname, lineno,
d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	641 linecache.getline(filename, lineno)),
27933 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 localtrace_count(self, frame, why, arg):
28712 d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	645 if why == "line":
27933 44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	646 filename = frame.f_code.co_filename
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	647 lineno = frame.f_lineno
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	648 key = filename, lineno
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	649 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	650 return self.localtrace
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	651
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	652 def results(self):
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	653 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	654 outfile=self.outfile,
31881 f35b4acc594f Added --trackcalls command line arg to display crude caller/callee Skip Montanaro <skip@pobox.com> parents: 31551 diff changeset	655 calledfuncs=self._calledfuncs,
f35b4acc594f Added --trackcalls command line arg to display crude caller/callee Skip Montanaro <skip@pobox.com> parents: 31551 diff changeset	656 callers=self._callers)
27933 44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	657
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	658 def _err_exit(msg):
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	659 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	660 sys.exit(1)
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	661
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	662 def main(argv=None):
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	663 import getopt
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	664
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	665 if argv is None:
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	666 argv = sys.argv
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	667 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	668 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	669 ["help", "version", "trace", "count",
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	670 "report", "no-report", "summary",
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	671 "file=", "missing",
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	672 "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	673 "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	674 "trackcalls", "timing"])
27933 44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	675
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	676 except getopt.error, msg:
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	677 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	678 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	679 % sys.argv[0])
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	680 sys.exit(1)
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	681
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	682 trace = 0
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	683 count = 0
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	684 report = 0
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	685 no_report = 0
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	686 counts_file = None
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	687 missing = 0
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	688 ignore_modules = []
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	689 ignore_dirs = []
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	690 coverdir = None
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	691 summary = 0
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	692 listfuncs = False
31881 f35b4acc594f Added --trackcalls command line arg to display crude caller/callee Skip Montanaro <skip@pobox.com> parents: 31551 diff changeset	693 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	694 timing = False
27933 44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	695
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	696 for opt, val in opts:
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	697 if opt == "--help":
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	698 usage(sys.stdout)
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	699 sys.exit(0)
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	700
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	701 if opt == "--version":
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	702 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	703 sys.exit(0)
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	704
31881 f35b4acc594f Added --trackcalls command line arg to display crude caller/callee Skip Montanaro <skip@pobox.com> parents: 31551 diff changeset	705 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	706 countcallers = True
f35b4acc594f Added --trackcalls command line arg to display crude caller/callee Skip Montanaro <skip@pobox.com> parents: 31551 diff changeset	707 continue
f35b4acc594f Added --trackcalls command line arg to display crude caller/callee Skip Montanaro <skip@pobox.com> parents: 31551 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 == "-l" or opt == "--listfuncs":
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	710 listfuncs = True
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
45535 c57cb2f8c350 Add a timing flag to Trace so you can see where slowness occurs Neal Norwitz <nnorwitz@gmail.com> parents: 44775 diff changeset	713 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	714 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	715 continue
c57cb2f8c350 Add a timing flag to Trace so you can see where slowness occurs Neal Norwitz <nnorwitz@gmail.com> parents: 44775 diff changeset	716
27933 44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	717 if opt == "-t" or opt == "--trace":
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	718 trace = 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 == "-c" or opt == "--count":
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	722 count = 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 == "-r" or opt == "--report":
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	726 report = 1
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 == "-R" or opt == "--no-report":
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	730 no_report = 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 == "-f" or opt == "--file":
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	734 counts_file = 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 == "-m" or opt == "--missing":
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	738 missing = 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 == "-C" or opt == "--coverdir":
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	742 coverdir = val
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	743 continue
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	744
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	745 if opt == "-s" or opt == "--summary":
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	746 summary = 1
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	747 continue
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	748
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	749 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	750 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	751 ignore_modules.append(mod.strip())
27933 44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	752 continue
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	753
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	754 if opt == "--ignore-dir":
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	755 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	756 s = os.path.expandvars(s)
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	757 # 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	758
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	759 s = s.replace("$prefix",
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	760 os.path.join(sys.prefix, "lib",
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	761 "python" + sys.version[:3]))
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	762 s = s.replace("$exec_prefix",
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	763 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	764 "python" + sys.version[:3]))
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	765 s = os.path.normpath(s)
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	766 ignore_dirs.append(s)
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	767 continue
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	768
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	769 assert 0, "Should never get here"
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	770
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	771 if listfuncs and (count or trace):
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	772 _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	773
31881 f35b4acc594f Added --trackcalls command line arg to display crude caller/callee Skip Montanaro <skip@pobox.com> parents: 31551 diff changeset	774 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	775 _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	776 "--listfuncs, or --trackcalls")
27933 44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	777
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	778 if report and no_report:
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	779 _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	780
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	781 if report and not counts_file:
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	782 _err_exit("--report requires a --file")
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	783
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	784 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	785 _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	786
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	787 # everything is ready
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	788 if report:
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	789 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	790 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	791 else:
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	792 sys.argv = prog_argv
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	793 progname = prog_argv[0]
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	794 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	795
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	796 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	797 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	798 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	799 outfile=counts_file, timing=timing)
27933 44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	800 try:
31515 71a294bd7950 Replace backticks with repr() or "%r" Walter Dörwald <walter@livinglogic.de> parents: 30838 diff changeset	801 t.run('execfile(%r)' % (progname,))
27933 44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	802 except IOError, err:
28712 d3127b671d36 Holistic refactoring. Jeremy Hylton <jeremy@alum.mit.edu> parents: 28089 diff changeset	803 _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	804 except SystemExit:
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	805 pass
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	806
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	807 results = t.results()
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	808
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	809 if not no_report:
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	810 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	811
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	812 if __name__=='__main__':
44f3e315929b Copy the trace module here from Tools/scripts. Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	813 main()

Mercurial > cpython-lru_cache-weakref

annotate Lib/trace.py @ 58765:2ba7677bfa95 legacy-trunk