1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
|
"""Command-line support for coverage.py"""
import getopt, os, sys
USAGE = r"""
Coverage version %(__version__)s
Usage:
coverage -x [-p] MODULE.py [ARG1 ARG2 ...]
Execute module, passing the given command-line arguments, collecting
coverage data. With the -p option, write to a temporary file containing
the machine name and process ID.
coverage -e
Erase collected coverage data.
coverage -c
Combine data from multiple coverage files (as created by -p option above)
and store it into a single file representing the union of the coverage.
coverage -r [-m] [-i] [-o DIR,...] [FILE1 FILE2 ...]
Report on the statement coverage for the given files. With the -m
option, show line numbers of the statements that weren't executed.
coverage -a [-d DIR] [-i] [-o DIR,...] [FILE1 FILE2 ...]
Make annotated copies of the given files, marking statements that
are executed with > and statements that are missed with !. With
the -d option, make the copies in that directory. Without the -d
option, make each copy in the same directory as the original.
-h Print this help.
-i Ignore errors while reporting or annotating.
-o DIR,...
Omit reporting or annotating files when their filename path starts with
a directory listed in the omit list.
e.g. coverage -i -r -o c:\python25,lib\enthought\traits
Coverage data is saved in the file .coverage by default. Set the
COVERAGE_FILE environment variable to save it somewhere else.
""".strip()
class CoverageScript:
def __init__(self):
import coverage
self.covpkg = coverage
self.coverage = coverage.coverage()
def help(self, error=None): #pragma: no cover
if error:
print error
print
print USAGE % self.covpkg.__dict__
sys.exit(1)
def command_line(self, argv, help_fn=None):
# Collect the command-line options.
help_fn = help_fn or self.help
settings = {}
optmap = {
'-a': 'annotate',
'-c': 'combine',
'-d:': 'directory=',
'-e': 'erase',
'-h': 'help',
'-i': 'ignore-errors',
'-m': 'show-missing',
'-p': 'parallel-mode',
'-r': 'report',
'-x': 'execute',
'-o:': 'omit=',
}
short_opts = ''.join(map(lambda o: o[1:], optmap.keys()))
long_opts = optmap.values()
options, args = getopt.getopt(argv, short_opts, long_opts)
for o, a in options:
if optmap.has_key(o):
settings[optmap[o]] = True
elif optmap.has_key(o + ':'):
settings[optmap[o + ':']] = a
elif o[2:] in long_opts:
settings[o[2:]] = True
elif o[2:] + '=' in long_opts:
settings[o[2:]+'='] = a
if settings.get('help'):
help_fn()
# Check for conflicts and problems in the options.
for i in ['erase', 'execute']:
for j in ['annotate', 'report', 'combine']:
if settings.get(i) and settings.get(j):
help_fn("You can't specify the '%s' and '%s' "
"options at the same time." % (i, j))
args_needed = (settings.get('execute')
or settings.get('annotate')
or settings.get('report'))
action = (settings.get('erase')
or settings.get('combine')
or args_needed)
if not action:
help_fn("You must specify at least one of -e, -x, -c, -r, or -a.")
if not args_needed and args:
help_fn("Unexpected arguments: %s" % " ".join(args))
# Do something.
self.coverage.parallel_mode = settings.get('parallel-mode')
self.coverage.get_ready()
if settings.get('erase'):
self.coverage.erase()
if settings.get('execute'):
if not args:
help_fn("Nothing to do.")
# Create the runtime environment the script on the cmdline expects.
sys.argv = args
sys.path[0] = os.path.dirname(sys.argv[0])
import __main__ # TODO: I think this is useless...
self.coverage.start()
execfile(sys.argv[0], __main__.__dict__)
self.coverage.stop()
if settings.get('combine'):
self.coverage.combine()
ignore_errors = settings.get('ignore-errors')
show_missing = settings.get('show-missing')
directory = settings.get('directory=')
omit = settings.get('omit=')
if omit:
omit = omit.split(',')
if settings.get('report'):
self.coverage.report_engine(args, show_missing, ignore_errors, omit_prefixes=omit)
if settings.get('annotate'):
self.coverage.annotate(args, directory, ignore_errors, omit_prefixes=omit)
# Main entrypoint. This is installed as the script entrypoint, so don't
# refactor it away...
def main():
CoverageScript().command_line(sys.argv[1:])
|
