rust/src/etc/get-snapshot.py

75 lines
1.8 KiB
Python
Raw Normal View History

#!/usr/bin/env python
2011-11-21 15:11:40 -06:00
import os, tarfile, hashlib, re, shutil, sys
from snapshot import *
def unpack_snapshot(snap):
dl_path = os.path.join(download_dir_base, snap)
2011-05-03 01:37:52 -05:00
print("opening snapshot " + dl_path)
tar = tarfile.open(dl_path)
kernel = get_kernel()
2011-10-31 17:16:01 -05:00
for name in snapshot_files[kernel]:
p = "rust-stage0/" + name
2011-11-21 15:11:40 -06:00
stagep = os.path.join(triple, "stage0")
fp = os.path.join(stagep, name)
2011-05-03 01:37:52 -05:00
print("extracting " + fp)
tar.extract(p, download_unpack_base)
tp = os.path.join(download_unpack_base, p)
shutil.move(tp, fp)
tar.close()
shutil.rmtree(download_unpack_base)
2011-11-28 08:06:32 -06:00
def determine_curr_snapshot(triple):
2011-05-03 09:23:05 -05:00
i = 0
2011-11-28 08:06:32 -06:00
platform = get_platform(triple)
2011-05-03 09:23:05 -05:00
found_file = False
found_snap = False
hsh = None
date = None
rev = None
2011-05-03 18:22:00 -05:00
f = open(snapshotfile)
for line in f.readlines():
i += 1
parsed = parse_line(i, line)
if (not parsed): continue
if found_snap and parsed["type"] == "file":
if parsed["platform"] == platform:
hsh = parsed["hash"]
found_file = True
break;
elif parsed["type"] == "snapshot":
date = parsed["date"]
rev = parsed["rev"]
found_snap = True
2011-05-03 09:23:05 -05:00
if not found_snap:
raise Exception("no snapshot entries in file")
2011-05-03 09:23:05 -05:00
if not found_file:
raise Exception("no snapshot file found for platform %s, rev %s" %
(platform, rev))
2011-05-03 09:23:05 -05:00
return full_snapshot_name(date, rev, get_platform(), hsh)
# Main
2011-11-21 15:11:40 -06:00
triple = sys.argv[1]
2011-05-03 09:23:05 -05:00
snap = determine_curr_snapshot_for_platform()
dl = os.path.join(download_dir_base, snap)
2011-05-03 01:37:52 -05:00
url = download_url_base + "/" + snap
print("determined most recent snapshot: " + snap)
if (not os.path.exists(dl)):
get_url_to_file(url, dl)
if (snap_filename_hash_part(snap) == hash_file(dl)):
print("got download with ok hash")
else:
2011-05-03 01:37:52 -05:00
raise Exception("bad hash on download")
unpack_snapshot(snap)