FilesExpand file tree

main

/

tempfiles.py

Copy path

Blame

More file actions

Blame

More file actions

Latest commit

 

History

History

History

62 lines (48 loc) · 2.08 KB

main

/

tempfiles.py

Copy path

Top

File metadata and controls

• Code
• Blame

62 lines (48 loc) · 2.08 KB

Raw

Copy raw file

Download raw file

Open symbols panel

Edit and raw actions

1

2

3

4

5

6

7

8

9

10

11

12

13

14

15

16

17

18

19

20

21

22

23

24

25

26

27

28

29

30

31

32

33

34

35

36

37

38

39

40

41

42

43

44

45

46

47

48

49

50

51

52

53

54

55

56

57

58

59

60

61

62

# Copyright 2013 The Emscripten Authors. All rights reserved.

# Emscripten is available under two separate licenses, the MIT license and the

# University of Illinois/NCSA Open Source License. Both these licenses can be

# found in the LICENSE file.

import atexit

import sys

import tempfile

from . import utils

class TempFiles:

def __init__(self, tmpdir, save_debug_files):

self.tmpdir = tmpdir

self.save_debug_files = save_debug_files

self.to_clean = []

atexit.register(self.clean)

def note(self, filename):

self.to_clean.append(filename)

def get(self, suffix, prefix=None):

"""Return a named temp file with the given prefix."""

named_file = tempfile.NamedTemporaryFile(dir=self.tmpdir, suffix=suffix, prefix=prefix, delete=False)

self.note(named_file.name)

return named_file

def get_file(self, suffix):

"""Return an object representing a RAII-like access to a temp file.

The the result is a context manager object that can be used in a 'with' statement:

'with TempFiles.get_file(..) as filename:'.

The file will be deleted immediately once the 'with' block is exited.

"""

class TempFileObject:

def __enter__(self_): # noqa: DC02

self_.file = tempfile.NamedTemporaryFile(dir=self.tmpdir, suffix=suffix, delete=False)

self_.file.close() # NamedTemporaryFile passes out open file handles, but callers prefer filenames (and open their own handles manually if needed)

return self_.file.name

def __exit__(self_, _type, _value, _traceback): # noqa: DC02

if not self.save_debug_files:

utils.delete_file(self_.file.name)

return TempFileObject()

def get_dir(self):

"""Return a named temp directory with the given prefix."""

directory = tempfile.mkdtemp(dir=self.tmpdir)

self.note(directory)

return directory

def clean(self):

if self.save_debug_files:

print(f'not cleaning up temp files since in debug-save mode, see them in {self.tmpdir}', file=sys.stderr)

return

for filename in self.to_clean:

utils.delete_file(filename)

self.to_clean = []