--- /dev/null
+#!/usr/bin/env python
+
+# Copyright 2000, Mojam Media, Inc., all rights reserved.
+# Author: Skip Montanaro
+#
+# Copyright 1999, Bioreason, Inc., all rights reserved.
+# Author: Andrew Dalke
+#
+# Copyright 1995-1997, Automatrix, Inc., all rights reserved.
+# Author: Skip Montanaro
+#
+# Copyright 1991-1995, Stichting Mathematisch Centrum, all rights reserved.
+#
+#
+# Permission to use, copy, modify, and distribute this Python software and
+# its associated documentation for any purpose without fee is hereby
+# granted, provided that the above copyright notice appears in all copies,
+# and that both that copyright notice and this permission notice appear in
+# supporting documentation, and that the name of neither Automatrix,
+# Bioreason or Mojam Media be used in advertising or publicity pertaining to
+# distribution of the software without specific, written prior permission.
+#
+#
+# Summary of recent changes:
+# Support for files with the same basename (submodules in packages)
+# Expanded the idea of how to ignore files or modules
+# Split tracing and counting into different classes
+# Extracted count information and reporting from the count class
+# Added some ability to detect which missing lines could be executed
+# Added pseudo-pragma to prohibit complaining about unexecuted lines
+# Rewrote the main program
+
+# Summary of older changes:
+# Added run-time display of statements being executed
+# Incorporated portability and performance fixes from Greg Stein
+# Incorporated main program from Michael Scharf
+
+"""
+program/module to trace Python program or function execution
+
+Sample use, command line:
+ trace.py -c -f counts --ignore-dir '$prefix' spam.py eggs
+ trace.py -t --ignore-dir '$prefix' spam.py eggs
+
+Sample use, programmatically (still more complicated than it should be)
+ # create an Ignore option, telling it what you want to ignore
+ ignore = trace.Ignore(dirs = [sys.prefix, sys.exec_prefix])
+ # create a Coverage object, telling it what to ignore
+ coverage = trace.Coverage(ignore)
+ # run the new command using the given trace
+ trace.run(coverage.trace, 'main()')
+
+ # make a report, telling it where you want output
+ t = trace.create_results_log(coverage.results(),
+ '/usr/local/Automatrix/concerts/coverage')
+ show_missing = 1)
+
+ The Trace class can be instantited instead of the Coverage class if
+ runtime display of executable lines is desired instead of statement
+ converage measurement.
+"""
+
+import sys, os, string, marshal, tempfile, copy, operator
+
+def usage(outfile):
+ outfile.write("""Usage: %s [OPTIONS] <file> [ARGS]
+
+Execution:
+ --help Display this help then exit.
+ --version Output version information then exit.
+ -t,--trace Print the line to be executed to sys.stdout.
+ -c,--count Count the number of times a line is executed.
+ Results are written in the results file, if given.
+ -r,--report Generate a report from a results file; do not
+ execute any code.
+ (One of `-t', `-c' or `-r' must be specified)
+
+I/O:
+ -f,--file= File name for accumulating results over several runs.
+ (No file name means do not archive results)
+ -d,--logdir= Directory to use when writing annotated log files.
+ Log files are the module __name__ with `.` replaced
+ by os.sep and with '.pyl' added.
+ -m,--missing Annotate all executable lines which were not executed
+ with a '>>>>>> '.
+ -R,--no-report Do not generate the annotated reports. Useful if
+ you want to accumulate several over tests.
+
+Selection: Do not trace or log lines from ...
+ --ignore-module=[string] modules with the given __name__, and submodules
+ of that module
+ --ignore-dir=[string] files in the stated directory (multiple
+ directories can be joined by os.pathsep)
+
+ The selection options can be listed multiple times to ignore different
+modules.
+""" % sys.argv[0])
+
+
+class Ignore:
+ def __init__(self, modules = None, dirs = None):
+ self._mods = modules or []
+ self._dirs = dirs or []
+
+ self._ignore = { '<string>': 1 }
+
+
+ def names(self, filename, modulename):
+ if self._ignore.has_key(modulename):
+ return self._ignore[modulename]
+
+ # haven't seen this one before, so see if the module name is
+ # on the ignore list. Need to take some care since ignoring
+ # "cmp" musn't mean ignoring "cmpcache" but ignoring
+ # "Spam" must also mean ignoring "Spam.Eggs".
+ for mod in self._mods:
+ if mod == modulename: # Identical names, so ignore
+ self._ignore[modulename] = 1
+ return 1
+ # check if the module is a proper submodule of something on
+ # the ignore list
+ n = len(mod)
+ # (will not overflow since if the first n characters are the
+ # same and the name has not already occured, then the size
+ # of "name" is greater than that of "mod")
+ if mod == modulename[:n] and modulename[n] == '.':
+ self._ignore[modulename] = 1
+ return 1
+
+ # Now check that __file__ isn't in one of the directories
+ if filename is None:
+ # must be a built-in, so we must ignore
+ self._ignore[modulename] = 1
+ return 1
+
+ # Ignore a file when it contains one of the ignorable paths
+ for d in self._dirs:
+ # The '+ os.sep' is to ensure that d is a parent directory,
+ # as compared to cases like:
+ # d = "/usr/local"
+ # filename = "/usr/local.py"
+ # or
+ # d = "/usr/local.py"
+ # filename = "/usr/local.py"
+ if string.find(filename, d + os.sep) == 0:
+ self._ignore[modulename] = 1
+ return 1
+
+ # Tried the different ways, so we don't ignore this module
+ self._ignore[modulename] = 0
+ return 0
+
+
+def run(trace, cmd):
+ import __main__
+ dict = __main__.__dict__
+ sys.settrace(trace)
+ try:
+ exec cmd in dict, dict
+ finally:
+ sys.settrace(None)
+
+def runctx(trace, cmd, globals=None, locals=None):
+ if globals is None: globals = {}
+ if locals is None: locals = {}
+ sys.settrace(trace)
+ try:
+ exec cmd in dict, dict
+ finally:
+ sys.settrace(None)
+
+def runfunc(trace, func, *args, **kw):
+ result = None
+ sys.settrace(trace)
+ try:
+ result = apply(func, args, kw)
+ finally:
+ sys.settrace(None)
+ return result
+
+
+class CoverageResults:
+ def __init__(self, counts = {}, modules = {}):
+ self.counts = counts.copy() # map (filename, lineno) to count
+ self.modules = modules.copy() # map filenames to modules
+
+ def update(self, other):
+ """Merge in the data from another CoverageResults"""
+ counts = self.counts
+ other_counts = other.counts
+ modules = self.modules
+ other_modules = other.modules
+
+ for key in other_counts.keys():
+ counts[key] = counts.get(key, 0) + other_counts[key]
+
+ for key in other_modules.keys():
+ if modules.has_key(key):
+ # make sure they point to the same file
+ assert modules[key] == other_modules[key], \
+ "Strange! filename %s has two different module names" % \
+ (key, modules[key], other_module[key])
+ else:
+ modules[key] = other_modules[key]
+
+# Given a code string, return the SET_LINENO information
+def _find_LINENO_from_string(co_code):
+ """return all of the SET_LINENO information from a code string"""
+ import dis
+ linenos = {}
+
+ # This code was filched from the `dis' module then modified
+ n = len(co_code)
+ i = 0
+ prev_op = None
+ prev_lineno = 0
+ while i < n:
+ c = co_code[i]
+ op = ord(c)
+ if op == dis.SET_LINENO:
+ if prev_op == op:
+ # two SET_LINENO in a row, so the previous didn't
+ # indicate anything. This occurs with triple
+ # quoted strings (?). Remove the old one.
+ del linenos[prev_lineno]
+ prev_lineno = ord(co_code[i+1]) + ord(co_code[i+2])*256
+ linenos[prev_lineno] = 1
+ if op >= dis.HAVE_ARGUMENT:
+ i = i + 3
+ else:
+ i = i + 1
+ prev_op = op
+ return linenos
+
+def _find_LINENO(code):
+ """return all of the SET_LINENO information from a code object"""
+ import types
+
+ # get all of the lineno information from the code of this scope level
+ linenos = _find_LINENO_from_string(code.co_code)
+
+ # and check the constants for references to other code objects
+ for c in code.co_consts:
+ if type(c) == types.CodeType:
+ # find another code object, so recurse into it
+ linenos.update(_find_LINENO(c))
+ return linenos
+
+def find_executable_linenos(filename):
+ """return a dict of the line numbers from executable statements in a file
+
+ Works by finding all of the code-like objects in the module then searching
+ the byte code for 'SET_LINENO' terms (so this won't work one -O files).
+
+ """
+ import parser
+
+ prog = open(filename).read()
+ ast = parser.suite(prog)
+ code = parser.compileast(ast, filename)
+
+ # The only way I know to find line numbers is to look for the
+ # SET_LINENO instructions. Isn't there some way to get it from
+ # the AST?
+
+ return _find_LINENO(code)
+
+### XXX because os.path.commonprefix seems broken by my way of thinking...
+def commonprefix(dirs):
+ "Given a list of pathnames, returns the longest common leading component"
+ if not dirs: return ''
+ n = copy.copy(dirs)
+ for i in range(len(n)):
+ n[i] = n[i].split(os.sep)
+ prefix = n[0]
+ for item in n:
+ for i in range(len(prefix)):
+ if prefix[:i+1] <> item[:i+1]:
+ prefix = prefix[:i]
+ if i == 0: return ''
+ break
+ return os.sep.join(prefix)
+
+def create_results_log(results, dirname = ".", show_missing = 1,
+ save_counts = 0):
+ import re
+ # turn the counts data ("(filename, lineno) = count") into something
+ # accessible on a per-file basis
+ per_file = {}
+ for filename, lineno in results.counts.keys():
+ lines_hit = per_file[filename] = per_file.get(filename, {})
+ lines_hit[lineno] = results.counts[(filename, lineno)]
+
+ # try and merge existing counts and modules file from dirname
+ try:
+ counts = marshal.load(open(os.path.join(dirname, "counts")))
+ modules = marshal.load(open(os.path.join(dirname, "modules")))
+ results.update(results.__class__(counts, modules))
+ except IOError:
+ pass
+
+ # there are many places where this is insufficient, like a blank
+ # line embedded in a multiline string.
+ blank = re.compile(r'^\s*(#.*)?$')
+
+ # generate file paths for the coverage files we are going to write...
+ fnlist = []
+ tfdir = tempfile.gettempdir()
+ for key in per_file.keys():
+ filename = key
+
+ # skip some "files" we don't care about...
+ if filename == "<string>":
+ continue
+ # are these caused by code compiled using exec or something?
+ if filename.startswith(tfdir):
+ continue
+
+ # XXX this is almost certainly not portable!!!
+ fndir = os.path.dirname(filename)
+ if filename[:1] == os.sep:
+ coverpath = os.path.join(dirname, "."+fndir)
+ else:
+ coverpath = os.path.join(dirname, fndir)
+
+ if filename.endswith(".pyc") or filename.endswith(".pyo"):
+ filename = filename[:-1]
+
+ # Get the original lines from the .py file
+ try:
+ lines = open(filename, 'r').readlines()
+ except IOError, err:
+ sys.stderr.write(
+ "%s: Could not open %s for reading because: %s - skipping\n" % \
+ ("trace", `filename`, err.strerror))
+ continue
+
+ modulename = os.path.split(results.modules[key])[1]
+
+ # build list file name by appending a ".cover" to the module name
+ # and sticking it into the specified directory
+ listfilename = os.path.join(coverpath, modulename + ".cover")
+ #sys.stderr.write("modulename: %(modulename)s\n"
+ # "filename: %(filename)s\n"
+ # "coverpath: %(coverpath)s\n"
+ # "listfilename: %(listfilename)s\n"
+ # "dirname: %(dirname)s\n"
+ # % locals())
+ try:
+ outfile = open(listfilename, 'w')
+ except IOError, err:
+ sys.stderr.write(
+ '%s: Could not open %s for writing because: %s - skipping\n' %
+ ("trace", `listfilename`, err.strerror))
+ continue
+
+ # If desired, get a list of the line numbers which represent
+ # executable content (returned as a dict for better lookup speed)
+ if show_missing:
+ executable_linenos = find_executable_linenos(filename)
+ else:
+ executable_linenos = {}
+
+ lines_hit = per_file[key]
+ for i in range(len(lines)):
+ line = lines[i]
+
+ # do the blank/comment match to try to mark more lines
+ # (help the reader find stuff that hasn't been covered)
+ if lines_hit.has_key(i+1):
+ # count precedes the lines that we captured
+ outfile.write('%5d: ' % lines_hit[i+1])
+ elif blank.match(line):
+ # blank lines and comments are preceded by dots
+ outfile.write(' . ')
+ else:
+ # lines preceded by no marks weren't hit
+ # Highlight them if so indicated, unless the line contains
+ # '#pragma: NO COVER' (it is possible to embed this into
+ # the text as a non-comment; no easy fix)
+ if executable_linenos.has_key(i+1) and \
+ string.find(lines[i],
+ string.join(['#pragma', 'NO COVER'])) == -1:
+ outfile.write('>>>>>> ')
+ else:
+ outfile.write(' '*7)
+ outfile.write(string.expandtabs(lines[i], 8))
+
+ outfile.close()
+
+ if save_counts:
+ # try and store counts and module info into dirname
+ try:
+ marshal.dump(results.counts,
+ open(os.path.join(dirname, "counts"), "w"))
+ marshal.dump(results.modules,
+ open(os.path.join(dirname, "modules"), "w"))
+ except IOError, err:
+ sys.stderr.write("cannot save counts/modules files because %s" %
+ err.strerror)
+
+# There is a lot of code shared between these two classes even though
+# it is straightforward to make a super class to share code. However,
+# for performance reasons (remember, this is called at every step) I
+# wanted to keep everything to a single function call. Also, by
+# staying within a single scope, I don't have to temporarily nullify
+# sys.settrace, which would slow things down even more.
+
+class Coverage:
+ def __init__(self, ignore = Ignore()):
+ self.ignore = ignore
+ self.ignore_names = ignore._ignore # access ignore's cache (speed hack)
+
+ self.counts = {} # keys are (filename, linenumber)
+ self.modules = {} # maps filename -> module name
+
+ def trace(self, frame, why, arg):
+ if why == 'line':
+ # something is fishy about getting the file name
+ filename = frame.f_globals.get("__file__", None)
+ if filename is None:
+ filename = frame.f_code.co_filename
+ modulename = frame.f_globals["__name__"]
+
+ # We do this next block to keep from having to make methods
+ # calls, which also requires resetting the trace
+ ignore_it = self.ignore_names.get(modulename, -1)
+ if ignore_it == -1: # unknown filename
+ sys.settrace(None)
+ ignore_it = self.ignore.names(filename, modulename)
+ sys.settrace(self.trace)
+
+ # record the module name for every file
+ self.modules[filename] = modulename
+
+ if not ignore_it:
+ lineno = frame.f_lineno
+
+ # record the file name and line number of every trace
+ key = (filename, lineno)
+ self.counts[key] = self.counts.get(key, 0) + 1
+
+ return self.trace
+
+ def results(self):
+ return CoverageResults(self.counts, self.modules)
+
+class Trace:
+ def __init__(self, ignore = Ignore()):
+ self.ignore = ignore
+ self.ignore_names = ignore._ignore # access ignore's cache (speed hack)
+
+ self.files = {'<string>': None} # stores lines from the .py file, or None
+
+ def trace(self, frame, why, arg):
+ if why == 'line':
+ filename = frame.f_code.co_filename
+ modulename = frame.f_globals["__name__"]
+
+ # We do this next block to keep from having to make methods
+ # calls, which also requires resetting the trace
+ ignore_it = self.ignore_names.get(modulename, -1)
+ if ignore_it == -1: # unknown filename
+ sys.settrace(None)
+ ignore_it = self.ignore.names(filename, modulename)
+ sys.settrace(self.trace)
+
+ if not ignore_it:
+ lineno = frame.f_lineno
+ files = self.files
+
+ if filename != '<string>' and not files.has_key(filename):
+ files[filename] = map(string.rstrip,
+ open(filename).readlines())
+
+ # If you want to see filenames (the original behaviour), try:
+ # modulename = filename
+ # or, prettier but confusing when several files have the same name
+ # modulename = os.path.basename(filename)
+
+ if files[filename] != None:
+ print '%s(%d): %s' % (os.path.basename(filename), lineno,
+ files[filename][lineno-1])
+ else:
+ print '%s(%d): ??' % (modulename, lineno)
+
+ return self.trace
+
+
+def _err_exit(msg):
+ sys.stderr.write("%s: %s\n" % (sys.argv[0], msg))
+ sys.exit(1)
+
+def main(argv = None):
+ import getopt
+
+ if argv is None:
+ argv = sys.argv
+ try:
+ opts, prog_argv = getopt.getopt(argv[1:], "tcrRf:d:m",
+ ["help", "version", "trace", "count",
+ "report", "no-report",
+ "file=", "logdir=", "missing",
+ "ignore-module=", "ignore-dir="])
+
+ except getopt.error, msg:
+ sys.stderr.write("%s: %s\n" % (sys.argv[0], msg))
+ sys.stderr.write("Try `%s --help' for more information\n" % sys.argv[0])
+ sys.exit(1)
+
+ trace = 0
+ count = 0
+ report = 0
+ no_report = 0
+ counts_file = None
+ logdir = "."
+ missing = 0
+ ignore_modules = []
+ ignore_dirs = []
+
+ for opt, val in opts:
+ if opt == "--help":
+ usage(sys.stdout)
+ sys.exit(0)
+
+ if opt == "--version":
+ sys.stdout.write("trace 2.0\n")
+ sys.exit(0)
+
+ if opt == "-t" or opt == "--trace":
+ trace = 1
+ continue
+
+ if opt == "-c" or opt == "--count":
+ count = 1
+ continue
+
+ if opt == "-r" or opt == "--report":
+ report = 1
+ continue
+
+ if opt == "-R" or opt == "--no-report":
+ no_report = 1
+ continue
+
+ if opt == "-f" or opt == "--file":
+ counts_file = val
+ continue
+
+ if opt == "-d" or opt == "--logdir":
+ logdir = val
+ continue
+
+ if opt == "-m" or opt == "--missing":
+ missing = 1
+ continue
+
+ if opt == "--ignore-module":
+ ignore_modules.append(val)
+ continue
+
+ if opt == "--ignore-dir":
+ for s in string.split(val, os.pathsep):
+ s = os.path.expandvars(s)
+ # should I also call expanduser? (after all, could use $HOME)
+
+ s = string.replace(s, "$prefix",
+ os.path.join(sys.prefix, "lib",
+ "python" + sys.version[:3]))
+ s = string.replace(s, "$exec_prefix",
+ os.path.join(sys.exec_prefix, "lib",
+ "python" + sys.version[:3]))
+ s = os.path.normpath(s)
+ ignore_dirs.append(s)
+ continue
+
+ assert 0, "Should never get here"
+
+ if len(prog_argv) == 0:
+ _err_exit("missing name of file to run")
+
+ if count + trace + report > 1:
+ _err_exit("can only specify one of --trace, --count or --report")
+
+ if count + trace + report == 0:
+ _err_exit("must specify one of --trace, --count or --report")
+
+ if report and counts_file is None:
+ _err_exit("--report requires a --file")
+
+ if report and no_report:
+ _err_exit("cannot specify both --report and --no-report")
+
+ if logdir is not None:
+ # warn if the directory doesn't exist, but keep on going
+ # (is this the correct behaviour?)
+ if not os.path.isdir(logdir):
+ sys.stderr.write(
+ "trace: WARNING, --logdir directory %s is not available\n" %
+ `logdir`)
+
+ sys.argv = prog_argv
+ progname = prog_argv[0]
+ if eval(sys.version[:3])>1.3:
+ sys.path[0] = os.path.split(progname)[0] # ???
+
+ # everything is ready
+ ignore = Ignore(ignore_modules, ignore_dirs)
+ if trace:
+ t = Trace(ignore)
+ try:
+ run(t.trace, 'execfile(' + `progname` + ')')
+ except IOError, err:
+ _err_exit("Cannot run file %s because: %s" % \
+ (`sys.argv[0]`, err.strerror))
+
+ elif count:
+ t = Coverage(ignore)
+ try:
+ run(t.trace, 'execfile(' + `progname` + ')')
+ except IOError, err:
+ _err_exit("Cannot run file %s because: %s" % \
+ (`sys.argv[0]`, err.strerror))
+ except SystemExit:
+ pass
+
+ results = t.results()
+ # Add another lookup from the program's file name to its import name
+ # This give the right results, but I'm not sure why ...
+ results.modules[progname] = os.path.splitext(progname)[0]
+
+ if counts_file:
+ # add in archived data, if available
+ try:
+ old_counts, old_modules = marshal.load(open(counts_file, 'rb'))
+ except IOError:
+ pass
+ else:
+ results.update(CoverageResults(old_counts, old_modules))
+
+ if not no_report:
+ create_results_log(results, logdir, missing)
+
+ if counts_file:
+ try:
+ marshal.dump( (results.counts, results.modules),
+ open(counts_file, 'wb'))
+ except IOError, err:
+ _err_exit("Cannot save counts file %s because: %s" % \
+ (`counts_file`, err.strerror))
+
+ elif report:
+ old_counts, old_modules = marshal.load(open(counts_file, 'rb'))
+ results = CoverageResults(old_counts, old_modules)
+ create_results_log(results, logdir, missing)
+
+ else:
+ assert 0, "Should never get here"
+
+if __name__=='__main__':
+ main()
projects / python / commitdiff