add ymake export to ydb

1 files changed, 40 insertions, 0 deletions

diff --git a/build/scripts/with_coverage.py b/build/scripts/with_coverage.py
new file mode 100644
index 0000000000..d62435c3b8
--- /dev/null
+++ b/build/scripts/with_coverage.py
@@ -0,0 +1,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:])