Make entrypoint writes atomic to avoid overwriting symlinks

astral-sh · Jul 17, 2024 · fbf8c1d · fbf8c1d
1 parent f74235b
commit fbf8c1d
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/crates/install-wheel-rs/src/wheel.rs b/crates/install-wheel-rs/src/wheel.rs
@@ -604,7 +604,8 @@ pub(crate) fn write_file_recorded(
         relative_path.display()
     );
 
-    File::create(site_packages.join(relative_path))?.write_all(content.as_ref())?;
+    uv_fs::write_atomic_sync(site_packages.join(relative_path), content.as_ref())?;
+
     let hash = Sha256::new().chain_update(content.as_ref()).finalize();
     let encoded_hash = format!("sha256={}", BASE64URL_NOPAD.encode(&hash));
     record.push(RecordEntry {