blob: 4ed118f0d18b8cd3408bec29ac191e296c05e821 (
plain) (
blame)
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
|
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))
|
