add ymake export to ydb

1 files changed, 50 insertions, 0 deletions

diff --git a/build/scripts/fetch_from_mds.py b/build/scripts/fetch_from_mds.py
new file mode 100644
index 0000000000..5e4e656394
--- /dev/null
+++ b/build/scripts/fetch_from_mds.py
@@ -0,0 +1,50 @@
+import os
+import sys
+import logging
+import argparse
+
+import fetch_from
+
+MDS_PREFIX = "https://storage.yandex-team.ru/get-devtools/"
+
+
+def parse_args():
+    parser = argparse.ArgumentParser()
+    fetch_from.add_common_arguments(parser)
+
+    parser.add_argument('--key', required=True)
+
+    return parser.parse_args()
+
+
+def fetch(key):
+    parts = key.split("/")
+    if len(parts) != 3:
+        raise ValueError("Invalid MDS key '{}'".format(key))
+
+    _, sha1, file_name = parts
+
+    fetched_file = fetch_from.fetch_url(MDS_PREFIX + key, False, file_name, expected_sha1=sha1)
+
+    return fetched_file, file_name
+
+
+def main(args):
+    fetched_file, resource_file_name = fetch(args.key)
+
+    fetch_from.process(fetched_file, resource_file_name, args)
+
+
+if __name__ == '__main__':
+    args = parse_args()
+    fetch_from.setup_logging(args, os.path.basename(__file__))
+
+    try:
+        main(args)
+    except Exception as e:
+        logging.exception(e)
+        print >>sys.stderr, open(args.abs_log_path).read()
+        sys.stderr.flush()
+
+        import error
+        sys.exit(error.ExitCodes.INFRASTRUCTURE_ERROR if fetch_from.is_temporary(e) else 1)