annotate Lib/trace.py @ 27998:e27f58a50737 legacy-trunk

SF bug #690377: Dud footnote reference in distutils docs
author Raymond Hettinger <python@rcn.com>
date Fri, 21 Feb 2003 02:47:09 +0000
parents 226463445697
children 14f65d661084
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
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
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
35
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
36 Sample use, programmatically
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
37 # create a Trace object, telling it what to ignore, and whether to
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
38 # do tracing or line-counting or both.
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
39 trace = trace.Trace(ignoredirs=[sys.prefix, sys.exec_prefix,], trace=0,
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
40 count=1)
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
41 # run the new command using the given trace
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
42 trace.run(coverage.globaltrace, 'main()')
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
43 # make a report, telling it where you want output
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
44 r = trace.results()
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
45 r.write_results(show_missing=1)
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
46 """
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
47
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
48 import sys, os, tempfile, types, copy, operator, inspect, exceptions, marshal
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
49 try:
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
50 import cPickle
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
51 pickle = cPickle
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
52 except ImportError:
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
53 import pickle
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
54
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
55 # DEBUG_MODE=1 # make this true to get printouts which help you understand what's going on
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
56
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
57 def usage(outfile):
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
58 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
59
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
60 Meta-options:
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
61 --help Display this help then exit.
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
62 --version Output version information then exit.
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
63
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
64 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
65 -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
66 -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
67 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
68 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
69 See also `--coverdir', `--file', `--no-report' below.
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
70 -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
71 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
72 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
73 with `--count --file=FILE'.
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 Modifiers:
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
76 -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
77 -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
78 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
79 -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
80 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
81 <dir>/<package>/<module>.cover.
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
82 -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
83 with '>>>>>> '.
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
84 -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
85 (Can only be used with --count or --report.)
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
86
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
87 Filters, may be repeated multiple times:
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
88 --ignore-module=<mod> Ignore the given module and its submodules
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
89 (if it is a package).
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
90 --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
91 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
92 """ % sys.argv[0])
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
93
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
94 class Ignore:
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
95 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
96 self._mods = modules or []
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
97 self._dirs = dirs or []
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
98
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
99 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
100 self._ignore = { '<string>': 1 }
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
101
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
102 def names(self, filename, modulename):
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
103 if self._ignore.has_key(modulename):
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
104 return self._ignore[modulename]
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
105
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
106 # 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
107 # 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
108 # "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
109 # "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
110 for mod in self._mods:
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
111 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
112 self._ignore[modulename] = 1
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
113 return 1
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
114 # 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
115 # the ignore list
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
116 n = len(mod)
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
117 # (will not overflow since if the first n characters are the
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
118 # same and the name has not already occured, then the size
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
119 # 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
120 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
121 self._ignore[modulename] = 1
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
122 return 1
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
123
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
124 # 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
125 if filename is None:
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
126 # 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
127 self._ignore[modulename] = 1
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
128 return 1
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 # 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
131 for d in self._dirs:
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
132 # 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
133 # as compared to cases like:
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
134 # d = "/usr/local"
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
135 # filename = "/usr/local.py"
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
136 # or
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
137 # d = "/usr/local.py"
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
138 # filename = "/usr/local.py"
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
139 if filename.startswith(d + os.sep):
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
140 self._ignore[modulename] = 1
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
141 return 1
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
142
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
143 # 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
144 self._ignore[modulename] = 0
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
145 return 0
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
146
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
147 class CoverageResults:
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
148 def __init__(self, counts=None, calledfuncs=None, infile=None,
27963
226463445697 Whitespace normalization.
Tim Peters <tim.peters@gmail.com>
parents: 27933
diff changeset
149 outfile=None):
27933
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
150 self.counts = counts
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
151 if self.counts is None:
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
152 self.counts = {}
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
153 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
154 self.calledfuncs = calledfuncs
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
155 if self.calledfuncs is None:
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
156 self.calledfuncs = {}
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
157 self.calledfuncs = self.calledfuncs.copy()
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
158 self.infile = infile
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
159 self.outfile = outfile
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
160 if self.infile:
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
161 # try and merge existing counts file
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
162 try:
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
163 thingie = pickle.load(open(self.infile, 'r'))
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
164 if type(thingie) is types.DictType:
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
165 # backwards compatibility for old trace.py after
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
166 # Zooko touched it but before calledfuncs --Zooko
27963
226463445697 Whitespace normalization.
Tim Peters <tim.peters@gmail.com>
parents: 27933
diff changeset
167 # 2001-10-24
27933
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
168 self.update(self.__class__(thingie))
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
169 elif type(thingie) is types.TupleType and len(thingie) == 2:
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
170 counts, calledfuncs = thingie
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
171 self.update(self.__class__(counts, calledfuncs))
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
172 except (IOError, EOFError):
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
173 pass
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
174 except pickle.UnpicklingError:
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
175 # backwards compatibility for old trace.py before
27963
226463445697 Whitespace normalization.
Tim Peters <tim.peters@gmail.com>
parents: 27933
diff changeset
176 # Zooko touched it --Zooko 2001-10-24
27933
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
177 self.update(self.__class__(marshal.load(open(self.infile))))
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
178
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
179 def update(self, other):
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
180 """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
181 counts = self.counts
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
182 calledfuncs = self.calledfuncs
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
183 other_counts = other.counts
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
184 other_calledfuncs = other.calledfuncs
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
185
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
186 for key in other_counts.keys():
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
187 if key != 'calledfuncs':
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
188 # backwards compatibility for abortive attempt to
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
189 # stuff calledfuncs into self.counts, by Zooko
27963
226463445697 Whitespace normalization.
Tim Peters <tim.peters@gmail.com>
parents: 27933
diff changeset
190 # --Zooko 2001-10-24
27933
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
191 counts[key] = counts.get(key, 0) + other_counts[key]
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
192
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
193 for key in other_calledfuncs.keys():
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
194 calledfuncs[key] = 1
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
195
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
196 def write_results(self, show_missing = 1, summary = 0, coverdir = None):
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
197 """
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
198 @param coverdir
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
199 """
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
200 for filename, modulename, funcname in self.calledfuncs.keys():
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
201 print ("filename: %s, modulename: %s, funcname: %s"
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
202 % (filename, modulename, funcname))
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
203
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
204 import re
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
205 # 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
206 # accessible on a per-file basis
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
207 per_file = {}
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
208 for thingie in self.counts.keys():
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
209 if thingie != "calledfuncs":
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
210 # backwards compatibility for abortive attempt to
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
211 # stuff calledfuncs into self.counts, by Zooko --Zooko
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
212 # 2001-10-24
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
213 filename, lineno = thingie
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
214 lines_hit = per_file[filename] = per_file.get(filename, {})
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
215 lines_hit[lineno] = self.counts[(filename, lineno)]
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
216
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
217 # there are many places where this is insufficient, like a blank
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
218 # line embedded in a multiline string.
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
219 blank = re.compile(r'^\s*(#.*)?$')
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
220
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
221 # accumulate summary info, if needed
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
222 sums = {}
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
223
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
224 # generate file paths for the coverage files we are going to write...
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
225 fnlist = []
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
226 tfdir = tempfile.gettempdir()
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
227 for key in per_file.keys():
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
228 filename = key
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
229
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
230 # 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
231 if filename == "<string>":
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
232 continue
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
233 # are these caused by code compiled using exec or something?
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
234 if filename.startswith(tfdir):
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
235 continue
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
236
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
237 modulename = inspect.getmodulename(filename)
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
238
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
239 if filename.endswith(".pyc") or filename.endswith(".pyo"):
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
240 filename = filename[:-1]
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
241
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
242 if coverdir:
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
243 thiscoverdir = coverdir
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
244 else:
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
245 thiscoverdir = os.path.dirname(os.path.abspath(filename))
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
246
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
247 # the code from here to "<<<" is the contents of the `fileutil.make_dirs()' function in the Mojo Nation project. --Zooko 2001-10-14
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
248 # http://cvs.sourceforge.net/cgi-bin/viewcvs.cgi/mojonation/evil/common/fileutil.py?rev=HEAD&content-type=text/vnd.viewcvs-markup
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
249 tx = None
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
250 try:
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
251 os.makedirs(thiscoverdir)
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
252 except OSError, x:
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
253 tx = x
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
254
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
255 if not os.path.isdir(thiscoverdir):
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
256 if tx:
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
257 raise tx
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
258 raise exceptions.IOError, "unknown error prevented creation of directory: %s" % thiscoverdir # careful not to construct an IOError with a 2-tuple, as that has a special meaning...
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
259 # <<<
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
260
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
261 # build list file name by appending a ".cover" to the module name
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
262 # and sticking it into the specified directory
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
263 if "." in modulename:
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
264 # A module in a package
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
265 finalname = modulename.split(".")[-1]
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
266 listfilename = os.path.join(thiscoverdir, finalname + ".cover")
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
267 else:
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
268 listfilename = os.path.join(thiscoverdir, modulename + ".cover")
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
269
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
270 # Get the original lines from the .py file
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
271 try:
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
272 lines = open(filename, 'r').readlines()
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
273 except IOError, err:
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
274 sys.stderr.write("trace: Could not open %s for reading because: %s - skipping\n" % (`filename`, err))
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
275 continue
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
276
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
277 try:
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
278 outfile = open(listfilename, 'w')
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
279 except IOError, err:
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
280 sys.stderr.write(
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
281 '%s: Could not open %s for writing because: %s" \
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
282 "- skipping\n' % ("trace", `listfilename`, err))
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
283 continue
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
284
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
285 # 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
286 # 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
287 if show_missing:
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
288 executable_linenos = find_executable_linenos(filename)
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
289 else:
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
290 executable_linenos = {}
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
291
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
292 n_lines = 0
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
293 n_hits = 0
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
294 lines_hit = per_file[key]
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
295 for i in range(len(lines)):
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
296 line = lines[i]
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
297
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
298 # do the blank/comment match to try to mark more lines
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
299 # (help the reader find stuff that hasn't been covered)
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
300 if lines_hit.has_key(i+1):
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
301 # count precedes the lines that we captured
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
302 outfile.write('%5d: ' % lines_hit[i+1])
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
303 n_hits = n_hits + 1
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
304 n_lines = n_lines + 1
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
305 elif blank.match(line):
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
306 # blank lines and comments are preceded by dots
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
307 outfile.write(' . ')
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
308 else:
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
309 # lines preceded by no marks weren't hit
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
310 # Highlight them if so indicated, unless the line contains
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
311 # '#pragma: NO COVER' (it is possible to embed this into
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
312 # the text as a non-comment; no easy fix)
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
313 if executable_linenos.has_key(i+1) and \
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
314 lines[i].find(' '.join(['#pragma', 'NO COVER'])) == -1:
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
315 outfile.write('>>>>>> ')
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
316 else:
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
317 outfile.write(' '*7)
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
318 n_lines = n_lines + 1
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
319 outfile.write(lines[i].expandtabs(8))
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
320
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
321 outfile.close()
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
322
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
323 if summary and n_lines:
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
324 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
325 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
326
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
327 if summary and sums:
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
328 mods = sums.keys()
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
329 mods.sort()
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
330 print "lines cov% module (path)"
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
331 for m in mods:
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
332 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
333 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
334
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
335 if self.outfile:
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
336 # 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
337 try:
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
338 pickle.dump((self.counts, self.calledfuncs),
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
339 open(self.outfile, 'w'), 1)
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
340 except IOError, err:
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
341 sys.stderr.write("cannot save counts files because %s" % err)
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
342
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
343 def _find_LINENO_from_code(code):
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
344 """return the numbers of the lines containing the source code that
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
345 was compiled into code"""
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
346 linenos = {}
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
347
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
348 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
349 table_length = len(line_increments)
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
350
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
351 lineno = code.co_firstlineno
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
352
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
353 for li in line_increments:
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
354 linenos[lineno] = 1
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
355 lineno += li
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
356 linenos[lineno] = 1
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
357
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
358 return linenos
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
359
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
360 def _find_LINENO(code):
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
361 """return all of the lineno information from a code object"""
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
362 import types
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
363
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
364 # get all of the lineno information from the code of this scope level
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
365 linenos = _find_LINENO_from_code(code)
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
366
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
367 # 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
368 for c in code.co_consts:
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
369 if type(c) == types.CodeType:
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
370 # find another code object, so recurse into it
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
371 linenos.update(_find_LINENO(c))
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
372 return linenos
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
373
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
374 def find_executable_linenos(filename):
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
375 """return a dict of the line numbers from executable statements in a file
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
376
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
377 """
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
378 import parser
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 assert filename.endswith('.py')
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
381
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
382 prog = open(filename).read()
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
383 ast = parser.suite(prog)
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
384 code = parser.compileast(ast, filename)
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
385
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
386 return _find_LINENO(code)
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
387
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
388 ### XXX because os.path.commonprefix seems broken by my way of thinking...
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
389 def commonprefix(dirs):
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
390 "Given a list of pathnames, returns the longest common leading component"
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
391 if not dirs: return ''
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
392 n = copy.copy(dirs)
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
393 for i in range(len(n)):
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
394 n[i] = n[i].split(os.sep)
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
395 prefix = n[0]
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
396 for item in n:
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
397 for i in range(len(prefix)):
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
398 if prefix[:i+1] <> item[:i+1]:
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
399 prefix = prefix[:i]
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
400 if i == 0: return ''
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
401 break
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
402 return os.sep.join(prefix)
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
403
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
404 class Trace:
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
405 def __init__(self, count=1, trace=1, countfuncs=0, ignoremods=(),
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
406 ignoredirs=(), infile=None, outfile=None):
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
407 """
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
408 @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
409 line is executed
27933
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
410 @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
411 being counted
27933
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
412 @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
413 (filename, modulename, funcname,) for functions
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
414 that were called at least once; This overrides
27963
226463445697 Whitespace normalization.
Tim Peters <tim.peters@gmail.com>
parents: 27933
diff changeset
415 `count' and `trace'
27933
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
416 @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
417 @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
418 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
419 @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
420 added into the results
27933
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
421 @param outfile file in which to write the results
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
422 """
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
423 self.infile = infile
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
424 self.outfile = outfile
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
425 self.ignore = Ignore(ignoremods, ignoredirs)
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
426 self.counts = {} # keys are (filename, linenumber)
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
427 self.blabbed = {} # for debugging
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
428 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
429 self.donothing = 0
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
430 self.trace = trace
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
431 self._calledfuncs = {}
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
432 if countfuncs:
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
433 self.globaltrace = self.globaltrace_countfuncs
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
434 elif trace and count:
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
435 self.globaltrace = self.globaltrace_lt
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
436 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
437 elif trace:
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
438 self.globaltrace = self.globaltrace_lt
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
439 self.localtrace = self.localtrace_trace
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
440 elif count:
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
441 self.globaltrace = self.globaltrace_lt
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
442 self.localtrace = self.localtrace_count
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
443 else:
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
444 # Ahem -- do nothing? Okay.
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
445 self.donothing = 1
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
446
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
447 def run(self, cmd):
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
448 import __main__
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
449 dict = __main__.__dict__
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
450 if not self.donothing:
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
451 sys.settrace(self.globaltrace)
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
452 try:
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
453 exec cmd in dict, dict
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
454 finally:
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
455 if not self.donothing:
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
456 sys.settrace(None)
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
457
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
458 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
459 if globals is None: globals = {}
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
460 if locals is None: locals = {}
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
461 if not self.donothing:
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
462 sys.settrace(self.globaltrace)
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
463 try:
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
464 exec cmd in globals, locals
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
465 finally:
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
466 if not self.donothing:
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
467 sys.settrace(None)
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
468
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
469 def runfunc(self, func, *args, **kw):
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
470 result = None
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
471 if not self.donothing:
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
472 sys.settrace(self.globaltrace)
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
473 try:
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
474 result = apply(func, args, kw)
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
475 finally:
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
476 if not self.donothing:
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
477 sys.settrace(None)
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
478 return result
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
479
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
480 def globaltrace_countfuncs(self, frame, why, arg):
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
481 """
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
482 Handles `call' events (why == 'call') and adds the (filename, modulename, funcname,) to the self._calledfuncs dict.
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
483 """
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
484 if why == 'call':
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
485 filename, lineno, funcname, context, lineindex = \
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
486 inspect.getframeinfo(frame, 0)
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
487 if filename:
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
488 modulename = inspect.getmodulename(filename)
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
489 else:
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
490 modulename = None
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
491 self._calledfuncs[(filename, modulename, funcname,)] = 1
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
492
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
493 def globaltrace_lt(self, frame, why, arg):
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
494 """
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
495 Handles `call' events (why == 'call') and if the code block being entered is to be ignored then it returns `None', else it returns `self.localtrace'.
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
496 """
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
497 if why == 'call':
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
498 filename, lineno, funcname, context, lineindex = \
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
499 inspect.getframeinfo(frame, 0)
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
500 if filename:
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
501 modulename = inspect.getmodulename(filename)
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
502 if modulename is not None:
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
503 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
504 if not ignore_it:
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
505 if self.trace:
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
506 print (" --- modulename: %s, funcname: %s"
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
507 % (modulename, funcname))
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
508 return self.localtrace
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
509 else:
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
510 # XXX why no filename?
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
511 return None
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
512
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
513 def localtrace_trace_and_count(self, frame, why, arg):
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
514 if why == 'line':
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
515 # record the file name and line number of every trace
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
516 # XXX I wish inspect offered me an optimized
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
517 # `getfilename(frame)' to use in place of the presumably
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
518 # heavier `getframeinfo()'. --Zooko 2001-10-14
27963
226463445697 Whitespace normalization.
Tim Peters <tim.peters@gmail.com>
parents: 27933
diff changeset
519
27933
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
520 filename, lineno, funcname, context, lineindex = \
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
521 inspect.getframeinfo(frame, 1)
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
522 key = filename, lineno
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
523 self.counts[key] = self.counts.get(key, 0) + 1
27963
226463445697 Whitespace normalization.
Tim Peters <tim.peters@gmail.com>
parents: 27933
diff changeset
524
27933
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
525 # XXX not convinced that this memoizing is a performance
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
526 # win -- I don't know enough about Python guts to tell.
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
527 # --Zooko 2001-10-14
27963
226463445697 Whitespace normalization.
Tim Peters <tim.peters@gmail.com>
parents: 27933
diff changeset
528
27933
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
529 bname = self.pathtobasename.get(filename)
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
530 if bname is None:
27963
226463445697 Whitespace normalization.
Tim Peters <tim.peters@gmail.com>
parents: 27933
diff changeset
531
27933
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
532 # Using setdefault faster than two separate lines?
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
533 # --Zooko 2001-10-14
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
534 bname = self.pathtobasename.setdefault(filename,
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
535 os.path.basename(filename))
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
536 try:
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
537 print "%s(%d): %s" % (bname, lineno, context[lineindex]),
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
538 except IndexError:
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
539 # Uh.. sometimes getframeinfo gives me a context of
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
540 # length 1 and a lineindex of -2. Oh well.
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
541 pass
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
542 return self.localtrace
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
543
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
544 def localtrace_trace(self, frame, why, arg):
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
545 if why == 'line':
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
546 # XXX shouldn't do the count increment when arg is
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
547 # exception? But be careful to return self.localtrace
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
548 # when arg is exception! ? --Zooko 2001-10-14
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
549
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
550 # record the file name and line number of every trace XXX
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
551 # I wish inspect offered me an optimized
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
552 # `getfilename(frame)' to use in place of the presumably
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
553 # heavier `getframeinfo()'. --Zooko 2001-10-14
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
554 filename, lineno, funcname, context, lineindex = \
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
555 inspect.getframeinfo(frame)
27963
226463445697 Whitespace normalization.
Tim Peters <tim.peters@gmail.com>
parents: 27933
diff changeset
556
27933
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
557 # XXX not convinced that this memoizing is a performance
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
558 # win -- I don't know enough about Python guts to tell.
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
559 # --Zooko 2001-10-14
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
560 bname = self.pathtobasename.get(filename)
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
561 if bname is None:
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
562 # Using setdefault faster than two separate lines?
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
563 # --Zooko 2001-10-14
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
564 bname = self.pathtobasename.setdefault(filename, os.path.basename(filename))
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
565 if context is not None:
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
566 try:
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
567 print "%s(%d): %s" % (bname, lineno, context[lineindex]),
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
568 except IndexError:
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
569 # Uh.. sometimes getframeinfo gives me a context of length 1 and a lineindex of -2. Oh well.
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
570 pass
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
571 else:
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
572 print "%s(???): ???" % bname
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
573 return self.localtrace
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
574
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
575 def localtrace_count(self, frame, why, arg):
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
576 if why == 'line':
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
577 filename = frame.f_code.co_filename
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
578 lineno = frame.f_lineno
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
579 key = filename, lineno
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
580 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
581 return self.localtrace
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
582
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
583 def results(self):
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
584 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
585 outfile=self.outfile,
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
586 calledfuncs=self._calledfuncs)
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
587
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
588 def _err_exit(msg):
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
589 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
590 sys.exit(1)
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
591
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
592 def main(argv=None):
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
593 import getopt
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
594
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
595 if argv is None:
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
596 argv = sys.argv
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
597 try:
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
598 opts, prog_argv = getopt.getopt(argv[1:], "tcrRf:d:msC:l",
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
599 ["help", "version", "trace", "count",
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
600 "report", "no-report", "summary",
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
601 "file=", "missing",
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
602 "ignore-module=", "ignore-dir=",
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
603 "coverdir=", "listfuncs",])
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
604
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
605 except getopt.error, msg:
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
606 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
607 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
608 % sys.argv[0])
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
609 sys.exit(1)
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
610
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
611 trace = 0
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
612 count = 0
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
613 report = 0
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
614 no_report = 0
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
615 counts_file = None
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
616 missing = 0
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
617 ignore_modules = []
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
618 ignore_dirs = []
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
619 coverdir = None
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
620 summary = 0
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
621 listfuncs = False
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
622
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
623 for opt, val in opts:
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
624 if opt == "--help":
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
625 usage(sys.stdout)
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
626 sys.exit(0)
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
627
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
628 if opt == "--version":
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
629 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
630 sys.exit(0)
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
631
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
632 if opt == "-l" or opt == "--listfuncs":
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
633 listfuncs = True
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
634 continue
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
635
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
636 if opt == "-t" or opt == "--trace":
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
637 trace = 1
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
638 continue
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
639
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
640 if opt == "-c" or opt == "--count":
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
641 count = 1
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
642 continue
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 if opt == "-r" or opt == "--report":
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
645 report = 1
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
646 continue
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
647
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
648 if opt == "-R" or opt == "--no-report":
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
649 no_report = 1
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
650 continue
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 if opt == "-f" or opt == "--file":
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
653 counts_file = val
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
654 continue
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
655
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
656 if opt == "-m" or opt == "--missing":
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
657 missing = 1
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
658 continue
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
659
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
660 if opt == "-C" or opt == "--coverdir":
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
661 coverdir = val
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
662 continue
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
663
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
664 if opt == "-s" or opt == "--summary":
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
665 summary = 1
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
666 continue
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
667
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
668 if opt == "--ignore-module":
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
669 ignore_modules.append(val)
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
670 continue
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
671
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
672 if opt == "--ignore-dir":
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
673 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
674 s = os.path.expandvars(s)
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
675 # 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
676
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
677 s = s.replace("$prefix",
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
678 os.path.join(sys.prefix, "lib",
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
679 "python" + sys.version[:3]))
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
680 s = s.replace("$exec_prefix",
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
681 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
682 "python" + sys.version[:3]))
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
683 s = os.path.normpath(s)
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
684 ignore_dirs.append(s)
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
685 continue
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
686
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
687 assert 0, "Should never get here"
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
688
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
689 if listfuncs and (count or trace):
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
690 _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
691
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
692 if not count and not trace and not report and not listfuncs:
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
693 _err_exit("must specify one of --trace, --count, --report or --listfuncs")
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
694
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
695 if report and no_report:
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
696 _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
697
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
698 if report and not counts_file:
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
699 _err_exit("--report requires a --file")
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 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
702 _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
703
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
704 # everything is ready
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
705 if report:
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
706 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
707 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
708 else:
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
709 sys.argv = prog_argv
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
710 progname = prog_argv[0]
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
711 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
712
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
713 t = Trace(count, trace, countfuncs=listfuncs,
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
714 ignoremods=ignore_modules, ignoredirs=ignore_dirs,
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
715 infile=counts_file, outfile=counts_file)
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
716 try:
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
717 t.run('execfile(' + `progname` + ')')
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
718 except IOError, err:
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
719 _err_exit("Cannot run file %s because: %s" % (`sys.argv[0]`, err))
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
720 except SystemExit:
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
721 pass
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
722
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
723 results = t.results()
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 not no_report:
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
726 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
727
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
728 if __name__=='__main__':
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
729 main()