Docker image fetch macro

commit_hash:0899d0942e1f7ad8dd12e28f6b3f088982660699

1 files changed, 34 insertions, 0 deletions

diff --git a/build/scripts/fetch_from_docker_repo.py b/build/scripts/fetch_from_docker_repo.py
new file mode 100644
index 0000000000..4ed118f0d1
--- /dev/null
+++ b/build/scripts/fetch_from_docker_repo.py
@@ -0,0 +1,34 @@
+import json
+import os
+import argparse
+import shutil
+
+def parse_args():
+    parser = argparse.ArgumentParser()
+    parser.add_argument('--link', required=True)
+    parser.add_argument('--tag', required=True)
+    parser.add_argument('--output-image-path', required=True)
+    parser.add_argument('--output-info-path', required=True)
+    parser.add_argument('--preloaded-path', required=True)
+
+    return parser.parse_args()
+
+
+def main(args):
+    # assume image was downloaded by runner
+    assert os.path.exists(args.preloaded_path)
+
+    output_dir = os.path.dirname(args.output_image_path)
+    if not os.path.exists(output_dir):
+        os.makedirs(output_dir)
+    shutil.copyfile(args.preloaded_path, args.output_image_path)
+    info = {
+        'link': args.link,
+        'tag': args.tag,
+    }
+    with open(args.output_info_path, 'w') as f:
+        f.write(json.dumps(info))
+
+if __name__ == '__main__':
+    args = parse_args()
+    exit(main(args))