summaryrefslogtreecommitdiff
path: root/fixcoverage.py
blob: f5d50d3274855a2f5e7f7de87a5cd585431878f4 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
import pickle
import re
import sys

def main(argv):
    source = argv[1]
    dest = argv[2]
    with open('.coverage', 'rb') as f:
        coverage_data = pickle.load(f)
    for filename in list(coverage_data['lines'].keys()):
        new_filename = re.sub(source, dest, filename)
        if new_filename != filename:
            coverage_data['lines'][new_filename] = coverage_data['lines'].pop(filename)
    with open('.coverage', 'wb') as f:
        pickle.dump(coverage_data, f)

if __name__ == '__main__':
    sys.exit(main(sys.argv))