Merge pull request #165 from vhernando/delete_tempfile

Delete temporary file
peter-wangxu · Feb 10, 2022 · d6ed33d · d6ed33d
2 parents d302298 + d638b6d
commit d6ed33d
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/persistqueue/queue.py b/persistqueue/queue.py
@@ -96,7 +96,7 @@ def __init__(self, path, maxsize=0, chunksize=100, tempdir=None,
                 raise ValueError("tempdir has to be located "
                                  "on same path filesystem")
         else:
-            _, tempdir = tempfile.mkstemp()
+            fd, tempdir = tempfile.mkstemp()
             if os.stat(self.path).st_dev != os.stat(tempdir).st_dev:
                 self.tempdir = self.path
                 log.warning("Default tempdir '%(dft_dir)s' is not on the "
@@ -105,6 +105,9 @@ def __init__(self, path, maxsize=0, chunksize=100, tempdir=None,
                                 "dft_dir": tempdir,
                                 "queue_path": self.path,
                                 "new_path": self.tempdir})
+            os.close(fd)
+            os.remove(tempdir)
+
         self.info = self._loadinfo()
         # truncate head case it contains garbage
         hnum, hcnt, hoffset = self.info['head']