build/scripts/with_coverage.py


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

# TODO prettyboy remove after ya-bin release

import os 
import sys 
import subprocess 
import tarfile 
import random 
import shutil 
 
 
def mkdir_p(path): 
    try: 
        os.makedirs(path) 
    except OSError: 
        pass 
 
 
def main(args): 
    coverage_path = os.path.abspath(args[0]) 
    coverage_dir = coverage_path + '.' + str(random.getrandbits(64)) 
 
    mkdir_p(coverage_dir) 
 
    env = os.environ.copy() 
    env['GCOV_PREFIX'] = coverage_dir 
 
    subprocess.check_call(args[1:], env=env) 
 
    arch_path = coverage_dir + '.archive' 
 
    with tarfile.open(arch_path, 'w:') as tar: 
        tar.add(coverage_dir, arcname='.') 
 
    os.rename(arch_path, coverage_path) 
 
    shutil.rmtree(coverage_dir) 
 
 
if __name__ == '__main__': 
    main(sys.argv[1:])