FilesExpand file tree

 batch-processing

/

_utils.py

Copy path

Blame

More file actions

Blame

More file actions

Latest commit

 

History

History

History

57 lines (42 loc) · 1.98 KB

 batch-processing

/

_utils.py

Copy path

Top

File metadata and controls

• Code
• Blame

57 lines (42 loc) · 1.98 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

import os

import sys

import sys, traceback

# Avoid "LookupError: unknown encoding: ascii" when open() called in a destructor

outnull_file = open(os.devnull, "w")

errnull_file = open(os.devnull, "w")

class suppress_stdout_stderr(object):

# NOTE: these must be "saved" here to avoid exceptions when using

# this context manager inside of a __del__ method

sys = sys

os = os

def __init__(self, disable: bool = True):

self.disable = disable

# Oddly enough this works better than the contextlib version

def __enter__(self):

if self.disable:

return self

# Check if sys.stdout and sys.stderr have fileno method

if not hasattr(self.sys.stdout, 'fileno') or not hasattr(self.sys.stderr, 'fileno'):

return self # Return the instance without making changes

self.old_stdout_fileno_undup = self.sys.stdout.fileno()

self.old_stderr_fileno_undup = self.sys.stderr.fileno()

self.old_stdout_fileno = self.os.dup(self.old_stdout_fileno_undup)

self.old_stderr_fileno = self.os.dup(self.old_stderr_fileno_undup)

self.old_stdout = self.sys.stdout

self.old_stderr = self.sys.stderr

self.os.dup2(outnull_file.fileno(), self.old_stdout_fileno_undup)

self.os.dup2(errnull_file.fileno(), self.old_stderr_fileno_undup)

self.sys.stdout = outnull_file

self.sys.stderr = errnull_file

return self

def __exit__(self, *_):

if self.disable:

return

# Check if sys.stdout and sys.stderr have fileno method

if hasattr(self.sys.stdout, 'fileno') and hasattr(self.sys.stderr, 'fileno'):

self.sys.stdout = self.old_stdout

self.sys.stderr = self.old_stderr

self.os.dup2(self.old_stdout_fileno, self.old_stdout_fileno_undup)

self.os.dup2(self.old_stderr_fileno, self.old_stderr_fileno_undup)

self.os.close(self.old_stdout_fileno)

self.os.close(self.old_stderr_fileno)