51 lines
1.7 KiB
Python
51 lines
1.7 KiB
Python
|
#!/usr/bin/env python
|
||
|
|
|
||
|
|
"""Simple script to concatenate coverage reports.
|
||
|
|
"""
|
||
|
|
|
||
|
|
import os
|
||
|
|
import sys
|
||
|
|
import argparse
|
||
|
|
import fnmatch
|
||
|
|
|
||
|
|
|
||
|
|
def main(arguments):
|
||
|
|
|
||
|
|
parser = argparse.ArgumentParser(description=__doc__,
|
||
|
|
formatter_class=argparse.RawDescriptionHelpFormatter)
|
||
|
|
parser.add_argument('dir', help="Input dir to search recursively for .cov files")
|
||
|
|
parser.add_argument('-o', '--outfile', help="Output file",
|
||
|
|
default=sys.stdout, type=argparse.FileType('w'))
|
||
|
|
|
||
|
|
args = parser.parse_args(arguments)
|
||
|
|
|
||
|
|
# Recursively find all matching .cov files.
|
||
|
|
matches = []
|
||
|
|
for root, dirnames, filenames in os.walk(args.dir):
|
||
|
|
for filename in fnmatch.filter(filenames, '*.cov'):
|
||
|
|
matches.append(os.path.join(root, filename))
|
||
|
|
|
||
|
|
# Write to output.
|
||
|
|
lines = {}
|
||
|
|
args.outfile.write('mode: atomic\n')
|
||
|
|
for m in matches:
|
||
|
|
if os.path.abspath(args.outfile.name) != os.path.abspath(m):
|
||
|
|
with open(m) as f:
|
||
|
|
for line in f:
|
||
|
|
if not line.startswith('mode:') and "vendor" not in line:
|
||
|
|
(position, stmt, count) = line.split(" ")
|
||
|
|
stmt = int(stmt)
|
||
|
|
count = int(count)
|
||
|
|
prev_count = 0
|
||
|
|
if position in lines:
|
||
|
|
(_, prev_stmt, prev_count) = lines[position]
|
||
|
|
assert prev_stmt == stmt
|
||
|
|
lines[position] = (position, stmt, prev_count + count)
|
||
|
|
|
||
|
|
for line in sorted(["%s %d %d\n" % lines[key] for key in lines.keys()]):
|
||
|
|
args.outfile.write(line)
|
||
|
|
|
||
|
|
|
||
|
|
if __name__ == '__main__':
|
||
|
|
sys.exit(main(sys.argv[1:]))
|