Strip .py suffix from file_stem

astral-sh · Sep 26, 2024 · b15f948 · b15f948
1 parent 284b919
commit b15f948
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/crates/uv/src/lib.rs b/crates/uv/src/lib.rs
@@ -69,13 +69,14 @@ async fn resolve_script_target(
     }
 
     let script_url = url::Url::parse(&maybe_url)?;
-    let file_name = script_url
+    let file_stem = script_url
         .path_segments()
         .and_then(std::iter::Iterator::last)
-        .unwrap_or("script.py");
+        .and_then(|segment| segment.strip_suffix(".py"))
+        .unwrap_or("script");
 
     let temp_file = tempfile::Builder::new()
-        .prefix(file_name)
+        .prefix(file_stem)
         .suffix(".py")
         .tempfile()?;
     let response = reqwest::get(script_url).await?;