add ymake export to ydb

1 files changed, 45 insertions, 0 deletions

diff --git a/build/scripts/wrapcc.py b/build/scripts/wrapcc.py
new file mode 100644
index 0000000000..88a9e6a4fc
--- /dev/null
+++ b/build/scripts/wrapcc.py
@@ -0,0 +1,45 @@
+from __future__ import print_function
+
+import os
+import sys
+import time
+import subprocess
+
+
+def need_retry(text):
+    return 'Stack dump' in text
+
+
+def retry_inf(cmd):
+    while True:
+        try:
+            yield subprocess.check_output(cmd, stderr=subprocess.STDOUT), None
+        except subprocess.CalledProcessError as e:
+            yield e.output, e
+
+
+def retry(cmd):
+    for n, (out, err) in enumerate(retry_inf(cmd)):
+        if out:
+            sys.stderr.write(out)
+
+        if n > 5:
+            raise Exception('all retries failed')
+        elif need_retry(out):
+            time.sleep(1 + n)
+        elif err:
+            raise err
+        else:
+            return
+
+
+if __name__ == '__main__':
+    cmd = sys.argv[1:]
+
+    if '-c' in cmd:
+        try:
+            retry(cmd)
+        except subprocess.CalledProcessError as e:
+            sys.exit(e.returncode)
+    else:
+        os.execv(cmd[0], cmd)