annotate Lib/trace.py @ 70817:dd775b1561ba

Fix markup.
author Georg Brandl <georg@python.org>
date Tue, 14 Jun 2011 21:09:55 +0200
parents c9792d3166eb
children 8a47d2322df0
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
rev   line source
59522
fcfe324663a6 convert shebang lines: python -> python3
Benjamin Peterson <benjamin@python.org>
parents: 59034
diff changeset
1 #!/usr/bin/env python3
27933
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
2
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
3 # portions copyright 2001, Autonomous Zones Industries, Inc., all rights...
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
4 # err... reserved and offered to the public under the terms of the
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
5 # Python 2.2 license.
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
6 # Author: Zooko O'Whielacronx
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
7 # http://zooko.com/
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
8 # mailto:zooko@zooko.com
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
9 #
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
10 # Copyright 2000, Mojam Media, Inc., all rights reserved.
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
11 # Author: Skip Montanaro
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
12 #
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
13 # Copyright 1999, Bioreason, Inc., all rights reserved.
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
14 # Author: Andrew Dalke
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
15 #
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
16 # Copyright 1995-1997, Automatrix, Inc., all rights reserved.
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
17 # Author: Skip Montanaro
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
18 #
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
19 # Copyright 1991-1995, Stichting Mathematisch Centrum, all rights reserved.
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
20 #
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
21 #
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
22 # Permission to use, copy, modify, and distribute this Python software and
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
23 # its associated documentation for any purpose without fee is hereby
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
24 # granted, provided that the above copyright notice appears in all copies,
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
25 # and that both that copyright notice and this permission notice appear in
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
26 # supporting documentation, and that the name of neither Automatrix,
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
27 # Bioreason or Mojam Media be used in advertising or publicity pertaining to
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
28 # distribution of the software without specific, written prior permission.
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
29 #
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
30 """program/module to trace Python program or function execution
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
31
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
32 Sample use, command line:
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
33 trace.py -c -f counts --ignore-dir '$prefix' spam.py eggs
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
34 trace.py -t --ignore-dir '$prefix' spam.py eggs
31892
081972ae2f84 Refactor common code out of globaltrace_trackcallers() and
Skip Montanaro <skip@pobox.com>
parents: 31881
diff changeset
35 trace.py --trackcalls spam.py eggs
27933
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
36
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
37 Sample use, programmatically
38453
29ae5e0fc348 Much-needed merge (using svnmerge.py this time) of trunk changes into p3yk.
Thomas Wouters <thomas@python.org>
parents: 35806
diff changeset
38 import sys
29ae5e0fc348 Much-needed merge (using svnmerge.py this time) of trunk changes into p3yk.
Thomas Wouters <thomas@python.org>
parents: 35806
diff changeset
39
29ae5e0fc348 Much-needed merge (using svnmerge.py this time) of trunk changes into p3yk.
Thomas Wouters <thomas@python.org>
parents: 35806
diff changeset
40 # create a Trace object, telling it what to ignore, and whether to
29ae5e0fc348 Much-needed merge (using svnmerge.py this time) of trunk changes into p3yk.
Thomas Wouters <thomas@python.org>
parents: 35806
diff changeset
41 # do tracing or line-counting or both.
29ae5e0fc348 Much-needed merge (using svnmerge.py this time) of trunk changes into p3yk.
Thomas Wouters <thomas@python.org>
parents: 35806
diff changeset
42 tracer = trace.Trace(ignoredirs=[sys.prefix, sys.exec_prefix,], trace=0,
29ae5e0fc348 Much-needed merge (using svnmerge.py this time) of trunk changes into p3yk.
Thomas Wouters <thomas@python.org>
parents: 35806
diff changeset
43 count=1)
29ae5e0fc348 Much-needed merge (using svnmerge.py this time) of trunk changes into p3yk.
Thomas Wouters <thomas@python.org>
parents: 35806
diff changeset
44 # run the new command using the given tracer
29ae5e0fc348 Much-needed merge (using svnmerge.py this time) of trunk changes into p3yk.
Thomas Wouters <thomas@python.org>
parents: 35806
diff changeset
45 tracer.run('main()')
29ae5e0fc348 Much-needed merge (using svnmerge.py this time) of trunk changes into p3yk.
Thomas Wouters <thomas@python.org>
parents: 35806
diff changeset
46 # make a report, placing output in /tmp
29ae5e0fc348 Much-needed merge (using svnmerge.py this time) of trunk changes into p3yk.
Thomas Wouters <thomas@python.org>
parents: 35806
diff changeset
47 r = tracer.results()
29ae5e0fc348 Much-needed merge (using svnmerge.py this time) of trunk changes into p3yk.
Thomas Wouters <thomas@python.org>
parents: 35806
diff changeset
48 r.write_results(show_missing=True, coverdir="/tmp")
27933
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
49 """
66308
4f29aa8f9f54 Issue #10371: Deprecated undocumented functions in the trace module.
Alexander Belopolsky <alexander.belopolsky@gmail.com>
parents: 66097
diff changeset
50 __all__ = ['Trace', 'CoverageResults']
52690
b43f14c103bb #5656: detect correct encoding of files when reporting coverage in trace.py, and ignore files in the temporary directory when reporting.
Georg Brandl <georg@python.org>
parents: 54437
diff changeset
51 import io
28712
d3127b671d36 Holistic refactoring.
Jeremy Hylton <jeremy@alum.mit.edu>
parents: 28089
diff changeset
52 import linecache
d3127b671d36 Holistic refactoring.
Jeremy Hylton <jeremy@alum.mit.edu>
parents: 28089
diff changeset
53 import os
d3127b671d36 Holistic refactoring.
Jeremy Hylton <jeremy@alum.mit.edu>
parents: 28089
diff changeset
54 import re
d3127b671d36 Holistic refactoring.
Jeremy Hylton <jeremy@alum.mit.edu>
parents: 28089
diff changeset
55 import sys
45548
8c3c5b4eef55 Merged revisions 61038,61042-61045,61047,61050,61053,61055-61056,61061-61062,61066,61068,61070,61081-61095 via svnmerge from
Christian Heimes <christian@cheimes.de>
parents: 44782
diff changeset
56 import time
28712
d3127b671d36 Holistic refactoring.
Jeremy Hylton <jeremy@alum.mit.edu>
parents: 28089
diff changeset
57 import token
d3127b671d36 Holistic refactoring.
Jeremy Hylton <jeremy@alum.mit.edu>
parents: 28089
diff changeset
58 import tokenize
64717
7dbfc7ef39f2 Merged revisions 84780-84781 via svnmerge from
Alexander Belopolsky <alexander.belopolsky@gmail.com>
parents: 63438
diff changeset
59 import inspect
31892
081972ae2f84 Refactor common code out of globaltrace_trackcallers() and
Skip Montanaro <skip@pobox.com>
parents: 31881
diff changeset
60 import gc
64925
235044b39a92 Merged revisions 84994 via svnmerge from
Alexander Belopolsky <alexander.belopolsky@gmail.com>
parents: 64717
diff changeset
61 import dis
42322
3e33215c66ed Getting rid of cPickle. Mmm, feels good!
Guido van Rossum <guido@python.org>
parents: 40910
diff changeset
62 import pickle
66308
4f29aa8f9f54 Issue #10371: Deprecated undocumented functions in the trace module.
Alexander Belopolsky <alexander.belopolsky@gmail.com>
parents: 66097
diff changeset
63 from warnings import warn as _warn
27933
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
64
66001
8911f475ad3d Merged revisions 86229 via svnmerge from
Alexander Belopolsky <alexander.belopolsky@gmail.com>
parents: 65115
diff changeset
65 try:
8911f475ad3d Merged revisions 86229 via svnmerge from
Alexander Belopolsky <alexander.belopolsky@gmail.com>
parents: 65115
diff changeset
66 import threading
8911f475ad3d Merged revisions 86229 via svnmerge from
Alexander Belopolsky <alexander.belopolsky@gmail.com>
parents: 65115
diff changeset
67 except ImportError:
8911f475ad3d Merged revisions 86229 via svnmerge from
Alexander Belopolsky <alexander.belopolsky@gmail.com>
parents: 65115
diff changeset
68 _settrace = sys.settrace
8911f475ad3d Merged revisions 86229 via svnmerge from
Alexander Belopolsky <alexander.belopolsky@gmail.com>
parents: 65115
diff changeset
69
8911f475ad3d Merged revisions 86229 via svnmerge from
Alexander Belopolsky <alexander.belopolsky@gmail.com>
parents: 65115
diff changeset
70 def _unsettrace():
8911f475ad3d Merged revisions 86229 via svnmerge from
Alexander Belopolsky <alexander.belopolsky@gmail.com>
parents: 65115
diff changeset
71 sys.settrace(None)
8911f475ad3d Merged revisions 86229 via svnmerge from
Alexander Belopolsky <alexander.belopolsky@gmail.com>
parents: 65115
diff changeset
72 else:
8911f475ad3d Merged revisions 86229 via svnmerge from
Alexander Belopolsky <alexander.belopolsky@gmail.com>
parents: 65115
diff changeset
73 def _settrace(func):
8911f475ad3d Merged revisions 86229 via svnmerge from
Alexander Belopolsky <alexander.belopolsky@gmail.com>
parents: 65115
diff changeset
74 threading.settrace(func)
8911f475ad3d Merged revisions 86229 via svnmerge from
Alexander Belopolsky <alexander.belopolsky@gmail.com>
parents: 65115
diff changeset
75 sys.settrace(func)
8911f475ad3d Merged revisions 86229 via svnmerge from
Alexander Belopolsky <alexander.belopolsky@gmail.com>
parents: 65115
diff changeset
76
8911f475ad3d Merged revisions 86229 via svnmerge from
Alexander Belopolsky <alexander.belopolsky@gmail.com>
parents: 65115
diff changeset
77 def _unsettrace():
8911f475ad3d Merged revisions 86229 via svnmerge from
Alexander Belopolsky <alexander.belopolsky@gmail.com>
parents: 65115
diff changeset
78 sys.settrace(None)
8911f475ad3d Merged revisions 86229 via svnmerge from
Alexander Belopolsky <alexander.belopolsky@gmail.com>
parents: 65115
diff changeset
79 threading.settrace(None)
8911f475ad3d Merged revisions 86229 via svnmerge from
Alexander Belopolsky <alexander.belopolsky@gmail.com>
parents: 65115
diff changeset
80
66308
4f29aa8f9f54 Issue #10371: Deprecated undocumented functions in the trace module.
Alexander Belopolsky <alexander.belopolsky@gmail.com>
parents: 66097
diff changeset
81 def _usage(outfile):
27933
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
82 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
83
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
84 Meta-options:
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
85 --help Display this help then exit.
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
86 --version Output version information then exit.
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
87
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
88 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
89 -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
90 -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
91 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
92 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
93 See also `--coverdir', `--file', `--no-report' below.
29619
f8069f7edd4a add mention of -l, --listfuncs to usage() small part of 542562
Skip Montanaro <skip@pobox.com>
parents: 29595
diff changeset
94 -l, --listfuncs Keep track of which functions are executed at least
29620
77eda0fb0e14 fix typo
Fred Drake <fdrake@acm.org>
parents: 29619
diff changeset
95 once and write the results to sys.stdout after the
29619
f8069f7edd4a add mention of -l, --listfuncs to usage() small part of 542562
Skip Montanaro <skip@pobox.com>
parents: 29595
diff changeset
96 program exits.
31881
f35b4acc594f Added --trackcalls command line arg to display crude caller/callee
Skip Montanaro <skip@pobox.com>
parents: 31551
diff changeset
97 -T, --trackcalls Keep track of caller/called pairs and write the
f35b4acc594f Added --trackcalls command line arg to display crude caller/callee
Skip Montanaro <skip@pobox.com>
parents: 31551
diff changeset
98 results to sys.stdout after the program exits.
27933
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
99 -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
100 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
101 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
102 with `--count --file=FILE'.
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
103
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
104 Modifiers:
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
105 -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
106 -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
107 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
108 -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
109 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
110 <dir>/<package>/<module>.cover.
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
111 -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
112 with '>>>>>> '.
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
113 -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
114 (Can only be used with --count or --report.)
45548
8c3c5b4eef55 Merged revisions 61038,61042-61045,61047,61050,61053,61055-61056,61061-61062,61066,61068,61070,61081-61095 via svnmerge from
Christian Heimes <christian@cheimes.de>
parents: 44782
diff changeset
115 -g, --timing Prefix each line with the time since the program started.
8c3c5b4eef55 Merged revisions 61038,61042-61045,61047,61050,61053,61055-61056,61061-61062,61066,61068,61070,61081-61095 via svnmerge from
Christian Heimes <christian@cheimes.de>
parents: 44782
diff changeset
116 Only used while tracing.
27933
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
117
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
118 Filters, may be repeated multiple times:
44782
ca2683a67943 Merged revisions 60080-60089,60091-60093 via svnmerge from
Georg Brandl <georg@python.org>
parents: 44025
diff changeset
119 --ignore-module=<mod> Ignore the given module(s) and its submodules
ca2683a67943 Merged revisions 60080-60089,60091-60093 via svnmerge from
Georg Brandl <georg@python.org>
parents: 44025
diff changeset
120 (if it is a package). Accepts comma separated
ca2683a67943 Merged revisions 60080-60089,60091-60093 via svnmerge from
Georg Brandl <georg@python.org>
parents: 44025
diff changeset
121 list of module names
27933
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
122 --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
123 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
124 """ % sys.argv[0])
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
125
28712
d3127b671d36 Holistic refactoring.
Jeremy Hylton <jeremy@alum.mit.edu>
parents: 28089
diff changeset
126 PRAGMA_NOCOVER = "#pragma NO COVER"
d3127b671d36 Holistic refactoring.
Jeremy Hylton <jeremy@alum.mit.edu>
parents: 28089
diff changeset
127
d3127b671d36 Holistic refactoring.
Jeremy Hylton <jeremy@alum.mit.edu>
parents: 28089
diff changeset
128 # Simple rx to find lines with no code.
d3127b671d36 Holistic refactoring.
Jeremy Hylton <jeremy@alum.mit.edu>
parents: 28089
diff changeset
129 rx_blank = re.compile(r'^\s*(#.*)?$')
d3127b671d36 Holistic refactoring.
Jeremy Hylton <jeremy@alum.mit.edu>
parents: 28089
diff changeset
130
66308
4f29aa8f9f54 Issue #10371: Deprecated undocumented functions in the trace module.
Alexander Belopolsky <alexander.belopolsky@gmail.com>
parents: 66097
diff changeset
131 class _Ignore:
66088
8ab1de0813b3 Streamlined code in trace.Ignore and added unit tests.
Alexander Belopolsky <alexander.belopolsky@gmail.com>
parents: 66068
diff changeset
132 def __init__(self, modules=None, dirs=None):
8ab1de0813b3 Streamlined code in trace.Ignore and added unit tests.
Alexander Belopolsky <alexander.belopolsky@gmail.com>
parents: 66068
diff changeset
133 self._mods = set() if not modules else set(modules)
8ab1de0813b3 Streamlined code in trace.Ignore and added unit tests.
Alexander Belopolsky <alexander.belopolsky@gmail.com>
parents: 66068
diff changeset
134 self._dirs = [] if not dirs else [os.path.normpath(d)
8ab1de0813b3 Streamlined code in trace.Ignore and added unit tests.
Alexander Belopolsky <alexander.belopolsky@gmail.com>
parents: 66068
diff changeset
135 for d in dirs]
27933
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
136 self._ignore = { '<string>': 1 }
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
137
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
138 def names(self, filename, modulename):
39575
63237fcefd99 Get rid of dict.has_key(). Boy this has a lot of repercussions!
Guido van Rossum <guido@python.org>
parents: 39442
diff changeset
139 if modulename in self._ignore:
27933
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
140 return self._ignore[modulename]
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
141
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
142 # haven't seen this one before, so see if the module name is
66088
8ab1de0813b3 Streamlined code in trace.Ignore and added unit tests.
Alexander Belopolsky <alexander.belopolsky@gmail.com>
parents: 66068
diff changeset
143 # on the ignore list.
8ab1de0813b3 Streamlined code in trace.Ignore and added unit tests.
Alexander Belopolsky <alexander.belopolsky@gmail.com>
parents: 66068
diff changeset
144 if modulename in self._mods: # Identical names, so ignore
8ab1de0813b3 Streamlined code in trace.Ignore and added unit tests.
Alexander Belopolsky <alexander.belopolsky@gmail.com>
parents: 66068
diff changeset
145 self._ignore[modulename] = 1
8ab1de0813b3 Streamlined code in trace.Ignore and added unit tests.
Alexander Belopolsky <alexander.belopolsky@gmail.com>
parents: 66068
diff changeset
146 return 1
8ab1de0813b3 Streamlined code in trace.Ignore and added unit tests.
Alexander Belopolsky <alexander.belopolsky@gmail.com>
parents: 66068
diff changeset
147
8ab1de0813b3 Streamlined code in trace.Ignore and added unit tests.
Alexander Belopolsky <alexander.belopolsky@gmail.com>
parents: 66068
diff changeset
148 # check if the module is a proper submodule of something on
8ab1de0813b3 Streamlined code in trace.Ignore and added unit tests.
Alexander Belopolsky <alexander.belopolsky@gmail.com>
parents: 66068
diff changeset
149 # the ignore list
27933
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
150 for mod in self._mods:
66088
8ab1de0813b3 Streamlined code in trace.Ignore and added unit tests.
Alexander Belopolsky <alexander.belopolsky@gmail.com>
parents: 66068
diff changeset
151 # Need to take some care since ignoring
8ab1de0813b3 Streamlined code in trace.Ignore and added unit tests.
Alexander Belopolsky <alexander.belopolsky@gmail.com>
parents: 66068
diff changeset
152 # "cmp" mustn't mean ignoring "cmpcache" but ignoring
8ab1de0813b3 Streamlined code in trace.Ignore and added unit tests.
Alexander Belopolsky <alexander.belopolsky@gmail.com>
parents: 66068
diff changeset
153 # "Spam" must also mean ignoring "Spam.Eggs".
8ab1de0813b3 Streamlined code in trace.Ignore and added unit tests.
Alexander Belopolsky <alexander.belopolsky@gmail.com>
parents: 66068
diff changeset
154 if modulename.startswith(mod + '.'):
27933
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
155 self._ignore[modulename] = 1
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
156 return 1
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
157
66088
8ab1de0813b3 Streamlined code in trace.Ignore and added unit tests.
Alexander Belopolsky <alexander.belopolsky@gmail.com>
parents: 66068
diff changeset
158 # Now check that filename isn't in one of the directories
27933
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
159 if filename is None:
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
160 # 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
161 self._ignore[modulename] = 1
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
162 return 1
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
163
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
164 # 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
165 for d in self._dirs:
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
166 # 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
167 # as compared to cases like:
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
168 # d = "/usr/local"
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
169 # filename = "/usr/local.py"
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
170 # or
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
171 # d = "/usr/local.py"
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
172 # filename = "/usr/local.py"
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
173 if filename.startswith(d + os.sep):
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
174 self._ignore[modulename] = 1
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
175 return 1
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
176
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
177 # 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
178 self._ignore[modulename] = 0
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
179 return 0
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
180
66308
4f29aa8f9f54 Issue #10371: Deprecated undocumented functions in the trace module.
Alexander Belopolsky <alexander.belopolsky@gmail.com>
parents: 66097
diff changeset
181 def _modname(path):
28712
d3127b671d36 Holistic refactoring.
Jeremy Hylton <jeremy@alum.mit.edu>
parents: 28089
diff changeset
182 """Return a plausible module name for the patch."""
28713
d08d15a70295 Add helper function to get module name taking packages into account.
Jeremy Hylton <jeremy@alum.mit.edu>
parents: 28712
diff changeset
183
28712
d3127b671d36 Holistic refactoring.
Jeremy Hylton <jeremy@alum.mit.edu>
parents: 28089
diff changeset
184 base = os.path.basename(path)
d3127b671d36 Holistic refactoring.
Jeremy Hylton <jeremy@alum.mit.edu>
parents: 28089
diff changeset
185 filename, ext = os.path.splitext(base)
d3127b671d36 Holistic refactoring.
Jeremy Hylton <jeremy@alum.mit.edu>
parents: 28089
diff changeset
186 return filename
d3127b671d36 Holistic refactoring.
Jeremy Hylton <jeremy@alum.mit.edu>
parents: 28089
diff changeset
187
66308
4f29aa8f9f54 Issue #10371: Deprecated undocumented functions in the trace module.
Alexander Belopolsky <alexander.belopolsky@gmail.com>
parents: 66097
diff changeset
188 def _fullmodname(path):
28727
051a4f12cc46 Only produce a dotted module name when writing output to a separate
Jeremy Hylton <jeremy@alum.mit.edu>
parents: 28713
diff changeset
189 """Return a plausible module name for the path."""
28713
d08d15a70295 Add helper function to get module name taking packages into account.
Jeremy Hylton <jeremy@alum.mit.edu>
parents: 28712
diff changeset
190
d08d15a70295 Add helper function to get module name taking packages into account.
Jeremy Hylton <jeremy@alum.mit.edu>
parents: 28712
diff changeset
191 # If the file 'path' is part of a package, then the filename isn't
d08d15a70295 Add helper function to get module name taking packages into account.
Jeremy Hylton <jeremy@alum.mit.edu>
parents: 28712
diff changeset
192 # enough to uniquely identify it. Try to do the right thing by
d08d15a70295 Add helper function to get module name taking packages into account.
Jeremy Hylton <jeremy@alum.mit.edu>
parents: 28712
diff changeset
193 # looking in sys.path for the longest matching prefix. We'll
d08d15a70295 Add helper function to get module name taking packages into account.
Jeremy Hylton <jeremy@alum.mit.edu>
parents: 28712
diff changeset
194 # assume that the rest is the package name.
d08d15a70295 Add helper function to get module name taking packages into account.
Jeremy Hylton <jeremy@alum.mit.edu>
parents: 28712
diff changeset
195
39610
3674f20ac781 Merge current trunk into p3yk. This includes the PyNumber_Index API change,
Thomas Wouters <thomas@python.org>
parents: 39575
diff changeset
196 comparepath = os.path.normcase(path)
28713
d08d15a70295 Add helper function to get module name taking packages into account.
Jeremy Hylton <jeremy@alum.mit.edu>
parents: 28712
diff changeset
197 longest = ""
d08d15a70295 Add helper function to get module name taking packages into account.
Jeremy Hylton <jeremy@alum.mit.edu>
parents: 28712
diff changeset
198 for dir in sys.path:
39610
3674f20ac781 Merge current trunk into p3yk. This includes the PyNumber_Index API change,
Thomas Wouters <thomas@python.org>
parents: 39575
diff changeset
199 dir = os.path.normcase(dir)
3674f20ac781 Merge current trunk into p3yk. This includes the PyNumber_Index API change,
Thomas Wouters <thomas@python.org>
parents: 39575
diff changeset
200 if comparepath.startswith(dir) and comparepath[len(dir)] == os.sep:
28713
d08d15a70295 Add helper function to get module name taking packages into account.
Jeremy Hylton <jeremy@alum.mit.edu>
parents: 28712
diff changeset
201 if len(dir) > len(longest):
d08d15a70295 Add helper function to get module name taking packages into account.
Jeremy Hylton <jeremy@alum.mit.edu>
parents: 28712
diff changeset
202 longest = dir
d08d15a70295 Add helper function to get module name taking packages into account.
Jeremy Hylton <jeremy@alum.mit.edu>
parents: 28712
diff changeset
203
30667
400674958025 The fullmodname() function chopped off the first character if the
Guido van Rossum <guido@python.org>
parents: 29864
diff changeset
204 if longest:
400674958025 The fullmodname() function chopped off the first character if the
Guido van Rossum <guido@python.org>
parents: 29864
diff changeset
205 base = path[len(longest) + 1:]
400674958025 The fullmodname() function chopped off the first character if the
Guido van Rossum <guido@python.org>
parents: 29864
diff changeset
206 else:
400674958025 The fullmodname() function chopped off the first character if the
Guido van Rossum <guido@python.org>
parents: 29864
diff changeset
207 base = path
63438
9ee196a7f9a2 Merged revisions 83393,83396,83398,83404-83405,83408 via svnmerge from
Georg Brandl <georg@python.org>
parents: 63133
diff changeset
208 # the drive letter is never part of the module name
9ee196a7f9a2 Merged revisions 83393,83396,83398,83404-83405,83408 via svnmerge from
Georg Brandl <georg@python.org>
parents: 63133
diff changeset
209 drive, base = os.path.splitdrive(base)
31551
a157cc20f1b7 Fix two small bugs: (1) on Windows, pathname munging didn't work
Guido van Rossum <guido@python.org>
parents: 31515
diff changeset
210 base = base.replace(os.sep, ".")
a157cc20f1b7 Fix two small bugs: (1) on Windows, pathname munging didn't work
Guido van Rossum <guido@python.org>
parents: 31515
diff changeset
211 if os.altsep:
a157cc20f1b7 Fix two small bugs: (1) on Windows, pathname munging didn't work
Guido van Rossum <guido@python.org>
parents: 31515
diff changeset
212 base = base.replace(os.altsep, ".")
28713
d08d15a70295 Add helper function to get module name taking packages into account.
Jeremy Hylton <jeremy@alum.mit.edu>
parents: 28712
diff changeset
213 filename, ext = os.path.splitext(base)
63438
9ee196a7f9a2 Merged revisions 83393,83396,83398,83404-83405,83408 via svnmerge from
Georg Brandl <georg@python.org>
parents: 63133
diff changeset
214 return filename.lstrip(".")
28713
d08d15a70295 Add helper function to get module name taking packages into account.
Jeremy Hylton <jeremy@alum.mit.edu>
parents: 28712
diff changeset
215
27933
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
216 class CoverageResults:
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
217 def __init__(self, counts=None, calledfuncs=None, infile=None,
31881
f35b4acc594f Added --trackcalls command line arg to display crude caller/callee
Skip Montanaro <skip@pobox.com>
parents: 31551
diff changeset
218 callers=None, outfile=None):
27933
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
219 self.counts = counts
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
220 if self.counts is None:
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
221 self.counts = {}
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
222 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
223 self.calledfuncs = calledfuncs
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
224 if self.calledfuncs is None:
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
225 self.calledfuncs = {}
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
226 self.calledfuncs = self.calledfuncs.copy()
31881
f35b4acc594f Added --trackcalls command line arg to display crude caller/callee
Skip Montanaro <skip@pobox.com>
parents: 31551
diff changeset
227 self.callers = callers
f35b4acc594f Added --trackcalls command line arg to display crude caller/callee
Skip Montanaro <skip@pobox.com>
parents: 31551
diff changeset
228 if self.callers is None:
f35b4acc594f Added --trackcalls command line arg to display crude caller/callee
Skip Montanaro <skip@pobox.com>
parents: 31551
diff changeset
229 self.callers = {}
f35b4acc594f Added --trackcalls command line arg to display crude caller/callee
Skip Montanaro <skip@pobox.com>
parents: 31551
diff changeset
230 self.callers = self.callers.copy()
27933
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
231 self.infile = infile
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
232 self.outfile = outfile
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
233 if self.infile:
29767
7770b7881346 Patch from Zooko to remove an experimental feature.
Jeremy Hylton <jeremy@alum.mit.edu>
parents: 29620
diff changeset
234 # Try to merge existing counts file.
27933
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
235 try:
31881
f35b4acc594f Added --trackcalls command line arg to display crude caller/callee
Skip Montanaro <skip@pobox.com>
parents: 31551
diff changeset
236 counts, calledfuncs, callers = \
f35b4acc594f Added --trackcalls command line arg to display crude caller/callee
Skip Montanaro <skip@pobox.com>
parents: 31551
diff changeset
237 pickle.load(open(self.infile, 'rb'))
f35b4acc594f Added --trackcalls command line arg to display crude caller/callee
Skip Montanaro <skip@pobox.com>
parents: 31551
diff changeset
238 self.update(self.__class__(counts, calledfuncs, callers))
40587
6b18a095c2e7 SF patch 1631942 by Collin Winter:
Guido van Rossum <guido@python.org>
parents: 39782
diff changeset
239 except (IOError, EOFError, ValueError) as err:
40775
4cdf21a0dbbc Fix most trivially-findable print statements.
Guido van Rossum <guido@python.org>
parents: 40587
diff changeset
240 print(("Skipping counts file %r: %s"
4cdf21a0dbbc Fix most trivially-findable print statements.
Guido van Rossum <guido@python.org>
parents: 40587
diff changeset
241 % (self.infile, err)), file=sys.stderr)
27933
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
242
52690
b43f14c103bb #5656: detect correct encoding of files when reporting coverage in trace.py, and ignore files in the temporary directory when reporting.
Georg Brandl <georg@python.org>
parents: 54437
diff changeset
243 def is_ignored_filename(self, filename):
b43f14c103bb #5656: detect correct encoding of files when reporting coverage in trace.py, and ignore files in the temporary directory when reporting.
Georg Brandl <georg@python.org>
parents: 54437
diff changeset
244 """Return True if the filename does not refer to a file
b43f14c103bb #5656: detect correct encoding of files when reporting coverage in trace.py, and ignore files in the temporary directory when reporting.
Georg Brandl <georg@python.org>
parents: 54437
diff changeset
245 we want to have reported.
b43f14c103bb #5656: detect correct encoding of files when reporting coverage in trace.py, and ignore files in the temporary directory when reporting.
Georg Brandl <georg@python.org>
parents: 54437
diff changeset
246 """
b43f14c103bb #5656: detect correct encoding of files when reporting coverage in trace.py, and ignore files in the temporary directory when reporting.
Georg Brandl <georg@python.org>
parents: 54437
diff changeset
247 return (filename == "<string>" or
b43f14c103bb #5656: detect correct encoding of files when reporting coverage in trace.py, and ignore files in the temporary directory when reporting.
Georg Brandl <georg@python.org>
parents: 54437
diff changeset
248 filename.startswith("<doctest "))
b43f14c103bb #5656: detect correct encoding of files when reporting coverage in trace.py, and ignore files in the temporary directory when reporting.
Georg Brandl <georg@python.org>
parents: 54437
diff changeset
249
27933
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
250 def update(self, other):
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
251 """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
252 counts = self.counts
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
253 calledfuncs = self.calledfuncs
31881
f35b4acc594f Added --trackcalls command line arg to display crude caller/callee
Skip Montanaro <skip@pobox.com>
parents: 31551
diff changeset
254 callers = self.callers
27933
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
255 other_counts = other.counts
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
256 other_calledfuncs = other.calledfuncs
31881
f35b4acc594f Added --trackcalls command line arg to display crude caller/callee
Skip Montanaro <skip@pobox.com>
parents: 31551
diff changeset
257 other_callers = other.callers
27933
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
258
64958
ff5d3e5c4bb5 Issue 9941: Minor code cleanup before implementing the context manager feature:
Alexander Belopolsky <alexander.belopolsky@gmail.com>
parents: 64924
diff changeset
259 for key in other_counts:
28712
d3127b671d36 Holistic refactoring.
Jeremy Hylton <jeremy@alum.mit.edu>
parents: 28089
diff changeset
260 counts[key] = counts.get(key, 0) + other_counts[key]
27933
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
261
64958
ff5d3e5c4bb5 Issue 9941: Minor code cleanup before implementing the context manager feature:
Alexander Belopolsky <alexander.belopolsky@gmail.com>
parents: 64924
diff changeset
262 for key in other_calledfuncs:
27933
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
263 calledfuncs[key] = 1
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
264
64958
ff5d3e5c4bb5 Issue 9941: Minor code cleanup before implementing the context manager feature:
Alexander Belopolsky <alexander.belopolsky@gmail.com>
parents: 64924
diff changeset
265 for key in other_callers:
31881
f35b4acc594f Added --trackcalls command line arg to display crude caller/callee
Skip Montanaro <skip@pobox.com>
parents: 31551
diff changeset
266 callers[key] = 1
f35b4acc594f Added --trackcalls command line arg to display crude caller/callee
Skip Montanaro <skip@pobox.com>
parents: 31551
diff changeset
267
28712
d3127b671d36 Holistic refactoring.
Jeremy Hylton <jeremy@alum.mit.edu>
parents: 28089
diff changeset
268 def write_results(self, show_missing=True, summary=False, coverdir=None):
27933
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
269 """
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
270 @param coverdir
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
271 """
31881
f35b4acc594f Added --trackcalls command line arg to display crude caller/callee
Skip Montanaro <skip@pobox.com>
parents: 31551
diff changeset
272 if self.calledfuncs:
40775
4cdf21a0dbbc Fix most trivially-findable print statements.
Guido van Rossum <guido@python.org>
parents: 40587
diff changeset
273 print()
4cdf21a0dbbc Fix most trivially-findable print statements.
Guido van Rossum <guido@python.org>
parents: 40587
diff changeset
274 print("functions called:")
64958
ff5d3e5c4bb5 Issue 9941: Minor code cleanup before implementing the context manager feature:
Alexander Belopolsky <alexander.belopolsky@gmail.com>
parents: 64924
diff changeset
275 calls = self.calledfuncs
63110
1e261cce4116 Merged revisions 82997 via svnmerge from
Alexander Belopolsky <alexander.belopolsky@gmail.com>
parents: 59034
diff changeset
276 for filename, modulename, funcname in sorted(calls):
40775
4cdf21a0dbbc Fix most trivially-findable print statements.
Guido van Rossum <guido@python.org>
parents: 40587
diff changeset
277 print(("filename: %s, modulename: %s, funcname: %s"
4cdf21a0dbbc Fix most trivially-findable print statements.
Guido van Rossum <guido@python.org>
parents: 40587
diff changeset
278 % (filename, modulename, funcname)))
31881
f35b4acc594f Added --trackcalls command line arg to display crude caller/callee
Skip Montanaro <skip@pobox.com>
parents: 31551
diff changeset
279
f35b4acc594f Added --trackcalls command line arg to display crude caller/callee
Skip Montanaro <skip@pobox.com>
parents: 31551
diff changeset
280 if self.callers:
40775
4cdf21a0dbbc Fix most trivially-findable print statements.
Guido van Rossum <guido@python.org>
parents: 40587
diff changeset
281 print()
4cdf21a0dbbc Fix most trivially-findable print statements.
Guido van Rossum <guido@python.org>
parents: 40587
diff changeset
282 print("calling relationships:")
31881
f35b4acc594f Added --trackcalls command line arg to display crude caller/callee
Skip Montanaro <skip@pobox.com>
parents: 31551
diff changeset
283 lastfile = lastcfile = ""
52690
b43f14c103bb #5656: detect correct encoding of files when reporting coverage in trace.py, and ignore files in the temporary directory when reporting.
Georg Brandl <georg@python.org>
parents: 54437
diff changeset
284 for ((pfile, pmod, pfunc), (cfile, cmod, cfunc)) \
64958
ff5d3e5c4bb5 Issue 9941: Minor code cleanup before implementing the context manager feature:
Alexander Belopolsky <alexander.belopolsky@gmail.com>
parents: 64924
diff changeset
285 in sorted(self.callers):
31881
f35b4acc594f Added --trackcalls command line arg to display crude caller/callee
Skip Montanaro <skip@pobox.com>
parents: 31551
diff changeset
286 if pfile != lastfile:
40775
4cdf21a0dbbc Fix most trivially-findable print statements.
Guido van Rossum <guido@python.org>
parents: 40587
diff changeset
287 print()
4cdf21a0dbbc Fix most trivially-findable print statements.
Guido van Rossum <guido@python.org>
parents: 40587
diff changeset
288 print("***", pfile, "***")
31881
f35b4acc594f Added --trackcalls command line arg to display crude caller/callee
Skip Montanaro <skip@pobox.com>
parents: 31551
diff changeset
289 lastfile = pfile
f35b4acc594f Added --trackcalls command line arg to display crude caller/callee
Skip Montanaro <skip@pobox.com>
parents: 31551
diff changeset
290 lastcfile = ""
f35b4acc594f Added --trackcalls command line arg to display crude caller/callee
Skip Montanaro <skip@pobox.com>
parents: 31551
diff changeset
291 if cfile != pfile and lastcfile != cfile:
40775
4cdf21a0dbbc Fix most trivially-findable print statements.
Guido van Rossum <guido@python.org>
parents: 40587
diff changeset
292 print(" -->", cfile)
31881
f35b4acc594f Added --trackcalls command line arg to display crude caller/callee
Skip Montanaro <skip@pobox.com>
parents: 31551
diff changeset
293 lastcfile = cfile
40775
4cdf21a0dbbc Fix most trivially-findable print statements.
Guido van Rossum <guido@python.org>
parents: 40587
diff changeset
294 print(" %s.%s -> %s.%s" % (pmod, pfunc, cmod, cfunc))
27933
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
295
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
296 # turn the counts data ("(filename, lineno) = count") into something
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
297 # accessible on a per-file basis
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
298 per_file = {}
64958
ff5d3e5c4bb5 Issue 9941: Minor code cleanup before implementing the context manager feature:
Alexander Belopolsky <alexander.belopolsky@gmail.com>
parents: 64924
diff changeset
299 for filename, lineno in self.counts:
28712
d3127b671d36 Holistic refactoring.
Jeremy Hylton <jeremy@alum.mit.edu>
parents: 28089
diff changeset
300 lines_hit = per_file[filename] = per_file.get(filename, {})
d3127b671d36 Holistic refactoring.
Jeremy Hylton <jeremy@alum.mit.edu>
parents: 28089
diff changeset
301 lines_hit[lineno] = self.counts[(filename, lineno)]
27933
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
302
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
303 # accumulate summary info, if needed
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
304 sums = {}
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
305
40809
f340cb045bf9 - PEP 3106: dict.iterkeys(), .iteritems(), .itervalues() are now gone;
Guido van Rossum <guido@python.org>
parents: 40775
diff changeset
306 for filename, count in per_file.items():
52690
b43f14c103bb #5656: detect correct encoding of files when reporting coverage in trace.py, and ignore files in the temporary directory when reporting.
Georg Brandl <georg@python.org>
parents: 54437
diff changeset
307 if self.is_ignored_filename(filename):
44025
ebc7389297d6 Merged revisions 59107-59186 via svnmerge from
Guido van Rossum <guido@python.org>
parents: 42552
diff changeset
308 continue
27933
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
309
39442
654c380cf8b9 Merged revisions 46753-51188 via svnmerge from
Thomas Wouters <thomas@python.org>
parents: 38453
diff changeset
310 if filename.endswith((".pyc", ".pyo")):
27933
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
311 filename = filename[:-1]
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
312
28712
d3127b671d36 Holistic refactoring.
Jeremy Hylton <jeremy@alum.mit.edu>
parents: 28089
diff changeset
313 if coverdir is None:
d3127b671d36 Holistic refactoring.
Jeremy Hylton <jeremy@alum.mit.edu>
parents: 28089
diff changeset
314 dir = os.path.dirname(os.path.abspath(filename))
66308
4f29aa8f9f54 Issue #10371: Deprecated undocumented functions in the trace module.
Alexander Belopolsky <alexander.belopolsky@gmail.com>
parents: 66097
diff changeset
315 modulename = _modname(filename)
27933
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
316 else:
28712
d3127b671d36 Holistic refactoring.
Jeremy Hylton <jeremy@alum.mit.edu>
parents: 28089
diff changeset
317 dir = coverdir
d3127b671d36 Holistic refactoring.
Jeremy Hylton <jeremy@alum.mit.edu>
parents: 28089
diff changeset
318 if not os.path.exists(dir):
d3127b671d36 Holistic refactoring.
Jeremy Hylton <jeremy@alum.mit.edu>
parents: 28089
diff changeset
319 os.makedirs(dir)
66308
4f29aa8f9f54 Issue #10371: Deprecated undocumented functions in the trace module.
Alexander Belopolsky <alexander.belopolsky@gmail.com>
parents: 66097
diff changeset
320 modulename = _fullmodname(filename)
27933
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
321
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
322 # 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
323 # 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
324 if show_missing:
66308
4f29aa8f9f54 Issue #10371: Deprecated undocumented functions in the trace module.
Alexander Belopolsky <alexander.belopolsky@gmail.com>
parents: 66097
diff changeset
325 lnotab = _find_executable_linenos(filename)
27933
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
326 else:
28712
d3127b671d36 Holistic refactoring.
Jeremy Hylton <jeremy@alum.mit.edu>
parents: 28089
diff changeset
327 lnotab = {}
27933
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
328
28712
d3127b671d36 Holistic refactoring.
Jeremy Hylton <jeremy@alum.mit.edu>
parents: 28089
diff changeset
329 source = linecache.getlines(filename)
d3127b671d36 Holistic refactoring.
Jeremy Hylton <jeremy@alum.mit.edu>
parents: 28089
diff changeset
330 coverpath = os.path.join(dir, modulename + ".cover")
66068
80473ad697ac Issue #10329: The trace module writes reports using the input Python script
Victor Stinner <victor.stinner@haypocalc.com>
parents: 65999
diff changeset
331 with open(filename, 'rb') as fp:
80473ad697ac Issue #10329: The trace module writes reports using the input Python script
Victor Stinner <victor.stinner@haypocalc.com>
parents: 65999
diff changeset
332 encoding, _ = tokenize.detect_encoding(fp.readline)
28712
d3127b671d36 Holistic refactoring.
Jeremy Hylton <jeremy@alum.mit.edu>
parents: 28089
diff changeset
333 n_hits, n_lines = self.write_results_file(coverpath, source,
66068
80473ad697ac Issue #10329: The trace module writes reports using the input Python script
Victor Stinner <victor.stinner@haypocalc.com>
parents: 65999
diff changeset
334 lnotab, count, encoding)
27933
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
335 if summary and n_lines:
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
336 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
337 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
338
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
339 if summary and sums:
40775
4cdf21a0dbbc Fix most trivially-findable print statements.
Guido van Rossum <guido@python.org>
parents: 40587
diff changeset
340 print("lines cov% module (path)")
64958
ff5d3e5c4bb5 Issue 9941: Minor code cleanup before implementing the context manager feature:
Alexander Belopolsky <alexander.belopolsky@gmail.com>
parents: 64924
diff changeset
341 for m in sorted(sums):
27933
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
342 n_lines, percent, modulename, filename = sums[m]
40775
4cdf21a0dbbc Fix most trivially-findable print statements.
Guido van Rossum <guido@python.org>
parents: 40587
diff changeset
343 print("%5d %3d%% %s (%s)" % sums[m])
27933
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
344
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
345 if self.outfile:
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
346 # 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
347 try:
31881
f35b4acc594f Added --trackcalls command line arg to display crude caller/callee
Skip Montanaro <skip@pobox.com>
parents: 31551
diff changeset
348 pickle.dump((self.counts, self.calledfuncs, self.callers),
30692
a3e86665f245 Open results files, which contain binary pickles, in binary mode.
Jeremy Hylton <jeremy@alum.mit.edu>
parents: 30668
diff changeset
349 open(self.outfile, 'wb'), 1)
40587
6b18a095c2e7 SF patch 1631942 by Collin Winter:
Guido van Rossum <guido@python.org>
parents: 39782
diff changeset
350 except IOError as err:
40775
4cdf21a0dbbc Fix most trivially-findable print statements.
Guido van Rossum <guido@python.org>
parents: 40587
diff changeset
351 print("Can't save counts files because %s" % err, file=sys.stderr)
28712
d3127b671d36 Holistic refactoring.
Jeremy Hylton <jeremy@alum.mit.edu>
parents: 28089
diff changeset
352
66068
80473ad697ac Issue #10329: The trace module writes reports using the input Python script
Victor Stinner <victor.stinner@haypocalc.com>
parents: 65999
diff changeset
353 def write_results_file(self, path, lines, lnotab, lines_hit, encoding=None):
28712
d3127b671d36 Holistic refactoring.
Jeremy Hylton <jeremy@alum.mit.edu>
parents: 28089
diff changeset
354 """Return a coverage results file in path."""
d3127b671d36 Holistic refactoring.
Jeremy Hylton <jeremy@alum.mit.edu>
parents: 28089
diff changeset
355
d3127b671d36 Holistic refactoring.
Jeremy Hylton <jeremy@alum.mit.edu>
parents: 28089
diff changeset
356 try:
66068
80473ad697ac Issue #10329: The trace module writes reports using the input Python script
Victor Stinner <victor.stinner@haypocalc.com>
parents: 65999
diff changeset
357 outfile = open(path, "w", encoding=encoding)
40587
6b18a095c2e7 SF patch 1631942 by Collin Winter:
Guido van Rossum <guido@python.org>
parents: 39782
diff changeset
358 except IOError as err:
40775
4cdf21a0dbbc Fix most trivially-findable print statements.
Guido van Rossum <guido@python.org>
parents: 40587
diff changeset
359 print(("trace: Could not open %r for writing: %s"
4cdf21a0dbbc Fix most trivially-findable print statements.
Guido van Rossum <guido@python.org>
parents: 40587
diff changeset
360 "- skipping" % (path, err)), file=sys.stderr)
31551
a157cc20f1b7 Fix two small bugs: (1) on Windows, pathname munging didn't work
Guido van Rossum <guido@python.org>
parents: 31515
diff changeset
361 return 0, 0
27933
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
362
28712
d3127b671d36 Holistic refactoring.
Jeremy Hylton <jeremy@alum.mit.edu>
parents: 28089
diff changeset
363 n_lines = 0
d3127b671d36 Holistic refactoring.
Jeremy Hylton <jeremy@alum.mit.edu>
parents: 28089
diff changeset
364 n_hits = 0
64958
ff5d3e5c4bb5 Issue 9941: Minor code cleanup before implementing the context manager feature:
Alexander Belopolsky <alexander.belopolsky@gmail.com>
parents: 64924
diff changeset
365 for lineno, line in enumerate(lines, 1):
28712
d3127b671d36 Holistic refactoring.
Jeremy Hylton <jeremy@alum.mit.edu>
parents: 28089
diff changeset
366 # do the blank/comment match to try to mark more lines
d3127b671d36 Holistic refactoring.
Jeremy Hylton <jeremy@alum.mit.edu>
parents: 28089
diff changeset
367 # (help the reader find stuff that hasn't been covered)
d3127b671d36 Holistic refactoring.
Jeremy Hylton <jeremy@alum.mit.edu>
parents: 28089
diff changeset
368 if lineno in lines_hit:
d3127b671d36 Holistic refactoring.
Jeremy Hylton <jeremy@alum.mit.edu>
parents: 28089
diff changeset
369 outfile.write("%5d: " % lines_hit[lineno])
d3127b671d36 Holistic refactoring.
Jeremy Hylton <jeremy@alum.mit.edu>
parents: 28089
diff changeset
370 n_hits += 1
d3127b671d36 Holistic refactoring.
Jeremy Hylton <jeremy@alum.mit.edu>
parents: 28089
diff changeset
371 n_lines += 1
d3127b671d36 Holistic refactoring.
Jeremy Hylton <jeremy@alum.mit.edu>
parents: 28089
diff changeset
372 elif rx_blank.match(line):
29864
84bccbeab5fa Make the prefix 7 spaces long, so the lines are properly aligned.
Walter Dörwald <walter@livinglogic.de>
parents: 29767
diff changeset
373 outfile.write(" ")
28712
d3127b671d36 Holistic refactoring.
Jeremy Hylton <jeremy@alum.mit.edu>
parents: 28089
diff changeset
374 else:
d3127b671d36 Holistic refactoring.
Jeremy Hylton <jeremy@alum.mit.edu>
parents: 28089
diff changeset
375 # lines preceded by no marks weren't hit
d3127b671d36 Holistic refactoring.
Jeremy Hylton <jeremy@alum.mit.edu>
parents: 28089
diff changeset
376 # Highlight them if so indicated, unless the line contains
d3127b671d36 Holistic refactoring.
Jeremy Hylton <jeremy@alum.mit.edu>
parents: 28089
diff changeset
377 # #pragma: NO COVER
64958
ff5d3e5c4bb5 Issue 9941: Minor code cleanup before implementing the context manager feature:
Alexander Belopolsky <alexander.belopolsky@gmail.com>
parents: 64924
diff changeset
378 if lineno in lnotab and not PRAGMA_NOCOVER in line:
28712
d3127b671d36 Holistic refactoring.
Jeremy Hylton <jeremy@alum.mit.edu>
parents: 28089
diff changeset
379 outfile.write(">>>>>> ")
29595
8911ca88981a Enable tracing of multi-threaded applications.
Jeremy Hylton <jeremy@alum.mit.edu>
parents: 28774
diff changeset
380 n_lines += 1
28712
d3127b671d36 Holistic refactoring.
Jeremy Hylton <jeremy@alum.mit.edu>
parents: 28089
diff changeset
381 else:
d3127b671d36 Holistic refactoring.
Jeremy Hylton <jeremy@alum.mit.edu>
parents: 28089
diff changeset
382 outfile.write(" ")
64958
ff5d3e5c4bb5 Issue 9941: Minor code cleanup before implementing the context manager feature:
Alexander Belopolsky <alexander.belopolsky@gmail.com>
parents: 64924
diff changeset
383 outfile.write(line.expandtabs(8))
28712
d3127b671d36 Holistic refactoring.
Jeremy Hylton <jeremy@alum.mit.edu>
parents: 28089
diff changeset
384 outfile.close()
d3127b671d36 Holistic refactoring.
Jeremy Hylton <jeremy@alum.mit.edu>
parents: 28089
diff changeset
385
d3127b671d36 Holistic refactoring.
Jeremy Hylton <jeremy@alum.mit.edu>
parents: 28089
diff changeset
386 return n_hits, n_lines
d3127b671d36 Holistic refactoring.
Jeremy Hylton <jeremy@alum.mit.edu>
parents: 28089
diff changeset
387
66308
4f29aa8f9f54 Issue #10371: Deprecated undocumented functions in the trace module.
Alexander Belopolsky <alexander.belopolsky@gmail.com>
parents: 66097
diff changeset
388 def _find_lines_from_code(code, strs):
28712
d3127b671d36 Holistic refactoring.
Jeremy Hylton <jeremy@alum.mit.edu>
parents: 28089
diff changeset
389 """Return dict where keys are lines in the line number table."""
27933
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
390 linenos = {}
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
391
64925
235044b39a92 Merged revisions 84994 via svnmerge from
Alexander Belopolsky <alexander.belopolsky@gmail.com>
parents: 64717
diff changeset
392 for _, lineno in dis.findlinestarts(code):
28712
d3127b671d36 Holistic refactoring.
Jeremy Hylton <jeremy@alum.mit.edu>
parents: 28089
diff changeset
393 if lineno not in strs:
d3127b671d36 Holistic refactoring.
Jeremy Hylton <jeremy@alum.mit.edu>
parents: 28089
diff changeset
394 linenos[lineno] = 1
27933
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
395
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
396 return linenos
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
397
66308
4f29aa8f9f54 Issue #10371: Deprecated undocumented functions in the trace module.
Alexander Belopolsky <alexander.belopolsky@gmail.com>
parents: 66097
diff changeset
398 def _find_lines(code, strs):
28712
d3127b671d36 Holistic refactoring.
Jeremy Hylton <jeremy@alum.mit.edu>
parents: 28089
diff changeset
399 """Return lineno dict for all code objects reachable from code."""
27933
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
400 # get all of the lineno information from the code of this scope level
66308
4f29aa8f9f54 Issue #10371: Deprecated undocumented functions in the trace module.
Alexander Belopolsky <alexander.belopolsky@gmail.com>
parents: 66097
diff changeset
401 linenos = _find_lines_from_code(code, strs)
27933
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
402
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
403 # 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
404 for c in code.co_consts:
64717
7dbfc7ef39f2 Merged revisions 84780-84781 via svnmerge from
Alexander Belopolsky <alexander.belopolsky@gmail.com>
parents: 63438
diff changeset
405 if inspect.iscode(c):
27933
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
406 # find another code object, so recurse into it
66308
4f29aa8f9f54 Issue #10371: Deprecated undocumented functions in the trace module.
Alexander Belopolsky <alexander.belopolsky@gmail.com>
parents: 66097
diff changeset
407 linenos.update(_find_lines(c, strs))
27933
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
408 return linenos
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
409
66308
4f29aa8f9f54 Issue #10371: Deprecated undocumented functions in the trace module.
Alexander Belopolsky <alexander.belopolsky@gmail.com>
parents: 66097
diff changeset
410 def _find_strings(filename, encoding=None):
28712
d3127b671d36 Holistic refactoring.
Jeremy Hylton <jeremy@alum.mit.edu>
parents: 28089
diff changeset
411 """Return a dict of possible docstring positions.
d3127b671d36 Holistic refactoring.
Jeremy Hylton <jeremy@alum.mit.edu>
parents: 28089
diff changeset
412
d3127b671d36 Holistic refactoring.
Jeremy Hylton <jeremy@alum.mit.edu>
parents: 28089
diff changeset
413 The dict maps line numbers to strings. There is an entry for
d3127b671d36 Holistic refactoring.
Jeremy Hylton <jeremy@alum.mit.edu>
parents: 28089
diff changeset
414 line that contains only a string or a part of a triple-quoted
d3127b671d36 Holistic refactoring.
Jeremy Hylton <jeremy@alum.mit.edu>
parents: 28089
diff changeset
415 string.
d3127b671d36 Holistic refactoring.
Jeremy Hylton <jeremy@alum.mit.edu>
parents: 28089
diff changeset
416 """
d3127b671d36 Holistic refactoring.
Jeremy Hylton <jeremy@alum.mit.edu>
parents: 28089
diff changeset
417 d = {}
d3127b671d36 Holistic refactoring.
Jeremy Hylton <jeremy@alum.mit.edu>
parents: 28089
diff changeset
418 # If the first token is a string, then it's the module docstring.
d3127b671d36 Holistic refactoring.
Jeremy Hylton <jeremy@alum.mit.edu>
parents: 28089
diff changeset
419 # Add this special case so that the test in the loop passes.
d3127b671d36 Holistic refactoring.
Jeremy Hylton <jeremy@alum.mit.edu>
parents: 28089
diff changeset
420 prev_ttype = token.INDENT
65781
1954c8e31cb3 close files correctly
Benjamin Peterson <benjamin@python.org>
parents: 65526
diff changeset
421 with open(filename, encoding=encoding) as f:
1954c8e31cb3 close files correctly
Benjamin Peterson <benjamin@python.org>
parents: 65526
diff changeset
422 tok = tokenize.generate_tokens(f.readline)
1954c8e31cb3 close files correctly
Benjamin Peterson <benjamin@python.org>
parents: 65526
diff changeset
423 for ttype, tstr, start, end, line in tok:
1954c8e31cb3 close files correctly
Benjamin Peterson <benjamin@python.org>
parents: 65526
diff changeset
424 if ttype == token.STRING:
1954c8e31cb3 close files correctly
Benjamin Peterson <benjamin@python.org>
parents: 65526
diff changeset
425 if prev_ttype == token.INDENT:
1954c8e31cb3 close files correctly
Benjamin Peterson <benjamin@python.org>
parents: 65526
diff changeset
426 sline, scol = start
1954c8e31cb3 close files correctly
Benjamin Peterson <benjamin@python.org>
parents: 65526
diff changeset
427 eline, ecol = end
1954c8e31cb3 close files correctly
Benjamin Peterson <benjamin@python.org>
parents: 65526
diff changeset
428 for i in range(sline, eline + 1):
1954c8e31cb3 close files correctly
Benjamin Peterson <benjamin@python.org>
parents: 65526
diff changeset
429 d[i] = 1
1954c8e31cb3 close files correctly
Benjamin Peterson <benjamin@python.org>
parents: 65526
diff changeset
430 prev_ttype = ttype
28712
d3127b671d36 Holistic refactoring.
Jeremy Hylton <jeremy@alum.mit.edu>
parents: 28089
diff changeset
431 return d
d3127b671d36 Holistic refactoring.
Jeremy Hylton <jeremy@alum.mit.edu>
parents: 28089
diff changeset
432
66308
4f29aa8f9f54 Issue #10371: Deprecated undocumented functions in the trace module.
Alexander Belopolsky <alexander.belopolsky@gmail.com>
parents: 66097
diff changeset
433 def _find_executable_linenos(filename):
28712
d3127b671d36 Holistic refactoring.
Jeremy Hylton <jeremy@alum.mit.edu>
parents: 28089
diff changeset
434 """Return dict where keys are line numbers in the line number table."""
d3127b671d36 Holistic refactoring.
Jeremy Hylton <jeremy@alum.mit.edu>
parents: 28089
diff changeset
435 try:
66097
306339c31758 Issue #10335: Add tokenize.open(), detect the file encoding using
Victor Stinner <victor.stinner@haypocalc.com>
parents: 66088
diff changeset
436 with tokenize.open(filename) as f:
65781
1954c8e31cb3 close files correctly
Benjamin Peterson <benjamin@python.org>
parents: 65526
diff changeset
437 prog = f.read()
66097
306339c31758 Issue #10335: Add tokenize.open(), detect the file encoding using
Victor Stinner <victor.stinner@haypocalc.com>
parents: 66088
diff changeset
438 encoding = f.encoding
40587
6b18a095c2e7 SF patch 1631942 by Collin Winter:
Guido van Rossum <guido@python.org>
parents: 39782
diff changeset
439 except IOError as err:
40775
4cdf21a0dbbc Fix most trivially-findable print statements.
Guido van Rossum <guido@python.org>
parents: 40587
diff changeset
440 print(("Not printing coverage data for %r: %s"
4cdf21a0dbbc Fix most trivially-findable print statements.
Guido van Rossum <guido@python.org>
parents: 40587
diff changeset
441 % (filename, err)), file=sys.stderr)
28712
d3127b671d36 Holistic refactoring.
Jeremy Hylton <jeremy@alum.mit.edu>
parents: 28089
diff changeset
442 return {}
d3127b671d36 Holistic refactoring.
Jeremy Hylton <jeremy@alum.mit.edu>
parents: 28089
diff changeset
443 code = compile(prog, filename, "exec")
66308
4f29aa8f9f54 Issue #10371: Deprecated undocumented functions in the trace module.
Alexander Belopolsky <alexander.belopolsky@gmail.com>
parents: 66097
diff changeset
444 strs = _find_strings(filename, encoding)
4f29aa8f9f54 Issue #10371: Deprecated undocumented functions in the trace module.
Alexander Belopolsky <alexander.belopolsky@gmail.com>
parents: 66097
diff changeset
445 return _find_lines(code, strs)
27933
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 class Trace:
31881
f35b4acc594f Added --trackcalls command line arg to display crude caller/callee
Skip Montanaro <skip@pobox.com>
parents: 31551
diff changeset
448 def __init__(self, count=1, trace=1, countfuncs=0, countcallers=0,
45548
8c3c5b4eef55 Merged revisions 61038,61042-61045,61047,61050,61053,61055-61056,61061-61062,61066,61068,61070,61081-61095 via svnmerge from
Christian Heimes <christian@cheimes.de>
parents: 44782
diff changeset
449 ignoremods=(), ignoredirs=(), infile=None, outfile=None,
8c3c5b4eef55 Merged revisions 61038,61042-61045,61047,61050,61053,61055-61056,61061-61062,61066,61068,61070,61081-61095 via svnmerge from
Christian Heimes <christian@cheimes.de>
parents: 44782
diff changeset
450 timing=False):
27933
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
451 """
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
452 @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
453 line is executed
27933
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
454 @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
455 being counted
27933
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
456 @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
457 (filename, modulename, funcname,) for functions
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
458 that were called at least once; This overrides
27963
226463445697 Whitespace normalization.
Tim Peters <tim.peters@gmail.com>
parents: 27933
diff changeset
459 `count' and `trace'
27933
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
460 @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
461 @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
462 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
463 @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
464 added into the results
27933
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
465 @param outfile file in which to write the results
45548
8c3c5b4eef55 Merged revisions 61038,61042-61045,61047,61050,61053,61055-61056,61061-61062,61066,61068,61070,61081-61095 via svnmerge from
Christian Heimes <christian@cheimes.de>
parents: 44782
diff changeset
466 @param timing true iff timing information be displayed
27933
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
467 """
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
468 self.infile = infile
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
469 self.outfile = outfile
66308
4f29aa8f9f54 Issue #10371: Deprecated undocumented functions in the trace module.
Alexander Belopolsky <alexander.belopolsky@gmail.com>
parents: 66097
diff changeset
470 self.ignore = _Ignore(ignoremods, ignoredirs)
27933
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
471 self.counts = {} # keys are (filename, linenumber)
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
472 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
473 self.donothing = 0
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
474 self.trace = trace
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
475 self._calledfuncs = {}
31881
f35b4acc594f Added --trackcalls command line arg to display crude caller/callee
Skip Montanaro <skip@pobox.com>
parents: 31551
diff changeset
476 self._callers = {}
31892
081972ae2f84 Refactor common code out of globaltrace_trackcallers() and
Skip Montanaro <skip@pobox.com>
parents: 31881
diff changeset
477 self._caller_cache = {}
45548
8c3c5b4eef55 Merged revisions 61038,61042-61045,61047,61050,61053,61055-61056,61061-61062,61066,61068,61070,61081-61095 via svnmerge from
Christian Heimes <christian@cheimes.de>
parents: 44782
diff changeset
478 self.start_time = None
8c3c5b4eef55 Merged revisions 61038,61042-61045,61047,61050,61053,61055-61056,61061-61062,61066,61068,61070,61081-61095 via svnmerge from
Christian Heimes <christian@cheimes.de>
parents: 44782
diff changeset
479 if timing:
8c3c5b4eef55 Merged revisions 61038,61042-61045,61047,61050,61053,61055-61056,61061-61062,61066,61068,61070,61081-61095 via svnmerge from
Christian Heimes <christian@cheimes.de>
parents: 44782
diff changeset
480 self.start_time = time.time()
31881
f35b4acc594f Added --trackcalls command line arg to display crude caller/callee
Skip Montanaro <skip@pobox.com>
parents: 31551
diff changeset
481 if countcallers:
f35b4acc594f Added --trackcalls command line arg to display crude caller/callee
Skip Montanaro <skip@pobox.com>
parents: 31551
diff changeset
482 self.globaltrace = self.globaltrace_trackcallers
f35b4acc594f Added --trackcalls command line arg to display crude caller/callee
Skip Montanaro <skip@pobox.com>
parents: 31551
diff changeset
483 elif countfuncs:
27933
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
484 self.globaltrace = self.globaltrace_countfuncs
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
485 elif trace and count:
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
486 self.globaltrace = self.globaltrace_lt
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
487 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
488 elif trace:
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
489 self.globaltrace = self.globaltrace_lt
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
490 self.localtrace = self.localtrace_trace
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
491 elif count:
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
492 self.globaltrace = self.globaltrace_lt
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
493 self.localtrace = self.localtrace_count
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
494 else:
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
495 # Ahem -- do nothing? Okay.
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
496 self.donothing = 1
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
497
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
498 def run(self, cmd):
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
499 import __main__
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
500 dict = __main__.__dict__
64958
ff5d3e5c4bb5 Issue 9941: Minor code cleanup before implementing the context manager feature:
Alexander Belopolsky <alexander.belopolsky@gmail.com>
parents: 64924
diff changeset
501 self.runctx(cmd, dict, dict)
27933
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
502
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
503 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
504 if globals is None: globals = {}
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
505 if locals is None: locals = {}
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
506 if not self.donothing:
66001
8911f475ad3d Merged revisions 86229 via svnmerge from
Alexander Belopolsky <alexander.belopolsky@gmail.com>
parents: 65115
diff changeset
507 _settrace(self.globaltrace)
27933
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
508 try:
39782
c94f37054f19 Patch #1550800: make exec a function.
Georg Brandl <georg@python.org>
parents: 39610
diff changeset
509 exec(cmd, globals, locals)
27933
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
510 finally:
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
511 if not self.donothing:
66001
8911f475ad3d Merged revisions 86229 via svnmerge from
Alexander Belopolsky <alexander.belopolsky@gmail.com>
parents: 65115
diff changeset
512 _unsettrace()
27933
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
513
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
514 def runfunc(self, func, *args, **kw):
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
515 result = None
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
516 if not self.donothing:
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
517 sys.settrace(self.globaltrace)
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
518 try:
28089
14f65d661084 Get rid of many apply() calls.
Guido van Rossum <guido@python.org>
parents: 27963
diff changeset
519 result = func(*args, **kw)
27933
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
520 finally:
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
521 if not self.donothing:
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
522 sys.settrace(None)
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
523 return result
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
524
31892
081972ae2f84 Refactor common code out of globaltrace_trackcallers() and
Skip Montanaro <skip@pobox.com>
parents: 31881
diff changeset
525 def file_module_function_of(self, frame):
081972ae2f84 Refactor common code out of globaltrace_trackcallers() and
Skip Montanaro <skip@pobox.com>
parents: 31881
diff changeset
526 code = frame.f_code
081972ae2f84 Refactor common code out of globaltrace_trackcallers() and
Skip Montanaro <skip@pobox.com>
parents: 31881
diff changeset
527 filename = code.co_filename
081972ae2f84 Refactor common code out of globaltrace_trackcallers() and
Skip Montanaro <skip@pobox.com>
parents: 31881
diff changeset
528 if filename:
66308
4f29aa8f9f54 Issue #10371: Deprecated undocumented functions in the trace module.
Alexander Belopolsky <alexander.belopolsky@gmail.com>
parents: 66097
diff changeset
529 modulename = _modname(filename)
31892
081972ae2f84 Refactor common code out of globaltrace_trackcallers() and
Skip Montanaro <skip@pobox.com>
parents: 31881
diff changeset
530 else:
081972ae2f84 Refactor common code out of globaltrace_trackcallers() and
Skip Montanaro <skip@pobox.com>
parents: 31881
diff changeset
531 modulename = None
081972ae2f84 Refactor common code out of globaltrace_trackcallers() and
Skip Montanaro <skip@pobox.com>
parents: 31881
diff changeset
532
081972ae2f84 Refactor common code out of globaltrace_trackcallers() and
Skip Montanaro <skip@pobox.com>
parents: 31881
diff changeset
533 funcname = code.co_name
081972ae2f84 Refactor common code out of globaltrace_trackcallers() and
Skip Montanaro <skip@pobox.com>
parents: 31881
diff changeset
534 clsname = None
081972ae2f84 Refactor common code out of globaltrace_trackcallers() and
Skip Montanaro <skip@pobox.com>
parents: 31881
diff changeset
535 if code in self._caller_cache:
081972ae2f84 Refactor common code out of globaltrace_trackcallers() and
Skip Montanaro <skip@pobox.com>
parents: 31881
diff changeset
536 if self._caller_cache[code] is not None:
081972ae2f84 Refactor common code out of globaltrace_trackcallers() and
Skip Montanaro <skip@pobox.com>
parents: 31881
diff changeset
537 clsname = self._caller_cache[code]
081972ae2f84 Refactor common code out of globaltrace_trackcallers() and
Skip Montanaro <skip@pobox.com>
parents: 31881
diff changeset
538 else:
081972ae2f84 Refactor common code out of globaltrace_trackcallers() and
Skip Montanaro <skip@pobox.com>
parents: 31881
diff changeset
539 self._caller_cache[code] = None
081972ae2f84 Refactor common code out of globaltrace_trackcallers() and
Skip Montanaro <skip@pobox.com>
parents: 31881
diff changeset
540 ## use of gc.get_referrers() was suggested by Michael Hudson
081972ae2f84 Refactor common code out of globaltrace_trackcallers() and
Skip Montanaro <skip@pobox.com>
parents: 31881
diff changeset
541 # all functions which refer to this code object
081972ae2f84 Refactor common code out of globaltrace_trackcallers() and
Skip Montanaro <skip@pobox.com>
parents: 31881
diff changeset
542 funcs = [f for f in gc.get_referrers(code)
64717
7dbfc7ef39f2 Merged revisions 84780-84781 via svnmerge from
Alexander Belopolsky <alexander.belopolsky@gmail.com>
parents: 63438
diff changeset
543 if inspect.isfunction(f)]
31892
081972ae2f84 Refactor common code out of globaltrace_trackcallers() and
Skip Montanaro <skip@pobox.com>
parents: 31881
diff changeset
544 # require len(func) == 1 to avoid ambiguity caused by calls to
081972ae2f84 Refactor common code out of globaltrace_trackcallers() and
Skip Montanaro <skip@pobox.com>
parents: 31881
diff changeset
545 # new.function(): "In the face of ambiguity, refuse the
081972ae2f84 Refactor common code out of globaltrace_trackcallers() and
Skip Montanaro <skip@pobox.com>
parents: 31881
diff changeset
546 # temptation to guess."
081972ae2f84 Refactor common code out of globaltrace_trackcallers() and
Skip Montanaro <skip@pobox.com>
parents: 31881
diff changeset
547 if len(funcs) == 1:
081972ae2f84 Refactor common code out of globaltrace_trackcallers() and
Skip Montanaro <skip@pobox.com>
parents: 31881
diff changeset
548 dicts = [d for d in gc.get_referrers(funcs[0])
081972ae2f84 Refactor common code out of globaltrace_trackcallers() and
Skip Montanaro <skip@pobox.com>
parents: 31881
diff changeset
549 if isinstance(d, dict)]
081972ae2f84 Refactor common code out of globaltrace_trackcallers() and
Skip Montanaro <skip@pobox.com>
parents: 31881
diff changeset
550 if len(dicts) == 1:
081972ae2f84 Refactor common code out of globaltrace_trackcallers() and
Skip Montanaro <skip@pobox.com>
parents: 31881
diff changeset
551 classes = [c for c in gc.get_referrers(dicts[0])
081972ae2f84 Refactor common code out of globaltrace_trackcallers() and
Skip Montanaro <skip@pobox.com>
parents: 31881
diff changeset
552 if hasattr(c, "__bases__")]
081972ae2f84 Refactor common code out of globaltrace_trackcallers() and
Skip Montanaro <skip@pobox.com>
parents: 31881
diff changeset
553 if len(classes) == 1:
081972ae2f84 Refactor common code out of globaltrace_trackcallers() and
Skip Montanaro <skip@pobox.com>
parents: 31881
diff changeset
554 # ditto for new.classobj()
64717
7dbfc7ef39f2 Merged revisions 84780-84781 via svnmerge from
Alexander Belopolsky <alexander.belopolsky@gmail.com>
parents: 63438
diff changeset
555 clsname = classes[0].__name__
31892
081972ae2f84 Refactor common code out of globaltrace_trackcallers() and
Skip Montanaro <skip@pobox.com>
parents: 31881
diff changeset
556 # cache the result - assumption is that new.* is
081972ae2f84 Refactor common code out of globaltrace_trackcallers() and
Skip Montanaro <skip@pobox.com>
parents: 31881
diff changeset
557 # not called later to disturb this relationship
081972ae2f84 Refactor common code out of globaltrace_trackcallers() and
Skip Montanaro <skip@pobox.com>
parents: 31881
diff changeset
558 # _caller_cache could be flushed if functions in
081972ae2f84 Refactor common code out of globaltrace_trackcallers() and
Skip Montanaro <skip@pobox.com>
parents: 31881
diff changeset
559 # the new module get called.
081972ae2f84 Refactor common code out of globaltrace_trackcallers() and
Skip Montanaro <skip@pobox.com>
parents: 31881
diff changeset
560 self._caller_cache[code] = clsname
081972ae2f84 Refactor common code out of globaltrace_trackcallers() and
Skip Montanaro <skip@pobox.com>
parents: 31881
diff changeset
561 if clsname is not None:
081972ae2f84 Refactor common code out of globaltrace_trackcallers() and
Skip Montanaro <skip@pobox.com>
parents: 31881
diff changeset
562 funcname = "%s.%s" % (clsname, funcname)
081972ae2f84 Refactor common code out of globaltrace_trackcallers() and
Skip Montanaro <skip@pobox.com>
parents: 31881
diff changeset
563
081972ae2f84 Refactor common code out of globaltrace_trackcallers() and
Skip Montanaro <skip@pobox.com>
parents: 31881
diff changeset
564 return filename, modulename, funcname
081972ae2f84 Refactor common code out of globaltrace_trackcallers() and
Skip Montanaro <skip@pobox.com>
parents: 31881
diff changeset
565
31881
f35b4acc594f Added --trackcalls command line arg to display crude caller/callee
Skip Montanaro <skip@pobox.com>
parents: 31551
diff changeset
566 def globaltrace_trackcallers(self, frame, why, arg):
f35b4acc594f Added --trackcalls command line arg to display crude caller/callee
Skip Montanaro <skip@pobox.com>
parents: 31551
diff changeset
567 """Handler for call events.
f35b4acc594f Added --trackcalls command line arg to display crude caller/callee
Skip Montanaro <skip@pobox.com>
parents: 31551
diff changeset
568
f35b4acc594f Added --trackcalls command line arg to display crude caller/callee
Skip Montanaro <skip@pobox.com>
parents: 31551
diff changeset
569 Adds information about who called who to the self._callers dict.
f35b4acc594f Added --trackcalls command line arg to display crude caller/callee
Skip Montanaro <skip@pobox.com>
parents: 31551
diff changeset
570 """
f35b4acc594f Added --trackcalls command line arg to display crude caller/callee
Skip Montanaro <skip@pobox.com>
parents: 31551
diff changeset
571 if why == 'call':
f35b4acc594f Added --trackcalls command line arg to display crude caller/callee
Skip Montanaro <skip@pobox.com>
parents: 31551
diff changeset
572 # XXX Should do a better job of identifying methods
31892
081972ae2f84 Refactor common code out of globaltrace_trackcallers() and
Skip Montanaro <skip@pobox.com>
parents: 31881
diff changeset
573 this_func = self.file_module_function_of(frame)
081972ae2f84 Refactor common code out of globaltrace_trackcallers() and
Skip Montanaro <skip@pobox.com>
parents: 31881
diff changeset
574 parent_func = self.file_module_function_of(frame.f_back)
31881
f35b4acc594f Added --trackcalls command line arg to display crude caller/callee
Skip Montanaro <skip@pobox.com>
parents: 31551
diff changeset
575 self._callers[(parent_func, this_func)] = 1
f35b4acc594f Added --trackcalls command line arg to display crude caller/callee
Skip Montanaro <skip@pobox.com>
parents: 31551
diff changeset
576
27933
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
577 def globaltrace_countfuncs(self, frame, why, arg):
28712
d3127b671d36 Holistic refactoring.
Jeremy Hylton <jeremy@alum.mit.edu>
parents: 28089
diff changeset
578 """Handler for call events.
28774
c9bee11eaa8b Whitespace normalization.
Tim Peters <tim.peters@gmail.com>
parents: 28727
diff changeset
579
28712
d3127b671d36 Holistic refactoring.
Jeremy Hylton <jeremy@alum.mit.edu>
parents: 28089
diff changeset
580 Adds (filename, modulename, funcname) to the self._calledfuncs dict.
27933
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
581 """
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
582 if why == 'call':
31892
081972ae2f84 Refactor common code out of globaltrace_trackcallers() and
Skip Montanaro <skip@pobox.com>
parents: 31881
diff changeset
583 this_func = self.file_module_function_of(frame)
081972ae2f84 Refactor common code out of globaltrace_trackcallers() and
Skip Montanaro <skip@pobox.com>
parents: 31881
diff changeset
584 self._calledfuncs[this_func] = 1
27933
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
585
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
586 def globaltrace_lt(self, frame, why, arg):
28712
d3127b671d36 Holistic refactoring.
Jeremy Hylton <jeremy@alum.mit.edu>
parents: 28089
diff changeset
587 """Handler for call events.
d3127b671d36 Holistic refactoring.
Jeremy Hylton <jeremy@alum.mit.edu>
parents: 28089
diff changeset
588
d3127b671d36 Holistic refactoring.
Jeremy Hylton <jeremy@alum.mit.edu>
parents: 28089
diff changeset
589 If the code block being entered is to be ignored, returns `None',
d3127b671d36 Holistic refactoring.
Jeremy Hylton <jeremy@alum.mit.edu>
parents: 28089
diff changeset
590 else returns self.localtrace.
27933
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 if why == 'call':
28712
d3127b671d36 Holistic refactoring.
Jeremy Hylton <jeremy@alum.mit.edu>
parents: 28089
diff changeset
593 code = frame.f_code
40893
32c4897b5d46 Merged revisions 53623-53858 via svnmerge from
Thomas Wouters <thomas@python.org>
parents: 40809
diff changeset
594 filename = frame.f_globals.get('__file__', None)
27933
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
595 if filename:
66308
4f29aa8f9f54 Issue #10371: Deprecated undocumented functions in the trace module.
Alexander Belopolsky <alexander.belopolsky@gmail.com>
parents: 66097
diff changeset
596 # XXX _modname() doesn't work right for packages, so
28713
d08d15a70295 Add helper function to get module name taking packages into account.
Jeremy Hylton <jeremy@alum.mit.edu>
parents: 28712
diff changeset
597 # the ignore support won't work right for packages
66308
4f29aa8f9f54 Issue #10371: Deprecated undocumented functions in the trace module.
Alexander Belopolsky <alexander.belopolsky@gmail.com>
parents: 66097
diff changeset
598 modulename = _modname(filename)
27933
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
599 if modulename is not None:
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
600 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
601 if not ignore_it:
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
602 if self.trace:
40775
4cdf21a0dbbc Fix most trivially-findable print statements.
Guido van Rossum <guido@python.org>
parents: 40587
diff changeset
603 print((" --- modulename: %s, funcname: %s"
4cdf21a0dbbc Fix most trivially-findable print statements.
Guido van Rossum <guido@python.org>
parents: 40587
diff changeset
604 % (modulename, code.co_name)))
27933
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
605 return self.localtrace
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
606 else:
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
607 return None
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
608
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
609 def localtrace_trace_and_count(self, frame, why, arg):
28712
d3127b671d36 Holistic refactoring.
Jeremy Hylton <jeremy@alum.mit.edu>
parents: 28089
diff changeset
610 if why == "line":
27933
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
611 # record the file name and line number of every trace
28712
d3127b671d36 Holistic refactoring.
Jeremy Hylton <jeremy@alum.mit.edu>
parents: 28089
diff changeset
612 filename = frame.f_code.co_filename
d3127b671d36 Holistic refactoring.
Jeremy Hylton <jeremy@alum.mit.edu>
parents: 28089
diff changeset
613 lineno = frame.f_lineno
27933
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
614 key = filename, lineno
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
615 self.counts[key] = self.counts.get(key, 0) + 1
27963
226463445697 Whitespace normalization.
Tim Peters <tim.peters@gmail.com>
parents: 27933
diff changeset
616
45548
8c3c5b4eef55 Merged revisions 61038,61042-61045,61047,61050,61053,61055-61056,61061-61062,61066,61068,61070,61081-61095 via svnmerge from
Christian Heimes <christian@cheimes.de>
parents: 44782
diff changeset
617 if self.start_time:
8c3c5b4eef55 Merged revisions 61038,61042-61045,61047,61050,61053,61055-61056,61061-61062,61066,61068,61070,61081-61095 via svnmerge from
Christian Heimes <christian@cheimes.de>
parents: 44782
diff changeset
618 print('%.2f' % (time.time() - self.start_time), end=' ')
28712
d3127b671d36 Holistic refactoring.
Jeremy Hylton <jeremy@alum.mit.edu>
parents: 28089
diff changeset
619 bname = os.path.basename(filename)
40775
4cdf21a0dbbc Fix most trivially-findable print statements.
Guido van Rossum <guido@python.org>
parents: 40587
diff changeset
620 print("%s(%d): %s" % (bname, lineno,
65115
b9a75dd7420d Merged revisions 82805-82806,83523-83527,83536,83538,83542,83546-83548,83550-83555,83558,83560 via svnmerge from
Georg Brandl <georg@python.org>
parents: 64925
diff changeset
621 linecache.getline(filename, lineno)), end='')
27933
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
622 return self.localtrace
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
623
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
624 def localtrace_trace(self, frame, why, arg):
28712
d3127b671d36 Holistic refactoring.
Jeremy Hylton <jeremy@alum.mit.edu>
parents: 28089
diff changeset
625 if why == "line":
d3127b671d36 Holistic refactoring.
Jeremy Hylton <jeremy@alum.mit.edu>
parents: 28089
diff changeset
626 # record the file name and line number of every trace
d3127b671d36 Holistic refactoring.
Jeremy Hylton <jeremy@alum.mit.edu>
parents: 28089
diff changeset
627 filename = frame.f_code.co_filename
d3127b671d36 Holistic refactoring.
Jeremy Hylton <jeremy@alum.mit.edu>
parents: 28089
diff changeset
628 lineno = frame.f_lineno
27963
226463445697 Whitespace normalization.
Tim Peters <tim.peters@gmail.com>
parents: 27933
diff changeset
629
45548
8c3c5b4eef55 Merged revisions 61038,61042-61045,61047,61050,61053,61055-61056,61061-61062,61066,61068,61070,61081-61095 via svnmerge from
Christian Heimes <christian@cheimes.de>
parents: 44782
diff changeset
630 if self.start_time:
8c3c5b4eef55 Merged revisions 61038,61042-61045,61047,61050,61053,61055-61056,61061-61062,61066,61068,61070,61081-61095 via svnmerge from
Christian Heimes <christian@cheimes.de>
parents: 44782
diff changeset
631 print('%.2f' % (time.time() - self.start_time), end=' ')
28712
d3127b671d36 Holistic refactoring.
Jeremy Hylton <jeremy@alum.mit.edu>
parents: 28089
diff changeset
632 bname = os.path.basename(filename)
40775
4cdf21a0dbbc Fix most trivially-findable print statements.
Guido van Rossum <guido@python.org>
parents: 40587
diff changeset
633 print("%s(%d): %s" % (bname, lineno,
65115
b9a75dd7420d Merged revisions 82805-82806,83523-83527,83536,83538,83542,83546-83548,83550-83555,83558,83560 via svnmerge from
Georg Brandl <georg@python.org>
parents: 64925
diff changeset
634 linecache.getline(filename, lineno)), end='')
27933
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
635 return self.localtrace
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
636
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
637 def localtrace_count(self, frame, why, arg):
28712
d3127b671d36 Holistic refactoring.
Jeremy Hylton <jeremy@alum.mit.edu>
parents: 28089
diff changeset
638 if why == "line":
27933
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
639 filename = frame.f_code.co_filename
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
640 lineno = frame.f_lineno
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
641 key = filename, lineno
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
642 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
643 return self.localtrace
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
644
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
645 def results(self):
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
646 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
647 outfile=self.outfile,
31881
f35b4acc594f Added --trackcalls command line arg to display crude caller/callee
Skip Montanaro <skip@pobox.com>
parents: 31551
diff changeset
648 calledfuncs=self._calledfuncs,
f35b4acc594f Added --trackcalls command line arg to display crude caller/callee
Skip Montanaro <skip@pobox.com>
parents: 31551
diff changeset
649 callers=self._callers)
27933
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
650
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
651 def _err_exit(msg):
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
652 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
653 sys.exit(1)
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
654
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
655 def main(argv=None):
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
656 import getopt
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
657
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
658 if argv is None:
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
659 argv = sys.argv
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
660 try:
45548
8c3c5b4eef55 Merged revisions 61038,61042-61045,61047,61050,61053,61055-61056,61061-61062,61066,61068,61070,61081-61095 via svnmerge from
Christian Heimes <christian@cheimes.de>
parents: 44782
diff changeset
661 opts, prog_argv = getopt.getopt(argv[1:], "tcrRf:d:msC:lTg",
27933
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
662 ["help", "version", "trace", "count",
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
663 "report", "no-report", "summary",
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
664 "file=", "missing",
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
665 "ignore-module=", "ignore-dir=",
31881
f35b4acc594f Added --trackcalls command line arg to display crude caller/callee
Skip Montanaro <skip@pobox.com>
parents: 31551
diff changeset
666 "coverdir=", "listfuncs",
45548
8c3c5b4eef55 Merged revisions 61038,61042-61045,61047,61050,61053,61055-61056,61061-61062,61066,61068,61070,61081-61095 via svnmerge from
Christian Heimes <christian@cheimes.de>
parents: 44782
diff changeset
667 "trackcalls", "timing"])
27933
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
668
40587
6b18a095c2e7 SF patch 1631942 by Collin Winter:
Guido van Rossum <guido@python.org>
parents: 39782
diff changeset
669 except getopt.error as msg:
27933
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
670 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
671 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
672 % sys.argv[0])
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
673 sys.exit(1)
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
674
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
675 trace = 0
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
676 count = 0
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
677 report = 0
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
678 no_report = 0
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
679 counts_file = None
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
680 missing = 0
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
681 ignore_modules = []
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
682 ignore_dirs = []
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
683 coverdir = None
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
684 summary = 0
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
685 listfuncs = False
31881
f35b4acc594f Added --trackcalls command line arg to display crude caller/callee
Skip Montanaro <skip@pobox.com>
parents: 31551
diff changeset
686 countcallers = False
45548
8c3c5b4eef55 Merged revisions 61038,61042-61045,61047,61050,61053,61055-61056,61061-61062,61066,61068,61070,61081-61095 via svnmerge from
Christian Heimes <christian@cheimes.de>
parents: 44782
diff changeset
687 timing = False
27933
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 for opt, val in opts:
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
690 if opt == "--help":
69604
c9792d3166eb Prevent deprecation warning
Éric Araujo <merwok@netwok.org>
parents: 68257
diff changeset
691 _usage(sys.stdout)
27933
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
692 sys.exit(0)
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
693
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
694 if opt == "--version":
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
695 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
696 sys.exit(0)
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
697
31881
f35b4acc594f Added --trackcalls command line arg to display crude caller/callee
Skip Montanaro <skip@pobox.com>
parents: 31551
diff changeset
698 if opt == "-T" or opt == "--trackcalls":
f35b4acc594f Added --trackcalls command line arg to display crude caller/callee
Skip Montanaro <skip@pobox.com>
parents: 31551
diff changeset
699 countcallers = True
f35b4acc594f Added --trackcalls command line arg to display crude caller/callee
Skip Montanaro <skip@pobox.com>
parents: 31551
diff changeset
700 continue
f35b4acc594f Added --trackcalls command line arg to display crude caller/callee
Skip Montanaro <skip@pobox.com>
parents: 31551
diff changeset
701
27933
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
702 if opt == "-l" or opt == "--listfuncs":
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
703 listfuncs = True
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
704 continue
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
705
45548
8c3c5b4eef55 Merged revisions 61038,61042-61045,61047,61050,61053,61055-61056,61061-61062,61066,61068,61070,61081-61095 via svnmerge from
Christian Heimes <christian@cheimes.de>
parents: 44782
diff changeset
706 if opt == "-g" or opt == "--timing":
8c3c5b4eef55 Merged revisions 61038,61042-61045,61047,61050,61053,61055-61056,61061-61062,61066,61068,61070,61081-61095 via svnmerge from
Christian Heimes <christian@cheimes.de>
parents: 44782
diff changeset
707 timing = True
8c3c5b4eef55 Merged revisions 61038,61042-61045,61047,61050,61053,61055-61056,61061-61062,61066,61068,61070,61081-61095 via svnmerge from
Christian Heimes <christian@cheimes.de>
parents: 44782
diff changeset
708 continue
8c3c5b4eef55 Merged revisions 61038,61042-61045,61047,61050,61053,61055-61056,61061-61062,61066,61068,61070,61081-61095 via svnmerge from
Christian Heimes <christian@cheimes.de>
parents: 44782
diff changeset
709
27933
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
710 if opt == "-t" or opt == "--trace":
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
711 trace = 1
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
712 continue
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
713
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
714 if opt == "-c" or opt == "--count":
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
715 count = 1
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
716 continue
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
717
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
718 if opt == "-r" or opt == "--report":
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
719 report = 1
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
720 continue
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
721
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
722 if opt == "-R" or opt == "--no-report":
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
723 no_report = 1
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
724 continue
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
725
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
726 if opt == "-f" or opt == "--file":
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
727 counts_file = val
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
728 continue
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
729
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
730 if opt == "-m" or opt == "--missing":
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
731 missing = 1
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
732 continue
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
733
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
734 if opt == "-C" or opt == "--coverdir":
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
735 coverdir = val
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
736 continue
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
737
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
738 if opt == "-s" or opt == "--summary":
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
739 summary = 1
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
740 continue
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
741
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
742 if opt == "--ignore-module":
44782
ca2683a67943 Merged revisions 60080-60089,60091-60093 via svnmerge from
Georg Brandl <georg@python.org>
parents: 44025
diff changeset
743 for mod in val.split(","):
ca2683a67943 Merged revisions 60080-60089,60091-60093 via svnmerge from
Georg Brandl <georg@python.org>
parents: 44025
diff changeset
744 ignore_modules.append(mod.strip())
27933
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
745 continue
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
746
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
747 if opt == "--ignore-dir":
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
748 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
749 s = os.path.expandvars(s)
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
750 # 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
751
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
752 s = s.replace("$prefix",
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
753 os.path.join(sys.prefix, "lib",
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
754 "python" + sys.version[:3]))
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
755 s = s.replace("$exec_prefix",
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
756 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
757 "python" + sys.version[:3]))
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
758 s = os.path.normpath(s)
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
759 ignore_dirs.append(s)
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
760 continue
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
761
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
762 assert 0, "Should never get here"
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
763
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
764 if listfuncs and (count or trace):
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
765 _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
766
31881
f35b4acc594f Added --trackcalls command line arg to display crude caller/callee
Skip Montanaro <skip@pobox.com>
parents: 31551
diff changeset
767 if not (count or trace or report or listfuncs or countcallers):
f35b4acc594f Added --trackcalls command line arg to display crude caller/callee
Skip Montanaro <skip@pobox.com>
parents: 31551
diff changeset
768 _err_exit("must specify one of --trace, --count, --report, "
f35b4acc594f Added --trackcalls command line arg to display crude caller/callee
Skip Montanaro <skip@pobox.com>
parents: 31551
diff changeset
769 "--listfuncs, or --trackcalls")
27933
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
770
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
771 if report and no_report:
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
772 _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
773
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
774 if report and not counts_file:
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
775 _err_exit("--report requires a --file")
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
776
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
777 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
778 _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
779
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
780 # everything is ready
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
781 if report:
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
782 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
783 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
784 else:
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
785 sys.argv = prog_argv
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
786 progname = prog_argv[0]
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
787 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
788
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
789 t = Trace(count, trace, countfuncs=listfuncs,
31881
f35b4acc594f Added --trackcalls command line arg to display crude caller/callee
Skip Montanaro <skip@pobox.com>
parents: 31551
diff changeset
790 countcallers=countcallers, ignoremods=ignore_modules,
f35b4acc594f Added --trackcalls command line arg to display crude caller/callee
Skip Montanaro <skip@pobox.com>
parents: 31551
diff changeset
791 ignoredirs=ignore_dirs, infile=counts_file,
45548
8c3c5b4eef55 Merged revisions 61038,61042-61045,61047,61050,61053,61055-61056,61061-61062,61066,61068,61070,61081-61095 via svnmerge from
Christian Heimes <christian@cheimes.de>
parents: 44782
diff changeset
792 outfile=counts_file, timing=timing)
27933
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
793 try:
63133
9dbbe6e7b668 Merged revisions 83035 via svnmerge from
Alexander Belopolsky <alexander.belopolsky@gmail.com>
parents: 63110
diff changeset
794 with open(progname) as fp:
9dbbe6e7b668 Merged revisions 83035 via svnmerge from
Alexander Belopolsky <alexander.belopolsky@gmail.com>
parents: 63110
diff changeset
795 code = compile(fp.read(), progname, 'exec')
63438
9ee196a7f9a2 Merged revisions 83393,83396,83398,83404-83405,83408 via svnmerge from
Georg Brandl <georg@python.org>
parents: 63133
diff changeset
796 # try to emulate __main__ namespace as much as possible
9ee196a7f9a2 Merged revisions 83393,83396,83398,83404-83405,83408 via svnmerge from
Georg Brandl <georg@python.org>
parents: 63133
diff changeset
797 globs = {
9ee196a7f9a2 Merged revisions 83393,83396,83398,83404-83405,83408 via svnmerge from
Georg Brandl <georg@python.org>
parents: 63133
diff changeset
798 '__file__': progname,
9ee196a7f9a2 Merged revisions 83393,83396,83398,83404-83405,83408 via svnmerge from
Georg Brandl <georg@python.org>
parents: 63133
diff changeset
799 '__name__': '__main__',
9ee196a7f9a2 Merged revisions 83393,83396,83398,83404-83405,83408 via svnmerge from
Georg Brandl <georg@python.org>
parents: 63133
diff changeset
800 '__package__': None,
9ee196a7f9a2 Merged revisions 83393,83396,83398,83404-83405,83408 via svnmerge from
Georg Brandl <georg@python.org>
parents: 63133
diff changeset
801 '__cached__': None,
9ee196a7f9a2 Merged revisions 83393,83396,83398,83404-83405,83408 via svnmerge from
Georg Brandl <georg@python.org>
parents: 63133
diff changeset
802 }
9ee196a7f9a2 Merged revisions 83393,83396,83398,83404-83405,83408 via svnmerge from
Georg Brandl <georg@python.org>
parents: 63133
diff changeset
803 t.runctx(code, globs, globs)
40587
6b18a095c2e7 SF patch 1631942 by Collin Winter:
Guido van Rossum <guido@python.org>
parents: 39782
diff changeset
804 except IOError as err:
28712
d3127b671d36 Holistic refactoring.
Jeremy Hylton <jeremy@alum.mit.edu>
parents: 28089
diff changeset
805 _err_exit("Cannot run file %r because: %s" % (sys.argv[0], err))
27933
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
806 except SystemExit:
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
807 pass
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
808
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
809 results = t.results()
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
810
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
811 if not no_report:
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
812 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
813
66308
4f29aa8f9f54 Issue #10371: Deprecated undocumented functions in the trace module.
Alexander Belopolsky <alexander.belopolsky@gmail.com>
parents: 66097
diff changeset
814 # Deprecated API
4f29aa8f9f54 Issue #10371: Deprecated undocumented functions in the trace module.
Alexander Belopolsky <alexander.belopolsky@gmail.com>
parents: 66097
diff changeset
815 def usage(outfile):
4f29aa8f9f54 Issue #10371: Deprecated undocumented functions in the trace module.
Alexander Belopolsky <alexander.belopolsky@gmail.com>
parents: 66097
diff changeset
816 _warn("The trace.usage() function is deprecated",
4f29aa8f9f54 Issue #10371: Deprecated undocumented functions in the trace module.
Alexander Belopolsky <alexander.belopolsky@gmail.com>
parents: 66097
diff changeset
817 DeprecationWarning, 2)
4f29aa8f9f54 Issue #10371: Deprecated undocumented functions in the trace module.
Alexander Belopolsky <alexander.belopolsky@gmail.com>
parents: 66097
diff changeset
818 _usage(outfile)
4f29aa8f9f54 Issue #10371: Deprecated undocumented functions in the trace module.
Alexander Belopolsky <alexander.belopolsky@gmail.com>
parents: 66097
diff changeset
819
4f29aa8f9f54 Issue #10371: Deprecated undocumented functions in the trace module.
Alexander Belopolsky <alexander.belopolsky@gmail.com>
parents: 66097
diff changeset
820 class Ignore(_Ignore):
4f29aa8f9f54 Issue #10371: Deprecated undocumented functions in the trace module.
Alexander Belopolsky <alexander.belopolsky@gmail.com>
parents: 66097
diff changeset
821 def __init__(self, modules=None, dirs=None):
4f29aa8f9f54 Issue #10371: Deprecated undocumented functions in the trace module.
Alexander Belopolsky <alexander.belopolsky@gmail.com>
parents: 66097
diff changeset
822 _warn("The class trace.Ignore is deprecated",
4f29aa8f9f54 Issue #10371: Deprecated undocumented functions in the trace module.
Alexander Belopolsky <alexander.belopolsky@gmail.com>
parents: 66097
diff changeset
823 DeprecationWarning, 2)
4f29aa8f9f54 Issue #10371: Deprecated undocumented functions in the trace module.
Alexander Belopolsky <alexander.belopolsky@gmail.com>
parents: 66097
diff changeset
824 _Ignore.__init__(self, modules, dirs)
4f29aa8f9f54 Issue #10371: Deprecated undocumented functions in the trace module.
Alexander Belopolsky <alexander.belopolsky@gmail.com>
parents: 66097
diff changeset
825
4f29aa8f9f54 Issue #10371: Deprecated undocumented functions in the trace module.
Alexander Belopolsky <alexander.belopolsky@gmail.com>
parents: 66097
diff changeset
826 def modname(path):
4f29aa8f9f54 Issue #10371: Deprecated undocumented functions in the trace module.
Alexander Belopolsky <alexander.belopolsky@gmail.com>
parents: 66097
diff changeset
827 _warn("The trace.modname() function is deprecated",
4f29aa8f9f54 Issue #10371: Deprecated undocumented functions in the trace module.
Alexander Belopolsky <alexander.belopolsky@gmail.com>
parents: 66097
diff changeset
828 DeprecationWarning, 2)
4f29aa8f9f54 Issue #10371: Deprecated undocumented functions in the trace module.
Alexander Belopolsky <alexander.belopolsky@gmail.com>
parents: 66097
diff changeset
829 return _modname(path)
4f29aa8f9f54 Issue #10371: Deprecated undocumented functions in the trace module.
Alexander Belopolsky <alexander.belopolsky@gmail.com>
parents: 66097
diff changeset
830
4f29aa8f9f54 Issue #10371: Deprecated undocumented functions in the trace module.
Alexander Belopolsky <alexander.belopolsky@gmail.com>
parents: 66097
diff changeset
831 def fullmodname(path):
4f29aa8f9f54 Issue #10371: Deprecated undocumented functions in the trace module.
Alexander Belopolsky <alexander.belopolsky@gmail.com>
parents: 66097
diff changeset
832 _warn("The trace.fullmodname() function is deprecated",
4f29aa8f9f54 Issue #10371: Deprecated undocumented functions in the trace module.
Alexander Belopolsky <alexander.belopolsky@gmail.com>
parents: 66097
diff changeset
833 DeprecationWarning, 2)
4f29aa8f9f54 Issue #10371: Deprecated undocumented functions in the trace module.
Alexander Belopolsky <alexander.belopolsky@gmail.com>
parents: 66097
diff changeset
834 return _fullmodname(path)
4f29aa8f9f54 Issue #10371: Deprecated undocumented functions in the trace module.
Alexander Belopolsky <alexander.belopolsky@gmail.com>
parents: 66097
diff changeset
835
4f29aa8f9f54 Issue #10371: Deprecated undocumented functions in the trace module.
Alexander Belopolsky <alexander.belopolsky@gmail.com>
parents: 66097
diff changeset
836 def find_lines_from_code(code, strs):
4f29aa8f9f54 Issue #10371: Deprecated undocumented functions in the trace module.
Alexander Belopolsky <alexander.belopolsky@gmail.com>
parents: 66097
diff changeset
837 _warn("The trace.find_lines_from_code() function is deprecated",
4f29aa8f9f54 Issue #10371: Deprecated undocumented functions in the trace module.
Alexander Belopolsky <alexander.belopolsky@gmail.com>
parents: 66097
diff changeset
838 DeprecationWarning, 2)
4f29aa8f9f54 Issue #10371: Deprecated undocumented functions in the trace module.
Alexander Belopolsky <alexander.belopolsky@gmail.com>
parents: 66097
diff changeset
839 return _find_lines_from_code(code, strs)
4f29aa8f9f54 Issue #10371: Deprecated undocumented functions in the trace module.
Alexander Belopolsky <alexander.belopolsky@gmail.com>
parents: 66097
diff changeset
840
4f29aa8f9f54 Issue #10371: Deprecated undocumented functions in the trace module.
Alexander Belopolsky <alexander.belopolsky@gmail.com>
parents: 66097
diff changeset
841 def find_lines(code, strs):
4f29aa8f9f54 Issue #10371: Deprecated undocumented functions in the trace module.
Alexander Belopolsky <alexander.belopolsky@gmail.com>
parents: 66097
diff changeset
842 _warn("The trace.find_lines() function is deprecated",
4f29aa8f9f54 Issue #10371: Deprecated undocumented functions in the trace module.
Alexander Belopolsky <alexander.belopolsky@gmail.com>
parents: 66097
diff changeset
843 DeprecationWarning, 2)
4f29aa8f9f54 Issue #10371: Deprecated undocumented functions in the trace module.
Alexander Belopolsky <alexander.belopolsky@gmail.com>
parents: 66097
diff changeset
844 return _find_lines(code, strs)
4f29aa8f9f54 Issue #10371: Deprecated undocumented functions in the trace module.
Alexander Belopolsky <alexander.belopolsky@gmail.com>
parents: 66097
diff changeset
845
4f29aa8f9f54 Issue #10371: Deprecated undocumented functions in the trace module.
Alexander Belopolsky <alexander.belopolsky@gmail.com>
parents: 66097
diff changeset
846 def find_strings(filename, encoding=None):
4f29aa8f9f54 Issue #10371: Deprecated undocumented functions in the trace module.
Alexander Belopolsky <alexander.belopolsky@gmail.com>
parents: 66097
diff changeset
847 _warn("The trace.find_strings() function is deprecated",
4f29aa8f9f54 Issue #10371: Deprecated undocumented functions in the trace module.
Alexander Belopolsky <alexander.belopolsky@gmail.com>
parents: 66097
diff changeset
848 DeprecationWarning, 2)
4f29aa8f9f54 Issue #10371: Deprecated undocumented functions in the trace module.
Alexander Belopolsky <alexander.belopolsky@gmail.com>
parents: 66097
diff changeset
849 return _find_strings(filename, encoding=None)
4f29aa8f9f54 Issue #10371: Deprecated undocumented functions in the trace module.
Alexander Belopolsky <alexander.belopolsky@gmail.com>
parents: 66097
diff changeset
850
4f29aa8f9f54 Issue #10371: Deprecated undocumented functions in the trace module.
Alexander Belopolsky <alexander.belopolsky@gmail.com>
parents: 66097
diff changeset
851 def find_executable_linenos(filename):
4f29aa8f9f54 Issue #10371: Deprecated undocumented functions in the trace module.
Alexander Belopolsky <alexander.belopolsky@gmail.com>
parents: 66097
diff changeset
852 _warn("The trace.find_executable_linenos() function is deprecated",
4f29aa8f9f54 Issue #10371: Deprecated undocumented functions in the trace module.
Alexander Belopolsky <alexander.belopolsky@gmail.com>
parents: 66097
diff changeset
853 DeprecationWarning, 2)
4f29aa8f9f54 Issue #10371: Deprecated undocumented functions in the trace module.
Alexander Belopolsky <alexander.belopolsky@gmail.com>
parents: 66097
diff changeset
854 return _find_executable_linenos(filename)
4f29aa8f9f54 Issue #10371: Deprecated undocumented functions in the trace module.
Alexander Belopolsky <alexander.belopolsky@gmail.com>
parents: 66097
diff changeset
855
27933
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
856 if __name__=='__main__':
44f3e315929b Copy the trace module here from Tools/scripts.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
857 main()