From 958dea1745b9000becaeed728f78daec3b9c13ba Mon Sep 17 00:00:00 2001 From: Richo Healey Date: Tue, 27 Jan 2015 00:10:27 -0800 Subject: [PATCH] make get-snapshot externally usable --- src/etc/get-snapshot.py | 38 +++++++++++++++++++++----------------- 1 file changed, 21 insertions(+), 17 deletions(-) diff --git a/src/etc/get-snapshot.py b/src/etc/get-snapshot.py index 442c50ed7731..432c6b249468 100755 --- a/src/etc/get-snapshot.py +++ b/src/etc/get-snapshot.py @@ -48,23 +48,27 @@ def unpack_snapshot(triple, dl_path): # The first is the O/S triple. # The second is an optional path to the snapshot to use. -triple = sys.argv[1] -if len(sys.argv) == 3: - dl_path = sys.argv[2] -else: - snap = determine_curr_snapshot(triple) - dl = os.path.join(download_dir_base, snap) - 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") +def main(argv): + triple = argv[1] + if len(argv) == 3: + dl_path = argv[2] else: - raise Exception("bad hash on download") + snap = determine_curr_snapshot(triple) + dl = os.path.join(download_dir_base, snap) + url = download_url_base + "/" + snap + print("determined most recent snapshot: " + snap) - dl_path = os.path.join(download_dir_base, snap) + if (not os.path.exists(dl)): + get_url_to_file(url, dl) -unpack_snapshot(triple, dl_path) + if (snap_filename_hash_part(snap) == hash_file(dl)): + print("got download with ok hash") + else: + raise Exception("bad hash on download") + + dl_path = os.path.join(download_dir_base, snap) + + unpack_snapshot(triple, dl_path) + +if __name__ == '__main__': + main(sys.argv)