# HG changeset patch
# User Amine Sehili <amine.sehili@gmail.com>
# Date 1570219398 -3600
# Node ID fbcbf6478b30ed77b40b0d760fe96ecf814545f1
# Parent  5515a3fb85d4a05b2a8465f061745688c955cea3
Fix bug in StreamSaverWorker

diff -r 5515a3fb85d4 -r fbcbf6478b30 auditok/workers.py
--- a/auditok/workers.py	Thu Oct 03 20:19:40 2019 +0100
+++ b/auditok/workers.py	Fri Oct 04 21:03:18 2019 +0100
@@ -172,7 +172,6 @@
         cache_size_sec=0.5,
         timeout=0.2,
     ):
-
         self._reader = audio_reader
         sample_size_bytes = self._reader.sw * self._reader.ch
         self._cache_size = cache_size_sec * self._reader.sr * sample_size_bytes
@@ -216,9 +215,12 @@
 
     def __del__(self):
         self._post_process()
+
         if (
-            self._tmp_output_filename != self._output_filename
-        ) and self._exported:
+            (self._tmp_output_filename != self._output_filename)
+            and self._exported
+            and os.path.exists(self._tmp_output_filename)
+        ):
             os.remove(self._tmp_output_filename)
 
     def _process_message(self, data):
@@ -264,14 +266,14 @@
 
     def save_stream(self):
         if self._exported:
-            return
-        if self._export_format == "wav":
+            return self._output_filename
+
+        if self._export_format in ("raw", "wav"):
+            if self._export_format == "raw":
+                self._export_raw()
             self._exported = True
-            return
-        if self._export_format == "raw":
-            self._export_raw()
-            self._exported = True
-            return
+            return self._output_filename
+
         try:
             self._export_with_ffmpeg_or_avconv()
         except AudioEncodingError: