Skip to content

Commit a7bda52

Browse files
1 parent bd932c3 commit a7bda52

File tree

2 files changed

+10
-6
lines changed

2 files changed

+10
-6
lines changed

fsspec/implementations/cached.py

Lines changed: 8 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -180,8 +180,8 @@ def save_cache(self):
180180
for c in cache.values():
181181
if isinstance(c["blocks"], set):
182182
c["blocks"] = list(c["blocks"])
183-
fn2 = tempfile.mktemp()
184-
with open(fn2, "wb") as f:
183+
fd2, fn2 = tempfile.mkstemp()
184+
with open(fd2, "wb") as f:
185185
pickle.dump(cache, f)
186186
self._mkcache()
187187
move(fn2, fn)
@@ -723,10 +723,13 @@ class LocalTempFile:
723723
"""A temporary local file, which will be uploaded on commit"""
724724

725725
def __init__(self, fs, path, fn=None, mode="wb", autocommit=True, seek=0):
726-
fn = fn or tempfile.mktemp()
726+
if fn:
727+
self.fn = fn
728+
self.fh = open(fn, mode)
729+
else:
730+
fd, self.fn = tempfile.mkstemp()
731+
self.fh = open(fd, mode)
727732
self.mode = mode
728-
self.fn = fn
729-
self.fh = open(fn, mode)
730733
if seek:
731734
self.fh.seek(seek)
732735
self.path = path

fsspec/implementations/tests/test_cached.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -124,7 +124,8 @@ def test_write():
124124
url = "simplecache::file://" + fn
125125
with fsspec.open(url, "wb") as f:
126126
f.write(b"hello")
127-
assert fn not in f.name
127+
if isinstance(f.name, str):
128+
assert fn not in f.name
128129
assert not os.listdir(tmp)
129130

130131
assert open(fn, "rb").read() == b"hello"

0 commit comments

Comments
 (0)