FIX: Replaced mktemp usage for PY2

maroonse · Nov 5, 2023 · ddb2858 · ddb2858
1 parent 17037e5
commit ddb2858
Showing 1 changed file with 2 additions and 5 deletions.
diff --git a/billiard/heap.py b/billiard/heap.py
@@ -90,15 +90,12 @@ def __init__(self, size, fd=-1):
                         f.write(b'\0' * (size % bs))
                         assert f.tell() == size
                 else:
-                    name = tempfile.mktemp(
+                    self.fd, name = tempfile.mkstemp(
                         prefix='pym-%d-' % (os.getpid(),),
                         dir=util.get_temp_dir(),
                     )
-                    self.fd = os.open(
-                        name, os.O_RDWR | os.O_CREAT | os.O_EXCL, 0o600,
-                    )
-                    util.Finalize(self, os.close, (self.fd,))
                     os.unlink(name)
+                    util.Finalize(self, os.close, (self.fd,))
                     os.ftruncate(self.fd, size)
             self.buffer = mmap.mmap(self.fd, self.size)