Avoid race conditions when downloading artefacts.

1 files changed, 5 insertions, 3 deletions

diff --git a/download_artefacts.py b/download_artefacts.py
index 10d47b85..cf82b4c0 100755
--- a/download_artefacts.py
+++ b/download_artefacts.py
@@ -65,14 +65,16 @@ def download1(wheel_url, dest_dir):
                 and file_path.stat().st_size == int(w.headers["Content-Length"])):
             logger.info(f"Already have {wheel_name}")
         else:
+            temp_file_path = file_path.with_suffix(".tmp")
             try:
-                with open(file_path, "wb") as f:
+                with open(temp_file_path, "wb") as f:
                     shutil.copyfileobj(w, f)
             except:
-                if file_path.exists():
-                    file_path.unlink()
+                if temp_file_path.exists():
+                    temp_file_path.unlink()
                 raise
             else:
+                temp_file_path.replace(file_path)
                 logger.info(f"Finished downloading {wheel_name}")
     return wheel_name