sift_py.data_import.tempfile

 1import gzip
 2import os
 3import tempfile
 4from pathlib import Path
 5
 6
 7class NamedTemporaryFile:
 8    """
 9    Created a named temporary file.
10
11    Works on both Windows and Unix systems.
12
13    See https://stackoverflow.com/questions/23212435/permission-denied-to-write-to-my-temporary-file
14    for more information on Windows compatibility.
15    """
16
17    def __init__(self, mode, suffix=""):
18        self.temp_dir = tempfile.mkdtemp()
19        self.name = Path(self.temp_dir) / f"tempfile{suffix}"
20        if suffix.endswith(".gz"):
21            self.file = gzip.open(self.name, mode, newline="")
22        else:
23            self.file = open(self.name, mode, newline="")
24
25    def close(self):
26        return self.file.close()
27
28    def __enter__(self):
29        return self.file
30
31    def __exit__(self, exc_type, exc_value, traceback):
32        if not self.file.closed:
33            self.file.close()
34
35        try:
36            os.remove(self.name)
37            os.rmdir(self.temp_dir)
38        except FileNotFoundError:
39            pass