981 lines
32 KiB
Python
981 lines
32 KiB
Python
|
# Licensed under the Apache License: http://www.apache.org/licenses/LICENSE-2.0
|
||
|
# For details: https://github.com/nedbat/coveragepy/blob/master/NOTICE.txt
|
||
|
|
||
|
"""Command-line support for coverage.py."""
|
||
|
|
||
|
import glob
|
||
|
import optparse # pylint: disable=deprecated-module
|
||
|
import os
|
||
|
import os.path
|
||
|
import shlex
|
||
|
import sys
|
||
|
import textwrap
|
||
|
import traceback
|
||
|
|
||
|
import coverage
|
||
|
from coverage import Coverage
|
||
|
from coverage import env
|
||
|
from coverage.collector import CTracer
|
||
|
from coverage.config import CoverageConfig
|
||
|
from coverage.control import DEFAULT_DATAFILE
|
||
|
from coverage.data import combinable_files, debug_data_file
|
||
|
from coverage.debug import info_header, short_stack, write_formatted_info
|
||
|
from coverage.exceptions import _BaseCoverageException, _ExceptionDuringRun, NoSource
|
||
|
from coverage.execfile import PyRunner
|
||
|
from coverage.results import Numbers, should_fail_under
|
||
|
|
||
|
# When adding to this file, alphabetization is important. Look for
|
||
|
# "alphabetize" comments throughout.
|
||
|
|
||
|
class Opts:
|
||
|
"""A namespace class for individual options we'll build parsers from."""
|
||
|
|
||
|
# Keep these entries alphabetized (roughly) by the option name as it
|
||
|
# appears on the command line.
|
||
|
|
||
|
append = optparse.make_option(
|
||
|
'-a', '--append', action='store_true',
|
||
|
help="Append coverage data to .coverage, otherwise it starts clean each time.",
|
||
|
)
|
||
|
keep = optparse.make_option(
|
||
|
'', '--keep', action='store_true',
|
||
|
help="Keep original coverage files, otherwise they are deleted.",
|
||
|
)
|
||
|
branch = optparse.make_option(
|
||
|
'', '--branch', action='store_true',
|
||
|
help="Measure branch coverage in addition to statement coverage.",
|
||
|
)
|
||
|
concurrency = optparse.make_option(
|
||
|
'', '--concurrency', action='store', metavar="LIBS",
|
||
|
help=(
|
||
|
"Properly measure code using a concurrency library. " +
|
||
|
"Valid values are: {}, or a comma-list of them."
|
||
|
).format(", ".join(sorted(CoverageConfig.CONCURRENCY_CHOICES))),
|
||
|
)
|
||
|
context = optparse.make_option(
|
||
|
'', '--context', action='store', metavar="LABEL",
|
||
|
help="The context label to record for this coverage run.",
|
||
|
)
|
||
|
contexts = optparse.make_option(
|
||
|
'', '--contexts', action='store', metavar="REGEX1,REGEX2,...",
|
||
|
help=(
|
||
|
"Only display data from lines covered in the given contexts. " +
|
||
|
"Accepts Python regexes, which must be quoted."
|
||
|
),
|
||
|
)
|
||
|
combine_datafile = optparse.make_option(
|
||
|
'', '--data-file', action='store', metavar="DATAFILE",
|
||
|
help=(
|
||
|
"Base name of the data files to operate on. " +
|
||
|
"Defaults to '.coverage'. [env: COVERAGE_FILE]"
|
||
|
),
|
||
|
)
|
||
|
input_datafile = optparse.make_option(
|
||
|
'', '--data-file', action='store', metavar="INFILE",
|
||
|
help=(
|
||
|
"Read coverage data for report generation from this file. " +
|
||
|
"Defaults to '.coverage'. [env: COVERAGE_FILE]"
|
||
|
),
|
||
|
)
|
||
|
output_datafile = optparse.make_option(
|
||
|
'', '--data-file', action='store', metavar="OUTFILE",
|
||
|
help=(
|
||
|
"Write the recorded coverage data to this file. " +
|
||
|
"Defaults to '.coverage'. [env: COVERAGE_FILE]"
|
||
|
),
|
||
|
)
|
||
|
debug = optparse.make_option(
|
||
|
'', '--debug', action='store', metavar="OPTS",
|
||
|
help="Debug options, separated by commas. [env: COVERAGE_DEBUG]",
|
||
|
)
|
||
|
directory = optparse.make_option(
|
||
|
'-d', '--directory', action='store', metavar="DIR",
|
||
|
help="Write the output files to DIR.",
|
||
|
)
|
||
|
fail_under = optparse.make_option(
|
||
|
'', '--fail-under', action='store', metavar="MIN", type="float",
|
||
|
help="Exit with a status of 2 if the total coverage is less than MIN.",
|
||
|
)
|
||
|
help = optparse.make_option(
|
||
|
'-h', '--help', action='store_true',
|
||
|
help="Get help on this command.",
|
||
|
)
|
||
|
ignore_errors = optparse.make_option(
|
||
|
'-i', '--ignore-errors', action='store_true',
|
||
|
help="Ignore errors while reading source files.",
|
||
|
)
|
||
|
include = optparse.make_option(
|
||
|
'', '--include', action='store', metavar="PAT1,PAT2,...",
|
||
|
help=(
|
||
|
"Include only files whose paths match one of these patterns. " +
|
||
|
"Accepts shell-style wildcards, which must be quoted."
|
||
|
),
|
||
|
)
|
||
|
pylib = optparse.make_option(
|
||
|
'-L', '--pylib', action='store_true',
|
||
|
help=(
|
||
|
"Measure coverage even inside the Python installed library, " +
|
||
|
"which isn't done by default."
|
||
|
),
|
||
|
)
|
||
|
show_missing = optparse.make_option(
|
||
|
'-m', '--show-missing', action='store_true',
|
||
|
help="Show line numbers of statements in each module that weren't executed.",
|
||
|
)
|
||
|
module = optparse.make_option(
|
||
|
'-m', '--module', action='store_true',
|
||
|
help=(
|
||
|
"<pyfile> is an importable Python module, not a script path, " +
|
||
|
"to be run as 'python -m' would run it."
|
||
|
),
|
||
|
)
|
||
|
omit = optparse.make_option(
|
||
|
'', '--omit', action='store', metavar="PAT1,PAT2,...",
|
||
|
help=(
|
||
|
"Omit files whose paths match one of these patterns. " +
|
||
|
"Accepts shell-style wildcards, which must be quoted."
|
||
|
),
|
||
|
)
|
||
|
output_xml = optparse.make_option(
|
||
|
'-o', '', action='store', dest="outfile", metavar="OUTFILE",
|
||
|
help="Write the XML report to this file. Defaults to 'coverage.xml'",
|
||
|
)
|
||
|
output_json = optparse.make_option(
|
||
|
'-o', '', action='store', dest="outfile", metavar="OUTFILE",
|
||
|
help="Write the JSON report to this file. Defaults to 'coverage.json'",
|
||
|
)
|
||
|
output_lcov = optparse.make_option(
|
||
|
'-o', '', action='store', dest='outfile', metavar="OUTFILE",
|
||
|
help="Write the LCOV report to this file. Defaults to 'coverage.lcov'",
|
||
|
)
|
||
|
json_pretty_print = optparse.make_option(
|
||
|
'', '--pretty-print', action='store_true',
|
||
|
help="Format the JSON for human readers.",
|
||
|
)
|
||
|
parallel_mode = optparse.make_option(
|
||
|
'-p', '--parallel-mode', action='store_true',
|
||
|
help=(
|
||
|
"Append the machine name, process id and random number to the " +
|
||
|
"data file name to simplify collecting data from " +
|
||
|
"many processes."
|
||
|
),
|
||
|
)
|
||
|
precision = optparse.make_option(
|
||
|
'', '--precision', action='store', metavar='N', type=int,
|
||
|
help=(
|
||
|
"Number of digits after the decimal point to display for " +
|
||
|
"reported coverage percentages."
|
||
|
),
|
||
|
)
|
||
|
quiet = optparse.make_option(
|
||
|
'-q', '--quiet', action='store_true',
|
||
|
help="Don't print messages about what is happening.",
|
||
|
)
|
||
|
rcfile = optparse.make_option(
|
||
|
'', '--rcfile', action='store',
|
||
|
help=(
|
||
|
"Specify configuration file. " +
|
||
|
"By default '.coveragerc', 'setup.cfg', 'tox.ini', and " +
|
||
|
"'pyproject.toml' are tried. [env: COVERAGE_RCFILE]"
|
||
|
),
|
||
|
)
|
||
|
show_contexts = optparse.make_option(
|
||
|
'--show-contexts', action='store_true',
|
||
|
help="Show contexts for covered lines.",
|
||
|
)
|
||
|
skip_covered = optparse.make_option(
|
||
|
'--skip-covered', action='store_true',
|
||
|
help="Skip files with 100% coverage.",
|
||
|
)
|
||
|
no_skip_covered = optparse.make_option(
|
||
|
'--no-skip-covered', action='store_false', dest='skip_covered',
|
||
|
help="Disable --skip-covered.",
|
||
|
)
|
||
|
skip_empty = optparse.make_option(
|
||
|
'--skip-empty', action='store_true',
|
||
|
help="Skip files with no code.",
|
||
|
)
|
||
|
sort = optparse.make_option(
|
||
|
'--sort', action='store', metavar='COLUMN',
|
||
|
help=(
|
||
|
"Sort the report by the named column: name, stmts, miss, branch, brpart, or cover. " +
|
||
|
"Default is name."
|
||
|
),
|
||
|
)
|
||
|
source = optparse.make_option(
|
||
|
'', '--source', action='store', metavar="SRC1,SRC2,...",
|
||
|
help="A list of directories or importable names of code to measure.",
|
||
|
)
|
||
|
timid = optparse.make_option(
|
||
|
'', '--timid', action='store_true',
|
||
|
help=(
|
||
|
"Use a simpler but slower trace method. Try this if you get " +
|
||
|
"seemingly impossible results!"
|
||
|
),
|
||
|
)
|
||
|
title = optparse.make_option(
|
||
|
'', '--title', action='store', metavar="TITLE",
|
||
|
help="A text string to use as the title on the HTML.",
|
||
|
)
|
||
|
version = optparse.make_option(
|
||
|
'', '--version', action='store_true',
|
||
|
help="Display version information and exit.",
|
||
|
)
|
||
|
|
||
|
|
||
|
class CoverageOptionParser(optparse.OptionParser):
|
||
|
"""Base OptionParser for coverage.py.
|
||
|
|
||
|
Problems don't exit the program.
|
||
|
Defaults are initialized for all options.
|
||
|
|
||
|
"""
|
||
|
|
||
|
def __init__(self, *args, **kwargs):
|
||
|
super().__init__(add_help_option=False, *args, **kwargs)
|
||
|
self.set_defaults(
|
||
|
# Keep these arguments alphabetized by their names.
|
||
|
action=None,
|
||
|
append=None,
|
||
|
branch=None,
|
||
|
concurrency=None,
|
||
|
context=None,
|
||
|
contexts=None,
|
||
|
data_file=None,
|
||
|
debug=None,
|
||
|
directory=None,
|
||
|
fail_under=None,
|
||
|
help=None,
|
||
|
ignore_errors=None,
|
||
|
include=None,
|
||
|
keep=None,
|
||
|
module=None,
|
||
|
omit=None,
|
||
|
parallel_mode=None,
|
||
|
precision=None,
|
||
|
pylib=None,
|
||
|
quiet=None,
|
||
|
rcfile=True,
|
||
|
show_contexts=None,
|
||
|
show_missing=None,
|
||
|
skip_covered=None,
|
||
|
skip_empty=None,
|
||
|
sort=None,
|
||
|
source=None,
|
||
|
timid=None,
|
||
|
title=None,
|
||
|
version=None,
|
||
|
)
|
||
|
|
||
|
self.disable_interspersed_args()
|
||
|
|
||
|
class OptionParserError(Exception):
|
||
|
"""Used to stop the optparse error handler ending the process."""
|
||
|
pass
|
||
|
|
||
|
def parse_args_ok(self, args=None, options=None):
|
||
|
"""Call optparse.parse_args, but return a triple:
|
||
|
|
||
|
(ok, options, args)
|
||
|
|
||
|
"""
|
||
|
try:
|
||
|
options, args = super().parse_args(args, options)
|
||
|
except self.OptionParserError:
|
||
|
return False, None, None
|
||
|
return True, options, args
|
||
|
|
||
|
def error(self, msg):
|
||
|
"""Override optparse.error so sys.exit doesn't get called."""
|
||
|
show_help(msg)
|
||
|
raise self.OptionParserError
|
||
|
|
||
|
|
||
|
class GlobalOptionParser(CoverageOptionParser):
|
||
|
"""Command-line parser for coverage.py global option arguments."""
|
||
|
|
||
|
def __init__(self):
|
||
|
super().__init__()
|
||
|
|
||
|
self.add_options([
|
||
|
Opts.help,
|
||
|
Opts.version,
|
||
|
])
|
||
|
|
||
|
|
||
|
class CmdOptionParser(CoverageOptionParser):
|
||
|
"""Parse one of the new-style commands for coverage.py."""
|
||
|
|
||
|
def __init__(self, action, options, defaults=None, usage=None, description=None):
|
||
|
"""Create an OptionParser for a coverage.py command.
|
||
|
|
||
|
`action` is the slug to put into `options.action`.
|
||
|
`options` is a list of Option's for the command.
|
||
|
`defaults` is a dict of default value for options.
|
||
|
`usage` is the usage string to display in help.
|
||
|
`description` is the description of the command, for the help text.
|
||
|
|
||
|
"""
|
||
|
if usage:
|
||
|
usage = "%prog " + usage
|
||
|
super().__init__(
|
||
|
usage=usage,
|
||
|
description=description,
|
||
|
)
|
||
|
self.set_defaults(action=action, **(defaults or {}))
|
||
|
self.add_options(options)
|
||
|
self.cmd = action
|
||
|
|
||
|
def __eq__(self, other):
|
||
|
# A convenience equality, so that I can put strings in unit test
|
||
|
# results, and they will compare equal to objects.
|
||
|
return (other == f"<CmdOptionParser:{self.cmd}>")
|
||
|
|
||
|
__hash__ = None # This object doesn't need to be hashed.
|
||
|
|
||
|
def get_prog_name(self):
|
||
|
"""Override of an undocumented function in optparse.OptionParser."""
|
||
|
program_name = super().get_prog_name()
|
||
|
|
||
|
# Include the sub-command for this parser as part of the command.
|
||
|
return f"{program_name} {self.cmd}"
|
||
|
|
||
|
# In lists of Opts, keep them alphabetized by the option names as they appear
|
||
|
# on the command line, since these lists determine the order of the options in
|
||
|
# the help output.
|
||
|
#
|
||
|
# In COMMANDS, keep the keys (command names) alphabetized.
|
||
|
|
||
|
GLOBAL_ARGS = [
|
||
|
Opts.debug,
|
||
|
Opts.help,
|
||
|
Opts.rcfile,
|
||
|
]
|
||
|
|
||
|
COMMANDS = {
|
||
|
'annotate': CmdOptionParser(
|
||
|
"annotate",
|
||
|
[
|
||
|
Opts.directory,
|
||
|
Opts.input_datafile,
|
||
|
Opts.ignore_errors,
|
||
|
Opts.include,
|
||
|
Opts.omit,
|
||
|
] + GLOBAL_ARGS,
|
||
|
usage="[options] [modules]",
|
||
|
description=(
|
||
|
"Make annotated copies of the given files, marking statements that are executed " +
|
||
|
"with > and statements that are missed with !."
|
||
|
),
|
||
|
),
|
||
|
|
||
|
'combine': CmdOptionParser(
|
||
|
"combine",
|
||
|
[
|
||
|
Opts.append,
|
||
|
Opts.combine_datafile,
|
||
|
Opts.keep,
|
||
|
Opts.quiet,
|
||
|
] + GLOBAL_ARGS,
|
||
|
usage="[options] <path1> <path2> ... <pathN>",
|
||
|
description=(
|
||
|
"Combine data from multiple coverage files collected " +
|
||
|
"with 'run -p'. The combined results are written to a single " +
|
||
|
"file representing the union of the data. The positional " +
|
||
|
"arguments are data files or directories containing data files. " +
|
||
|
"If no paths are provided, data files in the default data file's " +
|
||
|
"directory are combined."
|
||
|
),
|
||
|
),
|
||
|
|
||
|
'debug': CmdOptionParser(
|
||
|
"debug", GLOBAL_ARGS,
|
||
|
usage="<topic>",
|
||
|
description=(
|
||
|
"Display information about the internals of coverage.py, " +
|
||
|
"for diagnosing problems. " +
|
||
|
"Topics are: " +
|
||
|
"'data' to show a summary of the collected data; " +
|
||
|
"'sys' to show installation information; " +
|
||
|
"'config' to show the configuration; " +
|
||
|
"'premain' to show what is calling coverage; " +
|
||
|
"'pybehave' to show internal flags describing Python behavior."
|
||
|
),
|
||
|
),
|
||
|
|
||
|
'erase': CmdOptionParser(
|
||
|
"erase",
|
||
|
[
|
||
|
Opts.combine_datafile
|
||
|
] + GLOBAL_ARGS,
|
||
|
description="Erase previously collected coverage data.",
|
||
|
),
|
||
|
|
||
|
'help': CmdOptionParser(
|
||
|
"help", GLOBAL_ARGS,
|
||
|
usage="[command]",
|
||
|
description="Describe how to use coverage.py",
|
||
|
),
|
||
|
|
||
|
'html': CmdOptionParser(
|
||
|
"html",
|
||
|
[
|
||
|
Opts.contexts,
|
||
|
Opts.directory,
|
||
|
Opts.input_datafile,
|
||
|
Opts.fail_under,
|
||
|
Opts.ignore_errors,
|
||
|
Opts.include,
|
||
|
Opts.omit,
|
||
|
Opts.precision,
|
||
|
Opts.quiet,
|
||
|
Opts.show_contexts,
|
||
|
Opts.skip_covered,
|
||
|
Opts.no_skip_covered,
|
||
|
Opts.skip_empty,
|
||
|
Opts.title,
|
||
|
] + GLOBAL_ARGS,
|
||
|
usage="[options] [modules]",
|
||
|
description=(
|
||
|
"Create an HTML report of the coverage of the files. " +
|
||
|
"Each file gets its own page, with the source decorated to show " +
|
||
|
"executed, excluded, and missed lines."
|
||
|
),
|
||
|
),
|
||
|
|
||
|
'json': CmdOptionParser(
|
||
|
"json",
|
||
|
[
|
||
|
Opts.contexts,
|
||
|
Opts.input_datafile,
|
||
|
Opts.fail_under,
|
||
|
Opts.ignore_errors,
|
||
|
Opts.include,
|
||
|
Opts.omit,
|
||
|
Opts.output_json,
|
||
|
Opts.json_pretty_print,
|
||
|
Opts.quiet,
|
||
|
Opts.show_contexts,
|
||
|
] + GLOBAL_ARGS,
|
||
|
usage="[options] [modules]",
|
||
|
description="Generate a JSON report of coverage results.",
|
||
|
),
|
||
|
|
||
|
'lcov': CmdOptionParser(
|
||
|
"lcov",
|
||
|
[
|
||
|
Opts.input_datafile,
|
||
|
Opts.fail_under,
|
||
|
Opts.ignore_errors,
|
||
|
Opts.include,
|
||
|
Opts.output_lcov,
|
||
|
Opts.omit,
|
||
|
Opts.quiet,
|
||
|
] + GLOBAL_ARGS,
|
||
|
usage="[options] [modules]",
|
||
|
description="Generate an LCOV report of coverage results.",
|
||
|
),
|
||
|
|
||
|
'report': CmdOptionParser(
|
||
|
"report",
|
||
|
[
|
||
|
Opts.contexts,
|
||
|
Opts.input_datafile,
|
||
|
Opts.fail_under,
|
||
|
Opts.ignore_errors,
|
||
|
Opts.include,
|
||
|
Opts.omit,
|
||
|
Opts.precision,
|
||
|
Opts.sort,
|
||
|
Opts.show_missing,
|
||
|
Opts.skip_covered,
|
||
|
Opts.no_skip_covered,
|
||
|
Opts.skip_empty,
|
||
|
] + GLOBAL_ARGS,
|
||
|
usage="[options] [modules]",
|
||
|
description="Report coverage statistics on modules.",
|
||
|
),
|
||
|
|
||
|
'run': CmdOptionParser(
|
||
|
"run",
|
||
|
[
|
||
|
Opts.append,
|
||
|
Opts.branch,
|
||
|
Opts.concurrency,
|
||
|
Opts.context,
|
||
|
Opts.output_datafile,
|
||
|
Opts.include,
|
||
|
Opts.module,
|
||
|
Opts.omit,
|
||
|
Opts.pylib,
|
||
|
Opts.parallel_mode,
|
||
|
Opts.source,
|
||
|
Opts.timid,
|
||
|
] + GLOBAL_ARGS,
|
||
|
usage="[options] <pyfile> [program options]",
|
||
|
description="Run a Python program, measuring code execution.",
|
||
|
),
|
||
|
|
||
|
'xml': CmdOptionParser(
|
||
|
"xml",
|
||
|
[
|
||
|
Opts.input_datafile,
|
||
|
Opts.fail_under,
|
||
|
Opts.ignore_errors,
|
||
|
Opts.include,
|
||
|
Opts.omit,
|
||
|
Opts.output_xml,
|
||
|
Opts.quiet,
|
||
|
Opts.skip_empty,
|
||
|
] + GLOBAL_ARGS,
|
||
|
usage="[options] [modules]",
|
||
|
description="Generate an XML report of coverage results.",
|
||
|
),
|
||
|
}
|
||
|
|
||
|
|
||
|
def show_help(error=None, topic=None, parser=None):
|
||
|
"""Display an error message, or the named topic."""
|
||
|
assert error or topic or parser
|
||
|
|
||
|
program_path = sys.argv[0]
|
||
|
if program_path.endswith(os.path.sep + '__main__.py'):
|
||
|
# The path is the main module of a package; get that path instead.
|
||
|
program_path = os.path.dirname(program_path)
|
||
|
program_name = os.path.basename(program_path)
|
||
|
if env.WINDOWS:
|
||
|
# entry_points={'console_scripts':...} on Windows makes files
|
||
|
# called coverage.exe, coverage3.exe, and coverage-3.5.exe. These
|
||
|
# invoke coverage-script.py, coverage3-script.py, and
|
||
|
# coverage-3.5-script.py. argv[0] is the .py file, but we want to
|
||
|
# get back to the original form.
|
||
|
auto_suffix = "-script.py"
|
||
|
if program_name.endswith(auto_suffix):
|
||
|
program_name = program_name[:-len(auto_suffix)]
|
||
|
|
||
|
help_params = dict(coverage.__dict__)
|
||
|
help_params['program_name'] = program_name
|
||
|
if CTracer is not None:
|
||
|
help_params['extension_modifier'] = 'with C extension'
|
||
|
else:
|
||
|
help_params['extension_modifier'] = 'without C extension'
|
||
|
|
||
|
if error:
|
||
|
print(error, file=sys.stderr)
|
||
|
print(f"Use '{program_name} help' for help.", file=sys.stderr)
|
||
|
elif parser:
|
||
|
print(parser.format_help().strip())
|
||
|
print()
|
||
|
else:
|
||
|
help_msg = textwrap.dedent(HELP_TOPICS.get(topic, '')).strip()
|
||
|
if help_msg:
|
||
|
print(help_msg.format(**help_params))
|
||
|
else:
|
||
|
print(f"Don't know topic {topic!r}")
|
||
|
print("Full documentation is at {__url__}".format(**help_params))
|
||
|
|
||
|
|
||
|
OK, ERR, FAIL_UNDER = 0, 1, 2
|
||
|
|
||
|
|
||
|
class CoverageScript:
|
||
|
"""The command-line interface to coverage.py."""
|
||
|
|
||
|
def __init__(self):
|
||
|
self.global_option = False
|
||
|
self.coverage = None
|
||
|
|
||
|
def command_line(self, argv):
|
||
|
"""The bulk of the command line interface to coverage.py.
|
||
|
|
||
|
`argv` is the argument list to process.
|
||
|
|
||
|
Returns 0 if all is well, 1 if something went wrong.
|
||
|
|
||
|
"""
|
||
|
# Collect the command-line options.
|
||
|
if not argv:
|
||
|
show_help(topic='minimum_help')
|
||
|
return OK
|
||
|
|
||
|
# The command syntax we parse depends on the first argument. Global
|
||
|
# switch syntax always starts with an option.
|
||
|
self.global_option = argv[0].startswith('-')
|
||
|
if self.global_option:
|
||
|
parser = GlobalOptionParser()
|
||
|
else:
|
||
|
parser = COMMANDS.get(argv[0])
|
||
|
if not parser:
|
||
|
show_help(f"Unknown command: {argv[0]!r}")
|
||
|
return ERR
|
||
|
argv = argv[1:]
|
||
|
|
||
|
ok, options, args = parser.parse_args_ok(argv)
|
||
|
if not ok:
|
||
|
return ERR
|
||
|
|
||
|
# Handle help and version.
|
||
|
if self.do_help(options, args, parser):
|
||
|
return OK
|
||
|
|
||
|
# Listify the list options.
|
||
|
source = unshell_list(options.source)
|
||
|
omit = unshell_list(options.omit)
|
||
|
include = unshell_list(options.include)
|
||
|
debug = unshell_list(options.debug)
|
||
|
contexts = unshell_list(options.contexts)
|
||
|
|
||
|
if options.concurrency is not None:
|
||
|
concurrency = options.concurrency.split(",")
|
||
|
else:
|
||
|
concurrency = None
|
||
|
|
||
|
# Do something.
|
||
|
self.coverage = Coverage(
|
||
|
data_file=options.data_file or DEFAULT_DATAFILE,
|
||
|
data_suffix=options.parallel_mode,
|
||
|
cover_pylib=options.pylib,
|
||
|
timid=options.timid,
|
||
|
branch=options.branch,
|
||
|
config_file=options.rcfile,
|
||
|
source=source,
|
||
|
omit=omit,
|
||
|
include=include,
|
||
|
debug=debug,
|
||
|
concurrency=concurrency,
|
||
|
check_preimported=True,
|
||
|
context=options.context,
|
||
|
messages=not options.quiet,
|
||
|
)
|
||
|
|
||
|
if options.action == "debug":
|
||
|
return self.do_debug(args)
|
||
|
|
||
|
elif options.action == "erase":
|
||
|
self.coverage.erase()
|
||
|
return OK
|
||
|
|
||
|
elif options.action == "run":
|
||
|
return self.do_run(options, args)
|
||
|
|
||
|
elif options.action == "combine":
|
||
|
if options.append:
|
||
|
self.coverage.load()
|
||
|
data_paths = args or None
|
||
|
self.coverage.combine(data_paths, strict=True, keep=bool(options.keep))
|
||
|
self.coverage.save()
|
||
|
return OK
|
||
|
|
||
|
# Remaining actions are reporting, with some common options.
|
||
|
report_args = dict(
|
||
|
morfs=unglob_args(args),
|
||
|
ignore_errors=options.ignore_errors,
|
||
|
omit=omit,
|
||
|
include=include,
|
||
|
contexts=contexts,
|
||
|
)
|
||
|
|
||
|
# We need to be able to import from the current directory, because
|
||
|
# plugins may try to, for example, to read Django settings.
|
||
|
sys.path.insert(0, '')
|
||
|
|
||
|
self.coverage.load()
|
||
|
|
||
|
total = None
|
||
|
if options.action == "report":
|
||
|
total = self.coverage.report(
|
||
|
precision=options.precision,
|
||
|
show_missing=options.show_missing,
|
||
|
skip_covered=options.skip_covered,
|
||
|
skip_empty=options.skip_empty,
|
||
|
sort=options.sort,
|
||
|
**report_args
|
||
|
)
|
||
|
elif options.action == "annotate":
|
||
|
self.coverage.annotate(directory=options.directory, **report_args)
|
||
|
elif options.action == "html":
|
||
|
total = self.coverage.html_report(
|
||
|
directory=options.directory,
|
||
|
precision=options.precision,
|
||
|
skip_covered=options.skip_covered,
|
||
|
skip_empty=options.skip_empty,
|
||
|
show_contexts=options.show_contexts,
|
||
|
title=options.title,
|
||
|
**report_args
|
||
|
)
|
||
|
elif options.action == "xml":
|
||
|
total = self.coverage.xml_report(
|
||
|
outfile=options.outfile,
|
||
|
skip_empty=options.skip_empty,
|
||
|
**report_args
|
||
|
)
|
||
|
elif options.action == "json":
|
||
|
total = self.coverage.json_report(
|
||
|
outfile=options.outfile,
|
||
|
pretty_print=options.pretty_print,
|
||
|
show_contexts=options.show_contexts,
|
||
|
**report_args
|
||
|
)
|
||
|
elif options.action == "lcov":
|
||
|
total = self.coverage.lcov_report(
|
||
|
outfile=options.outfile,
|
||
|
**report_args
|
||
|
)
|
||
|
else:
|
||
|
# There are no other possible actions.
|
||
|
raise AssertionError
|
||
|
|
||
|
if total is not None:
|
||
|
# Apply the command line fail-under options, and then use the config
|
||
|
# value, so we can get fail_under from the config file.
|
||
|
if options.fail_under is not None:
|
||
|
self.coverage.set_option("report:fail_under", options.fail_under)
|
||
|
if options.precision is not None:
|
||
|
self.coverage.set_option("report:precision", options.precision)
|
||
|
|
||
|
fail_under = self.coverage.get_option("report:fail_under")
|
||
|
precision = self.coverage.get_option("report:precision")
|
||
|
if should_fail_under(total, fail_under, precision):
|
||
|
msg = "total of {total} is less than fail-under={fail_under:.{p}f}".format(
|
||
|
total=Numbers(precision=precision).display_covered(total),
|
||
|
fail_under=fail_under,
|
||
|
p=precision,
|
||
|
)
|
||
|
print("Coverage failure:", msg)
|
||
|
return FAIL_UNDER
|
||
|
|
||
|
return OK
|
||
|
|
||
|
def do_help(self, options, args, parser):
|
||
|
"""Deal with help requests.
|
||
|
|
||
|
Return True if it handled the request, False if not.
|
||
|
|
||
|
"""
|
||
|
# Handle help.
|
||
|
if options.help:
|
||
|
if self.global_option:
|
||
|
show_help(topic='help')
|
||
|
else:
|
||
|
show_help(parser=parser)
|
||
|
return True
|
||
|
|
||
|
if options.action == "help":
|
||
|
if args:
|
||
|
for a in args:
|
||
|
parser = COMMANDS.get(a)
|
||
|
if parser:
|
||
|
show_help(parser=parser)
|
||
|
else:
|
||
|
show_help(topic=a)
|
||
|
else:
|
||
|
show_help(topic='help')
|
||
|
return True
|
||
|
|
||
|
# Handle version.
|
||
|
if options.version:
|
||
|
show_help(topic='version')
|
||
|
return True
|
||
|
|
||
|
return False
|
||
|
|
||
|
def do_run(self, options, args):
|
||
|
"""Implementation of 'coverage run'."""
|
||
|
|
||
|
if not args:
|
||
|
if options.module:
|
||
|
# Specified -m with nothing else.
|
||
|
show_help("No module specified for -m")
|
||
|
return ERR
|
||
|
command_line = self.coverage.get_option("run:command_line")
|
||
|
if command_line is not None:
|
||
|
args = shlex.split(command_line)
|
||
|
if args and args[0] in {"-m", "--module"}:
|
||
|
options.module = True
|
||
|
args = args[1:]
|
||
|
if not args:
|
||
|
show_help("Nothing to do.")
|
||
|
return ERR
|
||
|
|
||
|
if options.append and self.coverage.get_option("run:parallel"):
|
||
|
show_help("Can't append to data files in parallel mode.")
|
||
|
return ERR
|
||
|
|
||
|
if options.concurrency == "multiprocessing":
|
||
|
# Can't set other run-affecting command line options with
|
||
|
# multiprocessing.
|
||
|
for opt_name in ['branch', 'include', 'omit', 'pylib', 'source', 'timid']:
|
||
|
# As it happens, all of these options have no default, meaning
|
||
|
# they will be None if they have not been specified.
|
||
|
if getattr(options, opt_name) is not None:
|
||
|
show_help(
|
||
|
"Options affecting multiprocessing must only be specified " +
|
||
|
"in a configuration file.\n" +
|
||
|
f"Remove --{opt_name} from the command line."
|
||
|
)
|
||
|
return ERR
|
||
|
|
||
|
os.environ["COVERAGE_RUN"] = "true"
|
||
|
|
||
|
runner = PyRunner(args, as_module=bool(options.module))
|
||
|
runner.prepare()
|
||
|
|
||
|
if options.append:
|
||
|
self.coverage.load()
|
||
|
|
||
|
# Run the script.
|
||
|
self.coverage.start()
|
||
|
code_ran = True
|
||
|
try:
|
||
|
runner.run()
|
||
|
except NoSource:
|
||
|
code_ran = False
|
||
|
raise
|
||
|
finally:
|
||
|
self.coverage.stop()
|
||
|
if code_ran:
|
||
|
self.coverage.save()
|
||
|
|
||
|
return OK
|
||
|
|
||
|
def do_debug(self, args):
|
||
|
"""Implementation of 'coverage debug'."""
|
||
|
|
||
|
if not args:
|
||
|
show_help("What information would you like: config, data, sys, premain, pybehave?")
|
||
|
return ERR
|
||
|
if args[1:]:
|
||
|
show_help("Only one topic at a time, please")
|
||
|
return ERR
|
||
|
|
||
|
if args[0] == "sys":
|
||
|
write_formatted_info(print, "sys", self.coverage.sys_info())
|
||
|
elif args[0] == "data":
|
||
|
print(info_header("data"))
|
||
|
data_file = self.coverage.config.data_file
|
||
|
debug_data_file(data_file)
|
||
|
for filename in combinable_files(data_file):
|
||
|
print("-----")
|
||
|
debug_data_file(filename)
|
||
|
elif args[0] == "config":
|
||
|
write_formatted_info(print, "config", self.coverage.config.debug_info())
|
||
|
elif args[0] == "premain":
|
||
|
print(info_header("premain"))
|
||
|
print(short_stack())
|
||
|
elif args[0] == "pybehave":
|
||
|
write_formatted_info(print, "pybehave", env.debug_info())
|
||
|
else:
|
||
|
show_help(f"Don't know what you mean by {args[0]!r}")
|
||
|
return ERR
|
||
|
|
||
|
return OK
|
||
|
|
||
|
|
||
|
def unshell_list(s):
|
||
|
"""Turn a command-line argument into a list."""
|
||
|
if not s:
|
||
|
return None
|
||
|
if env.WINDOWS:
|
||
|
# When running coverage.py as coverage.exe, some of the behavior
|
||
|
# of the shell is emulated: wildcards are expanded into a list of
|
||
|
# file names. So you have to single-quote patterns on the command
|
||
|
# line, but (not) helpfully, the single quotes are included in the
|
||
|
# argument, so we have to strip them off here.
|
||
|
s = s.strip("'")
|
||
|
return s.split(',')
|
||
|
|
||
|
|
||
|
def unglob_args(args):
|
||
|
"""Interpret shell wildcards for platforms that need it."""
|
||
|
if env.WINDOWS:
|
||
|
globbed = []
|
||
|
for arg in args:
|
||
|
if '?' in arg or '*' in arg:
|
||
|
globbed.extend(glob.glob(arg))
|
||
|
else:
|
||
|
globbed.append(arg)
|
||
|
args = globbed
|
||
|
return args
|
||
|
|
||
|
|
||
|
HELP_TOPICS = {
|
||
|
'help': """\
|
||
|
Coverage.py, version {__version__} {extension_modifier}
|
||
|
Measure, collect, and report on code coverage in Python programs.
|
||
|
|
||
|
usage: {program_name} <command> [options] [args]
|
||
|
|
||
|
Commands:
|
||
|
annotate Annotate source files with execution information.
|
||
|
combine Combine a number of data files.
|
||
|
debug Display information about the internals of coverage.py
|
||
|
erase Erase previously collected coverage data.
|
||
|
help Get help on using coverage.py.
|
||
|
html Create an HTML report.
|
||
|
json Create a JSON report of coverage results.
|
||
|
lcov Create an LCOV report of coverage results.
|
||
|
report Report coverage stats on modules.
|
||
|
run Run a Python program and measure code execution.
|
||
|
xml Create an XML report of coverage results.
|
||
|
|
||
|
Use "{program_name} help <command>" for detailed help on any command.
|
||
|
""",
|
||
|
|
||
|
'minimum_help': """\
|
||
|
Code coverage for Python, version {__version__} {extension_modifier}. Use '{program_name} help' for help.
|
||
|
""",
|
||
|
|
||
|
'version': """\
|
||
|
Coverage.py, version {__version__} {extension_modifier}
|
||
|
""",
|
||
|
}
|
||
|
|
||
|
|
||
|
def main(argv=None):
|
||
|
"""The main entry point to coverage.py.
|
||
|
|
||
|
This is installed as the script entry point.
|
||
|
|
||
|
"""
|
||
|
if argv is None:
|
||
|
argv = sys.argv[1:]
|
||
|
try:
|
||
|
status = CoverageScript().command_line(argv)
|
||
|
except _ExceptionDuringRun as err:
|
||
|
# An exception was caught while running the product code. The
|
||
|
# sys.exc_info() return tuple is packed into an _ExceptionDuringRun
|
||
|
# exception.
|
||
|
traceback.print_exception(*err.args) # pylint: disable=no-value-for-parameter
|
||
|
status = ERR
|
||
|
except _BaseCoverageException as err:
|
||
|
# A controlled error inside coverage.py: print the message to the user.
|
||
|
msg = err.args[0]
|
||
|
print(msg)
|
||
|
status = ERR
|
||
|
except SystemExit as err:
|
||
|
# The user called `sys.exit()`. Exit with their argument, if any.
|
||
|
if err.args:
|
||
|
status = err.args[0]
|
||
|
else:
|
||
|
status = None
|
||
|
return status
|
||
|
|
||
|
# Profiling using ox_profile. Install it from GitHub:
|
||
|
# pip install git+https://github.com/emin63/ox_profile.git
|
||
|
#
|
||
|
# $set_env.py: COVERAGE_PROFILE - Set to use ox_profile.
|
||
|
_profile = os.environ.get("COVERAGE_PROFILE", "")
|
||
|
if _profile: # pragma: debugging
|
||
|
from ox_profile.core.launchers import SimpleLauncher # pylint: disable=import-error
|
||
|
original_main = main
|
||
|
|
||
|
def main(argv=None): # pylint: disable=function-redefined
|
||
|
"""A wrapper around main that profiles."""
|
||
|
profiler = SimpleLauncher.launch()
|
||
|
try:
|
||
|
return original_main(argv)
|
||
|
finally:
|
||
|
data, _ = profiler.query(re_filter='coverage', max_records=100)
|
||
|
print(profiler.show(query=data, limit=100, sep='', col=''))
|
||
|
profiler.cancel()
|