|
3 | 3 |
|
4 | 4 |
|
5 | 5 | class suppress_stdout_stderr(object): |
| 6 | + # NOTE: these must be "saved" here to avoid exceptions when using |
| 7 | + # this context manager inside of a __del__ method |
| 8 | + open = open |
| 9 | + sys = sys |
| 10 | + os = os |
| 11 | + |
6 | 12 | # Oddly enough this works better than the contextlib version |
7 | 13 | def __enter__(self): |
8 | | - self.outnull_file = open(os.devnull, "w") |
9 | | - self.errnull_file = open(os.devnull, "w") |
| 14 | + self.outnull_file = self.open(self.os.devnull, "w") |
| 15 | + self.errnull_file = self.open(self.os.devnull, "w") |
10 | 16 |
|
11 | | - self.old_stdout_fileno_undup = sys.stdout.fileno() |
12 | | - self.old_stderr_fileno_undup = sys.stderr.fileno() |
| 17 | + self.old_stdout_fileno_undup = self.sys.stdout.fileno() |
| 18 | + self.old_stderr_fileno_undup = self.sys.stderr.fileno() |
13 | 19 |
|
14 | | - self.old_stdout_fileno = os.dup(sys.stdout.fileno()) |
15 | | - self.old_stderr_fileno = os.dup(sys.stderr.fileno()) |
| 20 | + self.old_stdout_fileno = self.os.dup(self.sys.stdout.fileno()) |
| 21 | + self.old_stderr_fileno = self.os.dup(self.sys.stderr.fileno()) |
16 | 22 |
|
17 | | - self.old_stdout = sys.stdout |
18 | | - self.old_stderr = sys.stderr |
| 23 | + self.old_stdout = self.sys.stdout |
| 24 | + self.old_stderr = self.sys.stderr |
19 | 25 |
|
20 | | - os.dup2(self.outnull_file.fileno(), self.old_stdout_fileno_undup) |
21 | | - os.dup2(self.errnull_file.fileno(), self.old_stderr_fileno_undup) |
| 26 | + self.os.dup2(self.outnull_file.fileno(), self.old_stdout_fileno_undup) |
| 27 | + self.os.dup2(self.errnull_file.fileno(), self.old_stderr_fileno_undup) |
22 | 28 |
|
23 | | - sys.stdout = self.outnull_file |
24 | | - sys.stderr = self.errnull_file |
| 29 | + self.sys.stdout = self.outnull_file |
| 30 | + self.sys.stderr = self.errnull_file |
25 | 31 | return self |
26 | 32 |
|
27 | 33 | def __exit__(self, *_): |
28 | | - sys.stdout = self.old_stdout |
29 | | - sys.stderr = self.old_stderr |
| 34 | + self.sys.stdout = self.old_stdout |
| 35 | + self.sys.stderr = self.old_stderr |
30 | 36 |
|
31 | | - os.dup2(self.old_stdout_fileno, self.old_stdout_fileno_undup) |
32 | | - os.dup2(self.old_stderr_fileno, self.old_stderr_fileno_undup) |
| 37 | + self.os.dup2(self.old_stdout_fileno, self.old_stdout_fileno_undup) |
| 38 | + self.os.dup2(self.old_stderr_fileno, self.old_stderr_fileno_undup) |
33 | 39 |
|
34 | | - os.close(self.old_stdout_fileno) |
35 | | - os.close(self.old_stderr_fileno) |
| 40 | + self.os.close(self.old_stdout_fileno) |
| 41 | + self.os.close(self.old_stderr_fileno) |
36 | 42 |
|
37 | 43 | self.outnull_file.close() |
38 | 44 | self.errnull_file.close() |
0 commit comments