Delete invalid files when they're invalid

Mews · Jun 8, 2024 · 2faee0f · 2faee0f
1 parent 4d51cef
commit 2faee0f
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/src/simplestretch/__init__.py b/src/simplestretch/__init__.py
@@ -2,7 +2,7 @@
 
 import soundfile
 from numpy import ndarray
-
+import pathlib
 
 
 def stretch_audio(
@@ -60,6 +60,9 @@ def stretch_audio(
         try:
             soundfile.write(output, audio, samplerate=stretched_samplerate)
         except soundfile.LibsndfileError as exc:
+            # Delete invalid file
+            pathlib.Path.unlink(output)
+
             exc.add_note("(Try saving it as a .wav file instead)")
             raise exc