[tmpfile] delete the temporary file if an error occurs

2026-04-01 20:12:24 -07:00 · 2026-04-01 20:12:24 -07:00 · ce46cc18dc
commit ce46cc18dc
parent 7c50f95fde
1 changed files with 5 additions and 1 deletions
--- a/masque/file/utils.py
+++ b/masque/file/utils.py
@ -145,7 +145,11 @@ def tmpfile(path: str | pathlib.Path) -> Iterator[IO[bytes]]:
    path = pathlib.Path(path)
    suffixes = ''.join(path.suffixes)
    with tempfile.NamedTemporaryFile(suffix=suffixes, delete=False) as tmp_stream:
-        yield tmp_stream
+        try:
+            yield tmp_stream
+        except Exception:
+            pathlib.Path(tmp_stream.name).unlink(missing_ok=True)
+            raise

    try:
        shutil.move(tmp_stream.name, path)