---

yaml
---
r: 277766
b: refs/heads/try
c: 15de207
h: refs/heads/master

Author: cyplo

[refs]

@@ -1,7 +1,7 @@
 ---
 refs/heads/master: 6dbb0e86aec11050480beb76eade6fb805010ba7
 refs/heads/snap-stage3: 235d77457d80b549dad3ac36d94f235208a1eafb
-refs/heads/try: 18dafe83b5411e76a13f5e813dae1bec71ef07a6
+refs/heads/try: 15de207b9c79c603aa10c30ae54ca07b295aadaf
 refs/tags/release-0.1: 1f5c5126e96c79d22cb7862f75304136e204f105
 refs/tags/release-0.2: c870d2dffb391e14efb05aa27898f1f6333a9596
 refs/tags/release-0.3: b5f0d0f648d9a6153664837026ba1be43d3e2503

branches/try/src/bootstrap/bootstrap.py

@@ -21,17 +21,23 @@
 
 def get(url, path, verbose=False):
     sha_url = url + ".sha256"
-    temp_file = tempfile.NamedTemporaryFile(delete=False)
-    temp_path = temp_file.name
-    sha_file = tempfile.NamedTemporaryFile(suffix=".sha256", delete=True)
-    sha_path = sha_file.name
-    download(sha_path, sha_url, verbose)
-    download(temp_path, url, verbose)
-    verify(sha_path, temp_path, verbose)
-    sha_file.close()
-    print("moving " + temp_path + " to " + path)
-    shutil.move(temp_path, path)
-    temp_file.close()
+    with tempfile.NamedTemporaryFile(delete=False) as temp_file:
+        temp_path = temp_file.name
+    with tempfile.NamedTemporaryFile(suffix=".sha256", delete=False) as sha_file:
+        sha_path = sha_file.name
+
+    try:
+        download(sha_path, sha_url, verbose)
+        download(temp_path, url, verbose)
+        verify(temp_path, sha_path, verbose)
+        print("moving " + temp_path + " to " + path)
+        shutil.move(temp_path, path)
+    finally:
+        print("removing " + sha_path)
+        os.unlink(sha_path)
+        if os.path.isfile(temp_path):
+            print("removing " + temp_path)
+            os.unlink(temp_path)
 
 
 def download(path, url, verbose):
@@ -46,9 +52,9 @@ def download(path, url, verbose):
         run(["curl", "-o", path, url], verbose=verbose)
 
 
-def verify(sha_path, temp_path, verbose):
-    print("verifying " + temp_path)
-    with open(temp_path, "rb") as f:
+def verify(path, sha_path, verbose):
+    print("verifying " + path)
+    with open(path, "rb") as f:
         found = hashlib.sha256(f.read()).hexdigest()
     with open(sha_path, "r") as f:
         expected, _ = f.readline().split()