cpython-lru_cache-weakref: Tools/scripts/trace.py annotate

annotate Tools/scripts/trace.py @ 24042:63fbbed10714 legacy-trunk

Simplify HTTPSConnection constructor. See discussion in SF bug 458463.

author	Jeremy Hylton <jeremy@alum.mit.edu>
date	Fri, 28 Jun 2002 23:38:14 +0000
parents	6392f03222e0
children	ec2c854bd196

rev	line source
13838 91d95a7b7858 Python code coverage tool by Skip Montanaro and Andrew Dalke Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	1 #!/usr/bin/env python
91d95a7b7858 Python code coverage tool by Skip Montanaro and Andrew Dalke Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	2
21071 6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	3 # portions copyright 2001, Autonomous Zones Industries, Inc., all rights...
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	4 # err... reserved and offered to the public under the terms of the
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	5 # Python 2.2 license.
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	6 # Author: Zooko O'Whielacronx
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	7 # http://zooko.com/
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	8 # mailto:zooko@zooko.com
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	9 #
13838 91d95a7b7858 Python code coverage tool by Skip Montanaro and Andrew Dalke Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	10 # Copyright 2000, Mojam Media, Inc., all rights reserved.
91d95a7b7858 Python code coverage tool by Skip Montanaro and Andrew Dalke Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	11 # Author: Skip Montanaro
91d95a7b7858 Python code coverage tool by Skip Montanaro and Andrew Dalke Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	12 #
91d95a7b7858 Python code coverage tool by Skip Montanaro and Andrew Dalke Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	13 # Copyright 1999, Bioreason, Inc., all rights reserved.
91d95a7b7858 Python code coverage tool by Skip Montanaro and Andrew Dalke Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	14 # Author: Andrew Dalke
91d95a7b7858 Python code coverage tool by Skip Montanaro and Andrew Dalke Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	15 #
91d95a7b7858 Python code coverage tool by Skip Montanaro and Andrew Dalke Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	16 # Copyright 1995-1997, Automatrix, Inc., all rights reserved.
91d95a7b7858 Python code coverage tool by Skip Montanaro and Andrew Dalke Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	17 # Author: Skip Montanaro
91d95a7b7858 Python code coverage tool by Skip Montanaro and Andrew Dalke Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	18 #
91d95a7b7858 Python code coverage tool by Skip Montanaro and Andrew Dalke Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	19 # Copyright 1991-1995, Stichting Mathematisch Centrum, all rights reserved.
91d95a7b7858 Python code coverage tool by Skip Montanaro and Andrew Dalke Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	20 #
91d95a7b7858 Python code coverage tool by Skip Montanaro and Andrew Dalke Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	21 #
91d95a7b7858 Python code coverage tool by Skip Montanaro and Andrew Dalke Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	22 # Permission to use, copy, modify, and distribute this Python software and
91d95a7b7858 Python code coverage tool by Skip Montanaro and Andrew Dalke Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	23 # its associated documentation for any purpose without fee is hereby
91d95a7b7858 Python code coverage tool by Skip Montanaro and Andrew Dalke Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	24 # granted, provided that the above copyright notice appears in all copies,
91d95a7b7858 Python code coverage tool by Skip Montanaro and Andrew Dalke Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	25 # and that both that copyright notice and this permission notice appear in
91d95a7b7858 Python code coverage tool by Skip Montanaro and Andrew Dalke Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	26 # supporting documentation, and that the name of neither Automatrix,
91d95a7b7858 Python code coverage tool by Skip Montanaro and Andrew Dalke Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	27 # Bioreason or Mojam Media be used in advertising or publicity pertaining to
91d95a7b7858 Python code coverage tool by Skip Montanaro and Andrew Dalke Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	28 # distribution of the software without specific, written prior permission.
91d95a7b7858 Python code coverage tool by Skip Montanaro and Andrew Dalke Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	29 #
91d95a7b7858 Python code coverage tool by Skip Montanaro and Andrew Dalke Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	30 #
21071 6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	31 # Cleaned up the usage message --GvR 11/28/01
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	32 #
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	33 # Summary of even more recent changes, --Zooko 2001-10-14
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	34 # Used new `inspect' module for better (?) determination of file<->module
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	35 # mappings, line numbers, and source code.
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	36 # Used new local trace function for faster (and better?) operation.
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	37 # Removed "speed hack", which, as far as I can tell, meant that it would
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	38 # ignore all files ??? (When I tried it, it would ignore only most of my
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	39 # files. In any case with the speed hack removed in favor of actually
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	40 # calling `Ignore.names()', it ignores only those files that I told it to
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	41 # ignore, so I am happy.)
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	42 # Rolled the `Coverage' class into `Trace', which now does either tracing or
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	43 # counting or both according to constructor flags.
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	44 # Moved the construction of the `Ignore' object inside the constructor of
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	45 # `Trace', simplifying usage.
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	46 # Changed function `create_results_log()' into method
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	47 # `CoverageResults.write_results()'.
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	48 # Add new mode "countfuncs" which is faster and which just reports which
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	49 # functions were invoked.
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	50
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	51 # Made `write_results' create `coverdir' if it doesn't already exist.
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	52 # Moved the `run' funcs into `Trace' for simpler usage.
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	53 # Use pickle instead of marshal for persistence.
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	54 #
13838 91d95a7b7858 Python code coverage tool by Skip Montanaro and Andrew Dalke Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	55 # Summary of recent changes:
91d95a7b7858 Python code coverage tool by Skip Montanaro and Andrew Dalke Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	56 # Support for files with the same basename (submodules in packages)
91d95a7b7858 Python code coverage tool by Skip Montanaro and Andrew Dalke Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	57 # Expanded the idea of how to ignore files or modules
91d95a7b7858 Python code coverage tool by Skip Montanaro and Andrew Dalke Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	58 # Split tracing and counting into different classes
91d95a7b7858 Python code coverage tool by Skip Montanaro and Andrew Dalke Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	59 # Extracted count information and reporting from the count class
91d95a7b7858 Python code coverage tool by Skip Montanaro and Andrew Dalke Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	60 # Added some ability to detect which missing lines could be executed
91d95a7b7858 Python code coverage tool by Skip Montanaro and Andrew Dalke Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	61 # Added pseudo-pragma to prohibit complaining about unexecuted lines
91d95a7b7858 Python code coverage tool by Skip Montanaro and Andrew Dalke Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	62 # Rewrote the main program
91d95a7b7858 Python code coverage tool by Skip Montanaro and Andrew Dalke Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	63
91d95a7b7858 Python code coverage tool by Skip Montanaro and Andrew Dalke Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	64 # Summary of older changes:
91d95a7b7858 Python code coverage tool by Skip Montanaro and Andrew Dalke Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	65 # Added run-time display of statements being executed
91d95a7b7858 Python code coverage tool by Skip Montanaro and Andrew Dalke Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	66 # Incorporated portability and performance fixes from Greg Stein
91d95a7b7858 Python code coverage tool by Skip Montanaro and Andrew Dalke Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	67 # Incorporated main program from Michael Scharf
91d95a7b7858 Python code coverage tool by Skip Montanaro and Andrew Dalke Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	68
91d95a7b7858 Python code coverage tool by Skip Montanaro and Andrew Dalke Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	69 """
91d95a7b7858 Python code coverage tool by Skip Montanaro and Andrew Dalke Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	70 program/module to trace Python program or function execution
91d95a7b7858 Python code coverage tool by Skip Montanaro and Andrew Dalke Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	71
91d95a7b7858 Python code coverage tool by Skip Montanaro and Andrew Dalke Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	72 Sample use, command line:
91d95a7b7858 Python code coverage tool by Skip Montanaro and Andrew Dalke Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	73 trace.py -c -f counts --ignore-dir '$prefix' spam.py eggs
91d95a7b7858 Python code coverage tool by Skip Montanaro and Andrew Dalke Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	74 trace.py -t --ignore-dir '$prefix' spam.py eggs
91d95a7b7858 Python code coverage tool by Skip Montanaro and Andrew Dalke Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	75
21071 6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	76 Sample use, programmatically
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	77 # create a Trace object, telling it what to ignore, and whether to do tracing
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	78 # or line-counting or both.
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	79 trace = trace.Trace(ignoredirs=[sys.prefix, sys.exec_prefix,], trace=0, count=1)
13838 91d95a7b7858 Python code coverage tool by Skip Montanaro and Andrew Dalke Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	80 # run the new command using the given trace
21071 6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	81 trace.run(coverage.globaltrace, 'main()')
13838 91d95a7b7858 Python code coverage tool by Skip Montanaro and Andrew Dalke Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	82 # make a report, telling it where you want output
21071 6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	83 trace.print_results(show_missing=1)
13838 91d95a7b7858 Python code coverage tool by Skip Montanaro and Andrew Dalke Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	84 """
91d95a7b7858 Python code coverage tool by Skip Montanaro and Andrew Dalke Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	85
21071 6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	86 import sys, os, string, tempfile, types, copy, operator, inspect, exceptions, marshal
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	87 try:
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	88 import cPickle
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	89 pickle = cPickle
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	90 except ImportError:
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	91 import pickle
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	92
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	93 true = 1
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	94 false = None
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	95
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	96 # DEBUG_MODE=1 # make this true to get printouts which help you understand what's going on
13838 91d95a7b7858 Python code coverage tool by Skip Montanaro and Andrew Dalke Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	97
91d95a7b7858 Python code coverage tool by Skip Montanaro and Andrew Dalke Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	98 def usage(outfile):
91d95a7b7858 Python code coverage tool by Skip Montanaro and Andrew Dalke Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	99 outfile.write("""Usage: %s [OPTIONS] <file> [ARGS]
91d95a7b7858 Python code coverage tool by Skip Montanaro and Andrew Dalke Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	100
21071 6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	101 Meta-options:
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	102 --help Display this help then exit.
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	103 --version Output version information then exit.
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	104
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	105 Otherwise, exactly one of the following three options must be given:
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	106 -t, --trace Print each line to sys.stdout before it is executed.
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	107 -c, --count Count the number of times each line is executed
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	108 and write the counts to <module>.cover for each
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	109 module executed, in the module's directory.
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	110 See also `--coverdir', `--file', `--no-report' below.
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	111 -r, --report Generate a report from a counts file; do not execute
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	112 any code. `--file' must specify the results file to
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	113 read, which must have been created in a previous run
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	114 with `--count --file=FILE'.
13838 91d95a7b7858 Python code coverage tool by Skip Montanaro and Andrew Dalke Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	115
21071 6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	116 Modifiers:
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	117 -f, --file=<file> File to accumulate counts over several runs.
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	118 -R, --no-report Do not generate the coverage report files.
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	119 Useful if you want to accumulate over several runs.
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	120 -C, --coverdir=<dir> Directory where the report files. The coverage
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	121 report for <package>.<module> is written to file
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	122 <dir>/<package>/<module>.cover.
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	123 -m, --missing Annotate executable lines that were not executed
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	124 with '>>>>>> '.
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	125 -s, --summary Write a brief summary on stdout for each file.
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	126 (Can only be used with --count or --report.)
13838 91d95a7b7858 Python code coverage tool by Skip Montanaro and Andrew Dalke Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	127
21071 6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	128 Filters, may be repeated multiple times:
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	129 --ignore-module=<mod> Ignore the given module and its submodules
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	130 (if it is a package).
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	131 --ignore-dir=<dir> Ignore files in the given directory (multiple
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	132 directories can be joined by os.pathsep).
13838 91d95a7b7858 Python code coverage tool by Skip Montanaro and Andrew Dalke Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	133 """ % sys.argv[0])
91d95a7b7858 Python code coverage tool by Skip Montanaro and Andrew Dalke Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	134
91d95a7b7858 Python code coverage tool by Skip Montanaro and Andrew Dalke Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	135 class Ignore:
91d95a7b7858 Python code coverage tool by Skip Montanaro and Andrew Dalke Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	136 def __init__(self, modules = None, dirs = None):
14657 946709e82be0 satisfy the tabnanny Jeremy Hylton <jeremy@alum.mit.edu> parents: 13838 diff changeset	137 self._mods = modules or []
946709e82be0 satisfy the tabnanny Jeremy Hylton <jeremy@alum.mit.edu> parents: 13838 diff changeset	138 self._dirs = dirs or []
13838 91d95a7b7858 Python code coverage tool by Skip Montanaro and Andrew Dalke Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	139
21071 6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	140 self._dirs = map(os.path.normpath, self._dirs)
14657 946709e82be0 satisfy the tabnanny Jeremy Hylton <jeremy@alum.mit.edu> parents: 13838 diff changeset	141 self._ignore = { '<string>': 1 }
13838 91d95a7b7858 Python code coverage tool by Skip Montanaro and Andrew Dalke Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	142
91d95a7b7858 Python code coverage tool by Skip Montanaro and Andrew Dalke Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	143 def names(self, filename, modulename):
14657 946709e82be0 satisfy the tabnanny Jeremy Hylton <jeremy@alum.mit.edu> parents: 13838 diff changeset	144 if self._ignore.has_key(modulename):
946709e82be0 satisfy the tabnanny Jeremy Hylton <jeremy@alum.mit.edu> parents: 13838 diff changeset	145 return self._ignore[modulename]
13838 91d95a7b7858 Python code coverage tool by Skip Montanaro and Andrew Dalke Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	146
14657 946709e82be0 satisfy the tabnanny Jeremy Hylton <jeremy@alum.mit.edu> parents: 13838 diff changeset	147 # haven't seen this one before, so see if the module name is
946709e82be0 satisfy the tabnanny Jeremy Hylton <jeremy@alum.mit.edu> parents: 13838 diff changeset	148 # on the ignore list. Need to take some care since ignoring
946709e82be0 satisfy the tabnanny Jeremy Hylton <jeremy@alum.mit.edu> parents: 13838 diff changeset	149 # "cmp" musn't mean ignoring "cmpcache" but ignoring
946709e82be0 satisfy the tabnanny Jeremy Hylton <jeremy@alum.mit.edu> parents: 13838 diff changeset	150 # "Spam" must also mean ignoring "Spam.Eggs".
946709e82be0 satisfy the tabnanny Jeremy Hylton <jeremy@alum.mit.edu> parents: 13838 diff changeset	151 for mod in self._mods:
946709e82be0 satisfy the tabnanny Jeremy Hylton <jeremy@alum.mit.edu> parents: 13838 diff changeset	152 if mod == modulename: # Identical names, so ignore
946709e82be0 satisfy the tabnanny Jeremy Hylton <jeremy@alum.mit.edu> parents: 13838 diff changeset	153 self._ignore[modulename] = 1
946709e82be0 satisfy the tabnanny Jeremy Hylton <jeremy@alum.mit.edu> parents: 13838 diff changeset	154 return 1
946709e82be0 satisfy the tabnanny Jeremy Hylton <jeremy@alum.mit.edu> parents: 13838 diff changeset	155 # check if the module is a proper submodule of something on
946709e82be0 satisfy the tabnanny Jeremy Hylton <jeremy@alum.mit.edu> parents: 13838 diff changeset	156 # the ignore list
946709e82be0 satisfy the tabnanny Jeremy Hylton <jeremy@alum.mit.edu> parents: 13838 diff changeset	157 n = len(mod)
946709e82be0 satisfy the tabnanny Jeremy Hylton <jeremy@alum.mit.edu> parents: 13838 diff changeset	158 # (will not overflow since if the first n characters are the
946709e82be0 satisfy the tabnanny Jeremy Hylton <jeremy@alum.mit.edu> parents: 13838 diff changeset	159 # same and the name has not already occured, then the size
946709e82be0 satisfy the tabnanny Jeremy Hylton <jeremy@alum.mit.edu> parents: 13838 diff changeset	160 # of "name" is greater than that of "mod")
946709e82be0 satisfy the tabnanny Jeremy Hylton <jeremy@alum.mit.edu> parents: 13838 diff changeset	161 if mod == modulename[:n] and modulename[n] == '.':
946709e82be0 satisfy the tabnanny Jeremy Hylton <jeremy@alum.mit.edu> parents: 13838 diff changeset	162 self._ignore[modulename] = 1
946709e82be0 satisfy the tabnanny Jeremy Hylton <jeremy@alum.mit.edu> parents: 13838 diff changeset	163 return 1
13838 91d95a7b7858 Python code coverage tool by Skip Montanaro and Andrew Dalke Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	164
14657 946709e82be0 satisfy the tabnanny Jeremy Hylton <jeremy@alum.mit.edu> parents: 13838 diff changeset	165 # Now check that __file__ isn't in one of the directories
946709e82be0 satisfy the tabnanny Jeremy Hylton <jeremy@alum.mit.edu> parents: 13838 diff changeset	166 if filename is None:
946709e82be0 satisfy the tabnanny Jeremy Hylton <jeremy@alum.mit.edu> parents: 13838 diff changeset	167 # must be a built-in, so we must ignore
946709e82be0 satisfy the tabnanny Jeremy Hylton <jeremy@alum.mit.edu> parents: 13838 diff changeset	168 self._ignore[modulename] = 1
946709e82be0 satisfy the tabnanny Jeremy Hylton <jeremy@alum.mit.edu> parents: 13838 diff changeset	169 return 1
13838 91d95a7b7858 Python code coverage tool by Skip Montanaro and Andrew Dalke Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	170
14657 946709e82be0 satisfy the tabnanny Jeremy Hylton <jeremy@alum.mit.edu> parents: 13838 diff changeset	171 # Ignore a file when it contains one of the ignorable paths
946709e82be0 satisfy the tabnanny Jeremy Hylton <jeremy@alum.mit.edu> parents: 13838 diff changeset	172 for d in self._dirs:
946709e82be0 satisfy the tabnanny Jeremy Hylton <jeremy@alum.mit.edu> parents: 13838 diff changeset	173 # The '+ os.sep' is to ensure that d is a parent directory,
946709e82be0 satisfy the tabnanny Jeremy Hylton <jeremy@alum.mit.edu> parents: 13838 diff changeset	174 # as compared to cases like:
946709e82be0 satisfy the tabnanny Jeremy Hylton <jeremy@alum.mit.edu> parents: 13838 diff changeset	175 # d = "/usr/local"
946709e82be0 satisfy the tabnanny Jeremy Hylton <jeremy@alum.mit.edu> parents: 13838 diff changeset	176 # filename = "/usr/local.py"
946709e82be0 satisfy the tabnanny Jeremy Hylton <jeremy@alum.mit.edu> parents: 13838 diff changeset	177 # or
946709e82be0 satisfy the tabnanny Jeremy Hylton <jeremy@alum.mit.edu> parents: 13838 diff changeset	178 # d = "/usr/local.py"
946709e82be0 satisfy the tabnanny Jeremy Hylton <jeremy@alum.mit.edu> parents: 13838 diff changeset	179 # filename = "/usr/local.py"
946709e82be0 satisfy the tabnanny Jeremy Hylton <jeremy@alum.mit.edu> parents: 13838 diff changeset	180 if string.find(filename, d + os.sep) == 0:
946709e82be0 satisfy the tabnanny Jeremy Hylton <jeremy@alum.mit.edu> parents: 13838 diff changeset	181 self._ignore[modulename] = 1
946709e82be0 satisfy the tabnanny Jeremy Hylton <jeremy@alum.mit.edu> parents: 13838 diff changeset	182 return 1
13838 91d95a7b7858 Python code coverage tool by Skip Montanaro and Andrew Dalke Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	183
14657 946709e82be0 satisfy the tabnanny Jeremy Hylton <jeremy@alum.mit.edu> parents: 13838 diff changeset	184 # Tried the different ways, so we don't ignore this module
946709e82be0 satisfy the tabnanny Jeremy Hylton <jeremy@alum.mit.edu> parents: 13838 diff changeset	185 self._ignore[modulename] = 0
946709e82be0 satisfy the tabnanny Jeremy Hylton <jeremy@alum.mit.edu> parents: 13838 diff changeset	186 return 0
13838 91d95a7b7858 Python code coverage tool by Skip Montanaro and Andrew Dalke Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	187
91d95a7b7858 Python code coverage tool by Skip Montanaro and Andrew Dalke Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	188 class CoverageResults:
21071 6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	189 def __init__(self, counts=None, calledfuncs=None, infile=None, outfile=None):
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	190 self.counts = counts
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	191 if self.counts is None:
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	192 self.counts = {}
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	193 self.counter = self.counts.copy() # map (filename, lineno) to count
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	194 self.calledfuncs = calledfuncs
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	195 if self.calledfuncs is None:
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	196 self.calledfuncs = {}
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	197 self.calledfuncs = self.calledfuncs.copy()
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	198 self.infile = infile
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	199 self.outfile = outfile
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	200 if self.infile:
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	201 # try and merge existing counts file
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	202 try:
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	203 thingie = pickle.load(open(self.infile, 'r'))
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	204 if type(thingie) is types.DictType:
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	205 # backwards compatibility for old trace.py after Zooko touched it but before calledfuncs --Zooko 2001-10-24
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	206 self.update(self.__class__(thingie))
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	207 elif type(thingie) is types.TupleType and len(thingie) == 2:
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	208 (counts, calledfuncs,) = thingie
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	209 self.update(self.__class__(counts, calledfuncs))
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	210 except (IOError, EOFError,):
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	211 pass
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	212 except pickle.UnpicklingError:
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	213 # backwards compatibility for old trace.py before Zooko touched it --Zooko 2001-10-24
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	214 self.update(self.__class__(marshal.load(open(self.infile))))
13838 91d95a7b7858 Python code coverage tool by Skip Montanaro and Andrew Dalke Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	215
91d95a7b7858 Python code coverage tool by Skip Montanaro and Andrew Dalke Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	216 def update(self, other):
14657 946709e82be0 satisfy the tabnanny Jeremy Hylton <jeremy@alum.mit.edu> parents: 13838 diff changeset	217 """Merge in the data from another CoverageResults"""
946709e82be0 satisfy the tabnanny Jeremy Hylton <jeremy@alum.mit.edu> parents: 13838 diff changeset	218 counts = self.counts
21071 6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	219 calledfuncs = self.calledfuncs
14657 946709e82be0 satisfy the tabnanny Jeremy Hylton <jeremy@alum.mit.edu> parents: 13838 diff changeset	220 other_counts = other.counts
21071 6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	221 other_calledfuncs = other.calledfuncs
13838 91d95a7b7858 Python code coverage tool by Skip Montanaro and Andrew Dalke Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	222
14657 946709e82be0 satisfy the tabnanny Jeremy Hylton <jeremy@alum.mit.edu> parents: 13838 diff changeset	223 for key in other_counts.keys():
21071 6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	224 if key != 'calledfuncs': # backwards compatibility for abortive attempt to stuff calledfuncs into self.counts, by Zooko --Zooko 2001-10-24
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	225 counts[key] = counts.get(key, 0) + other_counts[key]
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	226
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	227 for key in other_calledfuncs.keys():
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	228 calledfuncs[key] = 1
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	229
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	230 def write_results(self, show_missing = 1, summary = 0, coverdir = None):
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	231 """
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	232 @param coverdir
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	233 """
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	234 for (filename, modulename, funcname,) in self.calledfuncs.keys():
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	235 print "filename: %s, modulename: %s, funcname: %s" % (filename, modulename, funcname,)
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	236
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	237 import re
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	238 # turn the counts data ("(filename, lineno) = count") into something
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	239 # accessible on a per-file basis
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	240 per_file = {}
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	241 for thingie in self.counts.keys():
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	242 if thingie != "calledfuncs": # backwards compatibility for abortive attempt to stuff calledfuncs into self.counts, by Zooko --Zooko 2001-10-24
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	243 (filename, lineno,) = thingie
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	244 lines_hit = per_file[filename] = per_file.get(filename, {})
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	245 lines_hit[lineno] = self.counts[(filename, lineno)]
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	246
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	247 # there are many places where this is insufficient, like a blank
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	248 # line embedded in a multiline string.
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	249 blank = re.compile(r'^\s(#.)?$')
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	250
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	251 # accumulate summary info, if needed
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	252 sums = {}
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	253
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	254 # generate file paths for the coverage files we are going to write...
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	255 fnlist = []
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	256 tfdir = tempfile.gettempdir()
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	257 for key in per_file.keys():
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	258 filename = key
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	259
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	260 # skip some "files" we don't care about...
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	261 if filename == "<string>":
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	262 continue
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	263 # are these caused by code compiled using exec or something?
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	264 if filename.startswith(tfdir):
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	265 continue
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	266
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	267 modulename = inspect.getmodulename(filename)
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	268
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	269 if filename.endswith(".pyc") or filename.endswith(".pyo"):
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	270 filename = filename[:-1]
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	271
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	272 if coverdir:
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	273 thiscoverdir = coverdir
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	274 else:
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	275 thiscoverdir = os.path.dirname(os.path.abspath(filename))
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	276
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	277 # the code from here to "<<<" is the contents of the `fileutil.make_dirs()' function in the Mojo Nation project. --Zooko 2001-10-14
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	278 # http://cvs.sourceforge.net/cgi-bin/viewcvs.cgi/mojonation/evil/common/fileutil.py?rev=HEAD&content-type=text/vnd.viewcvs-markup
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	279 tx = None
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	280 try:
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	281 os.makedirs(thiscoverdir)
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	282 except OSError, x:
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	283 tx = x
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	284
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	285 if not os.path.isdir(thiscoverdir):
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	286 if tx:
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	287 raise tx
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	288 raise exceptions.IOError, "unknown error prevented creation of directory: %s" % thiscoverdir # careful not to construct an IOError with a 2-tuple, as that has a special meaning...
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	289 # <<<
13838 91d95a7b7858 Python code coverage tool by Skip Montanaro and Andrew Dalke Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	290
21071 6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	291 # build list file name by appending a ".cover" to the module name
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	292 # and sticking it into the specified directory
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	293 if "." in modulename:
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	294 # A module in a package
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	295 finalname = modulename.split(".")[-1]
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	296 listfilename = os.path.join(thiscoverdir, finalname + ".cover")
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	297 else:
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	298 listfilename = os.path.join(thiscoverdir, modulename + ".cover")
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	299
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	300 # Get the original lines from the .py file
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	301 try:
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	302 lines = open(filename, 'r').readlines()
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	303 except IOError, err:
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	304 sys.stderr.write("trace: Could not open %s for reading because: %s - skipping\n" % (`filename`, err))
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	305 continue
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	306
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	307 try:
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	308 outfile = open(listfilename, 'w')
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	309 except IOError, err:
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	310 sys.stderr.write(
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	311 '%s: Could not open %s for writing because: %s" \
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	312 "- skipping\n' % ("trace", `listfilename`, err))
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	313 continue
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	314
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	315 # If desired, get a list of the line numbers which represent
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	316 # executable content (returned as a dict for better lookup speed)
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	317 if show_missing:
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	318 executable_linenos = find_executable_linenos(filename)
14657 946709e82be0 satisfy the tabnanny Jeremy Hylton <jeremy@alum.mit.edu> parents: 13838 diff changeset	319 else:
21071 6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	320 executable_linenos = {}
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	321
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	322 n_lines = 0
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	323 n_hits = 0
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	324 lines_hit = per_file[key]
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	325 for i in range(len(lines)):
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	326 line = lines[i]
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	327
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	328 # do the blank/comment match to try to mark more lines
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	329 # (help the reader find stuff that hasn't been covered)
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	330 if lines_hit.has_key(i+1):
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	331 # count precedes the lines that we captured
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	332 outfile.write('%5d: ' % lines_hit[i+1])
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	333 n_hits = n_hits + 1
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	334 n_lines = n_lines + 1
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	335 elif blank.match(line):
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	336 # blank lines and comments are preceded by dots
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	337 outfile.write(' . ')
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	338 else:
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	339 # lines preceded by no marks weren't hit
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	340 # Highlight them if so indicated, unless the line contains
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	341 # '#pragma: NO COVER' (it is possible to embed this into
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	342 # the text as a non-comment; no easy fix)
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	343 if executable_linenos.has_key(i+1) and \
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	344 string.find(lines[i],
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	345 string.join(['#pragma', 'NO COVER'])) == -1:
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	346 outfile.write('>>>>>> ')
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	347 else:
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	348 outfile.write(' '*7)
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	349 n_lines = n_lines + 1
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	350 outfile.write(string.expandtabs(lines[i], 8))
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	351
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	352 outfile.close()
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	353
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	354 if summary and n_lines:
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	355 percent = int(100 * n_hits / n_lines)
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	356 sums[modulename] = n_lines, percent, modulename, filename
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	357
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	358 if summary and sums:
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	359 mods = sums.keys()
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	360 mods.sort()
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	361 print "lines cov% module (path)"
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	362 for m in mods:
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	363 n_lines, percent, modulename, filename = sums[m]
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	364 print "%5d %3d%% %s (%s)" % sums[m]
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	365
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	366 if self.outfile:
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	367 # try and store counts and module info into self.outfile
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	368 try:
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	369 pickle.dump((self.counts, self.calledfuncs,), open(self.outfile, 'w'), 1)
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	370 except IOError, err:
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	371 sys.stderr.write("cannot save counts files because %s" % err)
13838 91d95a7b7858 Python code coverage tool by Skip Montanaro and Andrew Dalke Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	372
91d95a7b7858 Python code coverage tool by Skip Montanaro and Andrew Dalke Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	373 # Given a code string, return the SET_LINENO information
91d95a7b7858 Python code coverage tool by Skip Montanaro and Andrew Dalke Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	374 def _find_LINENO_from_string(co_code):
91d95a7b7858 Python code coverage tool by Skip Montanaro and Andrew Dalke Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	375 """return all of the SET_LINENO information from a code string"""
91d95a7b7858 Python code coverage tool by Skip Montanaro and Andrew Dalke Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	376 import dis
91d95a7b7858 Python code coverage tool by Skip Montanaro and Andrew Dalke Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	377 linenos = {}
91d95a7b7858 Python code coverage tool by Skip Montanaro and Andrew Dalke Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	378
91d95a7b7858 Python code coverage tool by Skip Montanaro and Andrew Dalke Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	379 # This code was filched from the `dis' module then modified
91d95a7b7858 Python code coverage tool by Skip Montanaro and Andrew Dalke Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	380 n = len(co_code)
91d95a7b7858 Python code coverage tool by Skip Montanaro and Andrew Dalke Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	381 i = 0
91d95a7b7858 Python code coverage tool by Skip Montanaro and Andrew Dalke Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	382 prev_op = None
91d95a7b7858 Python code coverage tool by Skip Montanaro and Andrew Dalke Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	383 prev_lineno = 0
91d95a7b7858 Python code coverage tool by Skip Montanaro and Andrew Dalke Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	384 while i < n:
14657 946709e82be0 satisfy the tabnanny Jeremy Hylton <jeremy@alum.mit.edu> parents: 13838 diff changeset	385 c = co_code[i]
946709e82be0 satisfy the tabnanny Jeremy Hylton <jeremy@alum.mit.edu> parents: 13838 diff changeset	386 op = ord(c)
946709e82be0 satisfy the tabnanny Jeremy Hylton <jeremy@alum.mit.edu> parents: 13838 diff changeset	387 if op == dis.SET_LINENO:
946709e82be0 satisfy the tabnanny Jeremy Hylton <jeremy@alum.mit.edu> parents: 13838 diff changeset	388 if prev_op == op:
946709e82be0 satisfy the tabnanny Jeremy Hylton <jeremy@alum.mit.edu> parents: 13838 diff changeset	389 # two SET_LINENO in a row, so the previous didn't
946709e82be0 satisfy the tabnanny Jeremy Hylton <jeremy@alum.mit.edu> parents: 13838 diff changeset	390 # indicate anything. This occurs with triple
946709e82be0 satisfy the tabnanny Jeremy Hylton <jeremy@alum.mit.edu> parents: 13838 diff changeset	391 # quoted strings (?). Remove the old one.
946709e82be0 satisfy the tabnanny Jeremy Hylton <jeremy@alum.mit.edu> parents: 13838 diff changeset	392 del linenos[prev_lineno]
946709e82be0 satisfy the tabnanny Jeremy Hylton <jeremy@alum.mit.edu> parents: 13838 diff changeset	393 prev_lineno = ord(co_code[i+1]) + ord(co_code[i+2])*256
946709e82be0 satisfy the tabnanny Jeremy Hylton <jeremy@alum.mit.edu> parents: 13838 diff changeset	394 linenos[prev_lineno] = 1
946709e82be0 satisfy the tabnanny Jeremy Hylton <jeremy@alum.mit.edu> parents: 13838 diff changeset	395 if op >= dis.HAVE_ARGUMENT:
946709e82be0 satisfy the tabnanny Jeremy Hylton <jeremy@alum.mit.edu> parents: 13838 diff changeset	396 i = i + 3
946709e82be0 satisfy the tabnanny Jeremy Hylton <jeremy@alum.mit.edu> parents: 13838 diff changeset	397 else:
946709e82be0 satisfy the tabnanny Jeremy Hylton <jeremy@alum.mit.edu> parents: 13838 diff changeset	398 i = i + 1
946709e82be0 satisfy the tabnanny Jeremy Hylton <jeremy@alum.mit.edu> parents: 13838 diff changeset	399 prev_op = op
13838 91d95a7b7858 Python code coverage tool by Skip Montanaro and Andrew Dalke Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	400 return linenos
91d95a7b7858 Python code coverage tool by Skip Montanaro and Andrew Dalke Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	401
91d95a7b7858 Python code coverage tool by Skip Montanaro and Andrew Dalke Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	402 def _find_LINENO(code):
91d95a7b7858 Python code coverage tool by Skip Montanaro and Andrew Dalke Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	403 """return all of the SET_LINENO information from a code object"""
91d95a7b7858 Python code coverage tool by Skip Montanaro and Andrew Dalke Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	404 import types
91d95a7b7858 Python code coverage tool by Skip Montanaro and Andrew Dalke Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	405
91d95a7b7858 Python code coverage tool by Skip Montanaro and Andrew Dalke Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	406 # get all of the lineno information from the code of this scope level
91d95a7b7858 Python code coverage tool by Skip Montanaro and Andrew Dalke Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	407 linenos = _find_LINENO_from_string(code.co_code)
91d95a7b7858 Python code coverage tool by Skip Montanaro and Andrew Dalke Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	408
91d95a7b7858 Python code coverage tool by Skip Montanaro and Andrew Dalke Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	409 # and check the constants for references to other code objects
91d95a7b7858 Python code coverage tool by Skip Montanaro and Andrew Dalke Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	410 for c in code.co_consts:
14657 946709e82be0 satisfy the tabnanny Jeremy Hylton <jeremy@alum.mit.edu> parents: 13838 diff changeset	411 if type(c) == types.CodeType:
946709e82be0 satisfy the tabnanny Jeremy Hylton <jeremy@alum.mit.edu> parents: 13838 diff changeset	412 # find another code object, so recurse into it
946709e82be0 satisfy the tabnanny Jeremy Hylton <jeremy@alum.mit.edu> parents: 13838 diff changeset	413 linenos.update(_find_LINENO(c))
13838 91d95a7b7858 Python code coverage tool by Skip Montanaro and Andrew Dalke Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	414 return linenos
91d95a7b7858 Python code coverage tool by Skip Montanaro and Andrew Dalke Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	415
91d95a7b7858 Python code coverage tool by Skip Montanaro and Andrew Dalke Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	416 def find_executable_linenos(filename):
91d95a7b7858 Python code coverage tool by Skip Montanaro and Andrew Dalke Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	417 """return a dict of the line numbers from executable statements in a file
91d95a7b7858 Python code coverage tool by Skip Montanaro and Andrew Dalke Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	418
91d95a7b7858 Python code coverage tool by Skip Montanaro and Andrew Dalke Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	419 Works by finding all of the code-like objects in the module then searching
91d95a7b7858 Python code coverage tool by Skip Montanaro and Andrew Dalke Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	420 the byte code for 'SET_LINENO' terms (so this won't work one -O files).
91d95a7b7858 Python code coverage tool by Skip Montanaro and Andrew Dalke Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	421
91d95a7b7858 Python code coverage tool by Skip Montanaro and Andrew Dalke Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	422 """
91d95a7b7858 Python code coverage tool by Skip Montanaro and Andrew Dalke Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	423 import parser
91d95a7b7858 Python code coverage tool by Skip Montanaro and Andrew Dalke Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	424
17703 f3b8785bd022 Fix several bugs and add two features. Jeremy Hylton <jeremy@alum.mit.edu> parents: 15930 diff changeset	425 assert filename.endswith('.py')
f3b8785bd022 Fix several bugs and add two features. Jeremy Hylton <jeremy@alum.mit.edu> parents: 15930 diff changeset	426
13838 91d95a7b7858 Python code coverage tool by Skip Montanaro and Andrew Dalke Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	427 prog = open(filename).read()
91d95a7b7858 Python code coverage tool by Skip Montanaro and Andrew Dalke Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	428 ast = parser.suite(prog)
91d95a7b7858 Python code coverage tool by Skip Montanaro and Andrew Dalke Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	429 code = parser.compileast(ast, filename)
91d95a7b7858 Python code coverage tool by Skip Montanaro and Andrew Dalke Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	430
91d95a7b7858 Python code coverage tool by Skip Montanaro and Andrew Dalke Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	431 # The only way I know to find line numbers is to look for the
91d95a7b7858 Python code coverage tool by Skip Montanaro and Andrew Dalke Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	432 # SET_LINENO instructions. Isn't there some way to get it from
91d95a7b7858 Python code coverage tool by Skip Montanaro and Andrew Dalke Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	433 # the AST?
15930 14a27acc755a Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 14657 diff changeset	434
13838 91d95a7b7858 Python code coverage tool by Skip Montanaro and Andrew Dalke Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	435 return _find_LINENO(code)
91d95a7b7858 Python code coverage tool by Skip Montanaro and Andrew Dalke Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	436
91d95a7b7858 Python code coverage tool by Skip Montanaro and Andrew Dalke Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	437 ### XXX because os.path.commonprefix seems broken by my way of thinking...
91d95a7b7858 Python code coverage tool by Skip Montanaro and Andrew Dalke Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	438 def commonprefix(dirs):
91d95a7b7858 Python code coverage tool by Skip Montanaro and Andrew Dalke Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	439 "Given a list of pathnames, returns the longest common leading component"
91d95a7b7858 Python code coverage tool by Skip Montanaro and Andrew Dalke Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	440 if not dirs: return ''
91d95a7b7858 Python code coverage tool by Skip Montanaro and Andrew Dalke Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	441 n = copy.copy(dirs)
91d95a7b7858 Python code coverage tool by Skip Montanaro and Andrew Dalke Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	442 for i in range(len(n)):
91d95a7b7858 Python code coverage tool by Skip Montanaro and Andrew Dalke Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	443 n[i] = n[i].split(os.sep)
91d95a7b7858 Python code coverage tool by Skip Montanaro and Andrew Dalke Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	444 prefix = n[0]
91d95a7b7858 Python code coverage tool by Skip Montanaro and Andrew Dalke Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	445 for item in n:
91d95a7b7858 Python code coverage tool by Skip Montanaro and Andrew Dalke Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	446 for i in range(len(prefix)):
91d95a7b7858 Python code coverage tool by Skip Montanaro and Andrew Dalke Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	447 if prefix[:i+1] <> item[:i+1]:
91d95a7b7858 Python code coverage tool by Skip Montanaro and Andrew Dalke Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	448 prefix = prefix[:i]
91d95a7b7858 Python code coverage tool by Skip Montanaro and Andrew Dalke Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	449 if i == 0: return ''
91d95a7b7858 Python code coverage tool by Skip Montanaro and Andrew Dalke Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	450 break
91d95a7b7858 Python code coverage tool by Skip Montanaro and Andrew Dalke Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	451 return os.sep.join(prefix)
15930 14a27acc755a Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 14657 diff changeset	452
21071 6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	453 class Trace:
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	454 def __init__(self, count=1, trace=1, countfuncs=0, ignoremods=(), ignoredirs=(), infile=None, outfile=None):
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	455 """
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	456 @param count true iff it should count number of times each line is executed
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	457 @param trace true iff it should print out each line that is being counted
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	458 @param countfuncs true iff it should just output a list of (filename, modulename, funcname,) for functions that were called at least once; This overrides `count' and `trace'
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	459 @param ignoremods a list of the names of modules to ignore
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	460 @param ignoredirs a list of the names of directories to ignore all of the (recursive) contents of
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	461 @param infile file from which to read stored counts to be added into the results
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	462 @param outfile file in which to write the results
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	463 """
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	464 self.infile = infile
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	465 self.outfile = outfile
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	466 self.ignore = Ignore(ignoremods, ignoredirs)
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	467 self.counts = {} # keys are (filename, linenumber)
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	468 self.blabbed = {} # for debugging
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	469 self.pathtobasename = {} # for memoizing os.path.basename
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	470 self.donothing = 0
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	471 self.trace = trace
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	472 self._calledfuncs = {}
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	473 if countfuncs:
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	474 self.globaltrace = self.globaltrace_countfuncs
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	475 elif trace and count:
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	476 self.globaltrace = self.globaltrace_lt
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	477 self.localtrace = self.localtrace_trace_and_count
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	478 elif trace:
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	479 self.globaltrace = self.globaltrace_lt
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	480 self.localtrace = self.localtrace_trace
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	481 elif count:
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	482 self.globaltrace = self.globaltrace_lt
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	483 self.localtrace = self.localtrace_count
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	484 else:
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	485 # Ahem -- do nothing? Okay.
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	486 self.donothing = 1
13838 91d95a7b7858 Python code coverage tool by Skip Montanaro and Andrew Dalke Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	487
21071 6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	488 def run(self, cmd):
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	489 import __main__
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	490 dict = __main__.__dict__
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	491 if not self.donothing:
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	492 sys.settrace(self.globaltrace)
14657 946709e82be0 satisfy the tabnanny Jeremy Hylton <jeremy@alum.mit.edu> parents: 13838 diff changeset	493 try:
21071 6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	494 exec cmd in dict, dict
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	495 finally:
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	496 if not self.donothing:
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	497 sys.settrace(None)
13838 91d95a7b7858 Python code coverage tool by Skip Montanaro and Andrew Dalke Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	498
21071 6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	499 def runctx(self, cmd, globals=None, locals=None):
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	500 if globals is None: globals = {}
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	501 if locals is None: locals = {}
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	502 if not self.donothing:
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	503 sys.settrace(gself.lobaltrace)
14657 946709e82be0 satisfy the tabnanny Jeremy Hylton <jeremy@alum.mit.edu> parents: 13838 diff changeset	504 try:
21071 6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	505 exec cmd in dict, dict
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	506 finally:
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	507 if not self.donothing:
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	508 sys.settrace(None)
13838 91d95a7b7858 Python code coverage tool by Skip Montanaro and Andrew Dalke Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	509
21071 6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	510 def runfunc(self, func, args, *kw):
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	511 result = None
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	512 if not self.donothing:
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	513 sys.settrace(self.globaltrace)
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	514 try:
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	515 result = apply(func, args, kw)
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	516 finally:
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	517 if not self.donothing:
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	518 sys.settrace(None)
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	519 return result
13838 91d95a7b7858 Python code coverage tool by Skip Montanaro and Andrew Dalke Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	520
21071 6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	521 def globaltrace_countfuncs(self, frame, why, arg):
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	522 """
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	523 Handles `call' events (why == 'call') and adds the (filename, modulename, funcname,) to the self._calledfuncs dict.
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	524 """
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	525 if why == 'call':
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	526 (filename, lineno, funcname, context, lineindex,) = inspect.getframeinfo(frame, 0)
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	527 if filename:
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	528 modulename = inspect.getmodulename(filename)
14657 946709e82be0 satisfy the tabnanny Jeremy Hylton <jeremy@alum.mit.edu> parents: 13838 diff changeset	529 else:
21071 6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	530 modulename = None
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	531 self._calledfuncs[(filename, modulename, funcname,)] = 1
13838 91d95a7b7858 Python code coverage tool by Skip Montanaro and Andrew Dalke Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	532
21071 6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	533 def globaltrace_lt(self, frame, why, arg):
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	534 """
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	535 Handles `call' events (why == 'call') and if the code block being entered is to be ignored then it returns `None', else it returns `self.localtrace'.
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	536 """
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	537 if why == 'call':
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	538 (filename, lineno, funcname, context, lineindex,) = inspect.getframeinfo(frame, 0)
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	539 # if DEBUG_MODE and not filename:
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	540 # print "%s.globaltrace(frame: %s, why: %s, arg: %s): filename: %s, lineno: %s, funcname: %s, context: %s, lineindex: %s\n" % (self, frame, why, arg, filename, lineno, funcname, context, lineindex,)
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	541 if filename:
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	542 modulename = inspect.getmodulename(filename)
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	543 ignore_it = self.ignore.names(filename, modulename)
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	544 # if DEBUG_MODE and not self.blabbed.has_key((filename, modulename,)):
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	545 # self.blabbed[(filename, modulename,)] = None
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	546 # print "%s.globaltrace(frame: %s, why: %s, arg: %s, filename: %s, modulename: %s, ignore_it: %s\n" % (self, frame, why, arg, filename, modulename, ignore_it,)
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	547 if not ignore_it:
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	548 if self.trace:
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	549 print " --- modulename: %s, funcname: %s" % (modulename, funcname,)
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	550 # if DEBUG_MODE:
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	551 # print "%s.globaltrace(frame: %s, why: %s, arg: %s, filename: %s, modulename: %s, ignore_it: %s -- about to localtrace\n" % (self, frame, why, arg, filename, modulename, ignore_it,)
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	552 return self.localtrace
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	553 else:
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	554 # XXX why no filename?
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	555 return None
13838 91d95a7b7858 Python code coverage tool by Skip Montanaro and Andrew Dalke Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	556
21071 6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	557 def localtrace_trace_and_count(self, frame, why, arg):
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	558 if why == 'line':
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	559 # record the file name and line number of every trace
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	560 # XXX I wish inspect offered me an optimized `getfilename(frame)' to use in place of the presumably heavier `getframeinfo()'. --Zooko 2001-10-14
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	561 (filename, lineno, funcname, context, lineindex,) = inspect.getframeinfo(frame, 1)
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	562 key = (filename, lineno,)
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	563 self.counts[key] = self.counts.get(key, 0) + 1
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	564 # XXX not convinced that this memoizing is a performance win -- I don't know enough about Python guts to tell. --Zooko 2001-10-14
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	565 bname = self.pathtobasename.get(filename)
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	566 if bname is None:
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	567 # Using setdefault faster than two separate lines? --Zooko 2001-10-14
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	568 bname = self.pathtobasename.setdefault(filename, os.path.basename(filename))
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	569 try:
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	570 print "%s(%d): %s" % (bname, lineno, context[lineindex],),
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	571 except IndexError:
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	572 # Uh.. sometimes getframeinfo gives me a context of length 1 and a lineindex of -2. Oh well.
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	573 pass
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	574 return self.localtrace
17703 f3b8785bd022 Fix several bugs and add two features. Jeremy Hylton <jeremy@alum.mit.edu> parents: 15930 diff changeset	575
21071 6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	576 def localtrace_trace(self, frame, why, arg):
14657 946709e82be0 satisfy the tabnanny Jeremy Hylton <jeremy@alum.mit.edu> parents: 13838 diff changeset	577 if why == 'line':
21071 6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	578 # XXX shouldn't do the count increment when arg is exception? But be careful to return self.localtrace when arg is exception! ? --Zooko 2001-10-14
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	579 # record the file name and line number of every trace
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	580 # XXX I wish inspect offered me an optimized `getfilename(frame)' to use in place of the presumably heavier `getframeinfo()'. --Zooko 2001-10-14
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	581 (filename, lineno, funcname, context, lineindex,) = inspect.getframeinfo(frame)
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	582 # if DEBUG_MODE:
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	583 # print "%s.localtrace_trace(frame: %s, why: %s, arg: %s); filename: %s, lineno: %s, funcname: %s, context: %s, lineindex: %s\n" % (self, frame, why, arg, filename, lineno, funcname, context, lineindex,)
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	584 # XXX not convinced that this memoizing is a performance win -- I don't know enough about Python guts to tell. --Zooko 2001-10-14
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	585 bname = self.pathtobasename.get(filename)
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	586 if bname is None:
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	587 # Using setdefault faster than two separate lines? --Zooko 2001-10-14
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	588 bname = self.pathtobasename.setdefault(filename, os.path.basename(filename))
17703 f3b8785bd022 Fix several bugs and add two features. Jeremy Hylton <jeremy@alum.mit.edu> parents: 15930 diff changeset	589 try:
21071 6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	590 print "%s(%d): %s" % (bname, lineno, context[lineindex],),
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	591 except IndexError:
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	592 # Uh.. sometimes getframeinfo gives me a context of length 1 and a lineindex of -2. Oh well.
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	593 pass
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	594 return self.localtrace
13838 91d95a7b7858 Python code coverage tool by Skip Montanaro and Andrew Dalke Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	595
21071 6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	596 def localtrace_count(self, frame, why, arg):
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	597 if why == 'line':
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	598 # XXX shouldn't do the count increment when arg is exception? But be careful to return self.localtrace when arg is exception! ? --Zooko 2001-10-14
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	599 # record the file name and line number of every trace
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	600 # XXX I wish inspect offered me an optimized `getfilename(frame)' to use in place of the presumably heavier `getframeinfo()'. --Zooko 2001-10-14
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	601 (filename, lineno, funcname, context, lineindex,) = inspect.getframeinfo(frame)
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	602 key = (filename, lineno,)
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	603 self.counts[key] = self.counts.get(key, 0) + 1
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	604 return self.localtrace
13838 91d95a7b7858 Python code coverage tool by Skip Montanaro and Andrew Dalke Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	605
91d95a7b7858 Python code coverage tool by Skip Montanaro and Andrew Dalke Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	606 def results(self):
21071 6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	607 return CoverageResults(self.counts, infile=self.infile, outfile=self.outfile, calledfuncs=self._calledfuncs)
13838 91d95a7b7858 Python code coverage tool by Skip Montanaro and Andrew Dalke Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	608
91d95a7b7858 Python code coverage tool by Skip Montanaro and Andrew Dalke Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	609 def _err_exit(msg):
21071 6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	610 sys.stderr.write("%s: %s\n" % (sys.argv[0], msg))
14657 946709e82be0 satisfy the tabnanny Jeremy Hylton <jeremy@alum.mit.edu> parents: 13838 diff changeset	611 sys.exit(1)
13838 91d95a7b7858 Python code coverage tool by Skip Montanaro and Andrew Dalke Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	612
21071 6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	613 def main(argv=None):
13838 91d95a7b7858 Python code coverage tool by Skip Montanaro and Andrew Dalke Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	614 import getopt
91d95a7b7858 Python code coverage tool by Skip Montanaro and Andrew Dalke Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	615
91d95a7b7858 Python code coverage tool by Skip Montanaro and Andrew Dalke Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	616 if argv is None:
14657 946709e82be0 satisfy the tabnanny Jeremy Hylton <jeremy@alum.mit.edu> parents: 13838 diff changeset	617 argv = sys.argv
13838 91d95a7b7858 Python code coverage tool by Skip Montanaro and Andrew Dalke Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	618 try:
21071 6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	619 opts, prog_argv = getopt.getopt(argv[1:], "tcrRf:d:msC:l",
14657 946709e82be0 satisfy the tabnanny Jeremy Hylton <jeremy@alum.mit.edu> parents: 13838 diff changeset	620 ["help", "version", "trace", "count",
946709e82be0 satisfy the tabnanny Jeremy Hylton <jeremy@alum.mit.edu> parents: 13838 diff changeset	621 "report", "no-report",
21071 6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	622 "file=", "missing",
17703 f3b8785bd022 Fix several bugs and add two features. Jeremy Hylton <jeremy@alum.mit.edu> parents: 15930 diff changeset	623 "ignore-module=", "ignore-dir=",
21071 6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	624 "coverdir=", "listfuncs",])
13838 91d95a7b7858 Python code coverage tool by Skip Montanaro and Andrew Dalke Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	625
91d95a7b7858 Python code coverage tool by Skip Montanaro and Andrew Dalke Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	626 except getopt.error, msg:
21071 6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	627 sys.stderr.write("%s: %s\n" % (sys.argv[0], msg))
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	628 sys.stderr.write("Try `%s --help' for more information\n" % sys.argv[0])
14657 946709e82be0 satisfy the tabnanny Jeremy Hylton <jeremy@alum.mit.edu> parents: 13838 diff changeset	629 sys.exit(1)
13838 91d95a7b7858 Python code coverage tool by Skip Montanaro and Andrew Dalke Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	630
91d95a7b7858 Python code coverage tool by Skip Montanaro and Andrew Dalke Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	631 trace = 0
91d95a7b7858 Python code coverage tool by Skip Montanaro and Andrew Dalke Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	632 count = 0
91d95a7b7858 Python code coverage tool by Skip Montanaro and Andrew Dalke Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	633 report = 0
91d95a7b7858 Python code coverage tool by Skip Montanaro and Andrew Dalke Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	634 no_report = 0
91d95a7b7858 Python code coverage tool by Skip Montanaro and Andrew Dalke Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	635 counts_file = None
91d95a7b7858 Python code coverage tool by Skip Montanaro and Andrew Dalke Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	636 missing = 0
91d95a7b7858 Python code coverage tool by Skip Montanaro and Andrew Dalke Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	637 ignore_modules = []
91d95a7b7858 Python code coverage tool by Skip Montanaro and Andrew Dalke Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	638 ignore_dirs = []
17703 f3b8785bd022 Fix several bugs and add two features. Jeremy Hylton <jeremy@alum.mit.edu> parents: 15930 diff changeset	639 coverdir = None
f3b8785bd022 Fix several bugs and add two features. Jeremy Hylton <jeremy@alum.mit.edu> parents: 15930 diff changeset	640 summary = 0
21071 6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	641 listfuncs = false
13838 91d95a7b7858 Python code coverage tool by Skip Montanaro and Andrew Dalke Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	642
91d95a7b7858 Python code coverage tool by Skip Montanaro and Andrew Dalke Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	643 for opt, val in opts:
14657 946709e82be0 satisfy the tabnanny Jeremy Hylton <jeremy@alum.mit.edu> parents: 13838 diff changeset	644 if opt == "--help":
946709e82be0 satisfy the tabnanny Jeremy Hylton <jeremy@alum.mit.edu> parents: 13838 diff changeset	645 usage(sys.stdout)
946709e82be0 satisfy the tabnanny Jeremy Hylton <jeremy@alum.mit.edu> parents: 13838 diff changeset	646 sys.exit(0)
946709e82be0 satisfy the tabnanny Jeremy Hylton <jeremy@alum.mit.edu> parents: 13838 diff changeset	647
946709e82be0 satisfy the tabnanny Jeremy Hylton <jeremy@alum.mit.edu> parents: 13838 diff changeset	648 if opt == "--version":
946709e82be0 satisfy the tabnanny Jeremy Hylton <jeremy@alum.mit.edu> parents: 13838 diff changeset	649 sys.stdout.write("trace 2.0\n")
946709e82be0 satisfy the tabnanny Jeremy Hylton <jeremy@alum.mit.edu> parents: 13838 diff changeset	650 sys.exit(0)
946709e82be0 satisfy the tabnanny Jeremy Hylton <jeremy@alum.mit.edu> parents: 13838 diff changeset	651
21071 6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	652 if opt == "-l" or opt == "--listfuncs":
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	653 listfuncs = true
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	654 continue
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	655
14657 946709e82be0 satisfy the tabnanny Jeremy Hylton <jeremy@alum.mit.edu> parents: 13838 diff changeset	656 if opt == "-t" or opt == "--trace":
946709e82be0 satisfy the tabnanny Jeremy Hylton <jeremy@alum.mit.edu> parents: 13838 diff changeset	657 trace = 1
946709e82be0 satisfy the tabnanny Jeremy Hylton <jeremy@alum.mit.edu> parents: 13838 diff changeset	658 continue
13838 91d95a7b7858 Python code coverage tool by Skip Montanaro and Andrew Dalke Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	659
14657 946709e82be0 satisfy the tabnanny Jeremy Hylton <jeremy@alum.mit.edu> parents: 13838 diff changeset	660 if opt == "-c" or opt == "--count":
946709e82be0 satisfy the tabnanny Jeremy Hylton <jeremy@alum.mit.edu> parents: 13838 diff changeset	661 count = 1
946709e82be0 satisfy the tabnanny Jeremy Hylton <jeremy@alum.mit.edu> parents: 13838 diff changeset	662 continue
946709e82be0 satisfy the tabnanny Jeremy Hylton <jeremy@alum.mit.edu> parents: 13838 diff changeset	663
946709e82be0 satisfy the tabnanny Jeremy Hylton <jeremy@alum.mit.edu> parents: 13838 diff changeset	664 if opt == "-r" or opt == "--report":
946709e82be0 satisfy the tabnanny Jeremy Hylton <jeremy@alum.mit.edu> parents: 13838 diff changeset	665 report = 1
946709e82be0 satisfy the tabnanny Jeremy Hylton <jeremy@alum.mit.edu> parents: 13838 diff changeset	666 continue
13838 91d95a7b7858 Python code coverage tool by Skip Montanaro and Andrew Dalke Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	667
14657 946709e82be0 satisfy the tabnanny Jeremy Hylton <jeremy@alum.mit.edu> parents: 13838 diff changeset	668 if opt == "-R" or opt == "--no-report":
946709e82be0 satisfy the tabnanny Jeremy Hylton <jeremy@alum.mit.edu> parents: 13838 diff changeset	669 no_report = 1
946709e82be0 satisfy the tabnanny Jeremy Hylton <jeremy@alum.mit.edu> parents: 13838 diff changeset	670 continue
13838 91d95a7b7858 Python code coverage tool by Skip Montanaro and Andrew Dalke Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	671
14657 946709e82be0 satisfy the tabnanny Jeremy Hylton <jeremy@alum.mit.edu> parents: 13838 diff changeset	672 if opt == "-f" or opt == "--file":
946709e82be0 satisfy the tabnanny Jeremy Hylton <jeremy@alum.mit.edu> parents: 13838 diff changeset	673 counts_file = val
946709e82be0 satisfy the tabnanny Jeremy Hylton <jeremy@alum.mit.edu> parents: 13838 diff changeset	674 continue
13838 91d95a7b7858 Python code coverage tool by Skip Montanaro and Andrew Dalke Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	675
14657 946709e82be0 satisfy the tabnanny Jeremy Hylton <jeremy@alum.mit.edu> parents: 13838 diff changeset	676 if opt == "-m" or opt == "--missing":
946709e82be0 satisfy the tabnanny Jeremy Hylton <jeremy@alum.mit.edu> parents: 13838 diff changeset	677 missing = 1
946709e82be0 satisfy the tabnanny Jeremy Hylton <jeremy@alum.mit.edu> parents: 13838 diff changeset	678 continue
13838 91d95a7b7858 Python code coverage tool by Skip Montanaro and Andrew Dalke Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	679
17703 f3b8785bd022 Fix several bugs and add two features. Jeremy Hylton <jeremy@alum.mit.edu> parents: 15930 diff changeset	680 if opt == "-C" or opt == "--coverdir":
f3b8785bd022 Fix several bugs and add two features. Jeremy Hylton <jeremy@alum.mit.edu> parents: 15930 diff changeset	681 coverdir = val
f3b8785bd022 Fix several bugs and add two features. Jeremy Hylton <jeremy@alum.mit.edu> parents: 15930 diff changeset	682 continue
f3b8785bd022 Fix several bugs and add two features. Jeremy Hylton <jeremy@alum.mit.edu> parents: 15930 diff changeset	683
f3b8785bd022 Fix several bugs and add two features. Jeremy Hylton <jeremy@alum.mit.edu> parents: 15930 diff changeset	684 if opt == "-s" or opt == "--summary":
f3b8785bd022 Fix several bugs and add two features. Jeremy Hylton <jeremy@alum.mit.edu> parents: 15930 diff changeset	685 summary = 1
f3b8785bd022 Fix several bugs and add two features. Jeremy Hylton <jeremy@alum.mit.edu> parents: 15930 diff changeset	686 continue
f3b8785bd022 Fix several bugs and add two features. Jeremy Hylton <jeremy@alum.mit.edu> parents: 15930 diff changeset	687
14657 946709e82be0 satisfy the tabnanny Jeremy Hylton <jeremy@alum.mit.edu> parents: 13838 diff changeset	688 if opt == "--ignore-module":
946709e82be0 satisfy the tabnanny Jeremy Hylton <jeremy@alum.mit.edu> parents: 13838 diff changeset	689 ignore_modules.append(val)
946709e82be0 satisfy the tabnanny Jeremy Hylton <jeremy@alum.mit.edu> parents: 13838 diff changeset	690 continue
13838 91d95a7b7858 Python code coverage tool by Skip Montanaro and Andrew Dalke Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	691
14657 946709e82be0 satisfy the tabnanny Jeremy Hylton <jeremy@alum.mit.edu> parents: 13838 diff changeset	692 if opt == "--ignore-dir":
946709e82be0 satisfy the tabnanny Jeremy Hylton <jeremy@alum.mit.edu> parents: 13838 diff changeset	693 for s in string.split(val, os.pathsep):
946709e82be0 satisfy the tabnanny Jeremy Hylton <jeremy@alum.mit.edu> parents: 13838 diff changeset	694 s = os.path.expandvars(s)
946709e82be0 satisfy the tabnanny Jeremy Hylton <jeremy@alum.mit.edu> parents: 13838 diff changeset	695 # should I also call expanduser? (after all, could use $HOME)
13838 91d95a7b7858 Python code coverage tool by Skip Montanaro and Andrew Dalke Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	696
14657 946709e82be0 satisfy the tabnanny Jeremy Hylton <jeremy@alum.mit.edu> parents: 13838 diff changeset	697 s = string.replace(s, "$prefix",
946709e82be0 satisfy the tabnanny Jeremy Hylton <jeremy@alum.mit.edu> parents: 13838 diff changeset	698 os.path.join(sys.prefix, "lib",
946709e82be0 satisfy the tabnanny Jeremy Hylton <jeremy@alum.mit.edu> parents: 13838 diff changeset	699 "python" + sys.version[:3]))
946709e82be0 satisfy the tabnanny Jeremy Hylton <jeremy@alum.mit.edu> parents: 13838 diff changeset	700 s = string.replace(s, "$exec_prefix",
946709e82be0 satisfy the tabnanny Jeremy Hylton <jeremy@alum.mit.edu> parents: 13838 diff changeset	701 os.path.join(sys.exec_prefix, "lib",
946709e82be0 satisfy the tabnanny Jeremy Hylton <jeremy@alum.mit.edu> parents: 13838 diff changeset	702 "python" + sys.version[:3]))
946709e82be0 satisfy the tabnanny Jeremy Hylton <jeremy@alum.mit.edu> parents: 13838 diff changeset	703 s = os.path.normpath(s)
946709e82be0 satisfy the tabnanny Jeremy Hylton <jeremy@alum.mit.edu> parents: 13838 diff changeset	704 ignore_dirs.append(s)
946709e82be0 satisfy the tabnanny Jeremy Hylton <jeremy@alum.mit.edu> parents: 13838 diff changeset	705 continue
13838 91d95a7b7858 Python code coverage tool by Skip Montanaro and Andrew Dalke Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	706
14657 946709e82be0 satisfy the tabnanny Jeremy Hylton <jeremy@alum.mit.edu> parents: 13838 diff changeset	707 assert 0, "Should never get here"
13838 91d95a7b7858 Python code coverage tool by Skip Montanaro and Andrew Dalke Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	708
21071 6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	709 if listfuncs and (count or trace):
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	710 _err_exit("cannot specify both --listfuncs and (--trace or --count)")
13838 91d95a7b7858 Python code coverage tool by Skip Montanaro and Andrew Dalke Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	711
21071 6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	712 if not count and not trace and not report and not listfuncs:
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	713 _err_exit("must specify one of --trace, --count, --report or --listfuncs")
13838 91d95a7b7858 Python code coverage tool by Skip Montanaro and Andrew Dalke Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	714
91d95a7b7858 Python code coverage tool by Skip Montanaro and Andrew Dalke Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	715 if report and no_report:
14657 946709e82be0 satisfy the tabnanny Jeremy Hylton <jeremy@alum.mit.edu> parents: 13838 diff changeset	716 _err_exit("cannot specify both --report and --no-report")
13838 91d95a7b7858 Python code coverage tool by Skip Montanaro and Andrew Dalke Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	717
21071 6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	718 if report and not counts_file:
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	719 _err_exit("--report requires a --file")
13838 91d95a7b7858 Python code coverage tool by Skip Montanaro and Andrew Dalke Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	720
21071 6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	721 if no_report and len(prog_argv) == 0:
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	722 _err_exit("missing name of file to run")
13838 91d95a7b7858 Python code coverage tool by Skip Montanaro and Andrew Dalke Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	723
91d95a7b7858 Python code coverage tool by Skip Montanaro and Andrew Dalke Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	724 # everything is ready
21071 6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	725 if report:
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	726 results = CoverageResults(infile=counts_file, outfile=counts_file)
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	727 results.write_results(missing, summary=summary, coverdir=coverdir)
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	728 else:
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	729 sys.argv = prog_argv
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	730 progname = prog_argv[0]
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	731 if eval(sys.version[:3])>1.3:
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	732 sys.path[0] = os.path.split(progname)[0] # ???
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	733
6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	734 t = Trace(count, trace, countfuncs=listfuncs, ignoremods=ignore_modules, ignoredirs=ignore_dirs, infile=counts_file, outfile=counts_file)
14657 946709e82be0 satisfy the tabnanny Jeremy Hylton <jeremy@alum.mit.edu> parents: 13838 diff changeset	735 try:
21071 6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	736 t.run('execfile(' + `progname` + ')')
14657 946709e82be0 satisfy the tabnanny Jeremy Hylton <jeremy@alum.mit.edu> parents: 13838 diff changeset	737 except IOError, err:
21071 6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	738 _err_exit("Cannot run file %s because: %s" % (`sys.argv[0]`, err))
14657 946709e82be0 satisfy the tabnanny Jeremy Hylton <jeremy@alum.mit.edu> parents: 13838 diff changeset	739 except SystemExit:
946709e82be0 satisfy the tabnanny Jeremy Hylton <jeremy@alum.mit.edu> parents: 13838 diff changeset	740 pass
13838 91d95a7b7858 Python code coverage tool by Skip Montanaro and Andrew Dalke Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	741
14657 946709e82be0 satisfy the tabnanny Jeremy Hylton <jeremy@alum.mit.edu> parents: 13838 diff changeset	742 results = t.results()
13838 91d95a7b7858 Python code coverage tool by Skip Montanaro and Andrew Dalke Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	743
14657 946709e82be0 satisfy the tabnanny Jeremy Hylton <jeremy@alum.mit.edu> parents: 13838 diff changeset	744 if not no_report:
21071 6392f03222e0 Checking in Zooko's version per SF patch #476866, plus my changes to Guido van Rossum <guido@python.org> parents: 20322 diff changeset	745 results.write_results(missing, summary=summary, coverdir=coverdir)
13838 91d95a7b7858 Python code coverage tool by Skip Montanaro and Andrew Dalke Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	746
91d95a7b7858 Python code coverage tool by Skip Montanaro and Andrew Dalke Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	747 if __name__=='__main__':
91d95a7b7858 Python code coverage tool by Skip Montanaro and Andrew Dalke Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	748 main()

Mercurial > cpython-lru_cache-weakref

annotate Tools/scripts/trace.py @ 24042:63fbbed10714 legacy-trunk