如何替换 PathBuf 或 Path 的文件扩展名？

Question

My current solution is:

let temp = format!(
    "{}.png",
    path.file_stem().unwrap().to_string_lossy());
path.pop();
path.push(&temp);

Which is quite ugly, requiring at least 6 function calls and creating an new string.

Is there a more idiomatic, shorter, or more efficient way to do this?

Answer 1

PathBuf provides the method set_extension . It will add the extension if one does not exist yet, or replace it with the new one if it does.

let mut path = PathBuf::from("path/to/file");
path.set_extension("png");
assert_eq!(&path.to_string_lossy(), "path/to/file.png");

let mut path = PathBuf::from("path/to/file.jpg");
path.set_extension("png");
assert_eq!(&path.to_string_lossy(), "path/to/file.png");