mirror of
ssh://git.janware.com/srv/git/janware/proj/jw-python
synced 2026-01-15 18:03:31 +01:00
36 lines
1 KiB
Python
36 lines
1 KiB
Python
from __future__ import print_function
|
|
import os
|
|
import io
|
|
import sys
|
|
import traceback
|
|
from fcntl import fcntl, F_GETFL, F_SETFL
|
|
|
|
class MuteStdIO: # export
|
|
|
|
def __init__(self, mute_stderr=False, mute_stdout=True):
|
|
# TODO: arguments not implemented
|
|
pass
|
|
|
|
def __enter__(self):
|
|
rfd, wfd = os.pipe()
|
|
flags = fcntl(rfd, F_GETFL)
|
|
fcntl(rfd, F_SETFL, flags | os.O_NONBLOCK)
|
|
self.rfile = io.open(rfd, 'r')
|
|
self.fake_stdout_stream = io.open(wfd, 'w')
|
|
self.real_stdout_fd = os.dup(1)
|
|
#os.close(1)
|
|
os.dup2(wfd, 1)
|
|
|
|
def __exit__(self, type, value, traceback):
|
|
sys.stdout.flush()
|
|
os.dup2(self.real_stdout_fd, 1)
|
|
if type is not None:
|
|
#print("-------- Error while stdio was suppressed --------")
|
|
#traceback.print_stack()
|
|
#print(traceback)
|
|
print("-------- Captured output --------")
|
|
print(*self.rfile.readlines())
|
|
self.rfile.close()
|
|
#print('type = ' + str(type))
|
|
#print('value = ' + str(value))
|
|
raise type(value)
|