| import sys | |
| import tempfile | |
| class StdoutWrapper: | |
| def __init__(self): | |
| self.temp_file = tempfile.NamedTemporaryFile(mode="w+", delete=False) | |
| self.original_stdout = sys.stdout | |
| def write(self, message: str): | |
| self.temp_file.write(message) | |
| self.temp_file.flush() | |
| print(message, end="", file=self.original_stdout) | |
| def flush(self): | |
| self.temp_file.flush() | |
| def read(self): | |
| self.temp_file.seek(0) | |
| return self.temp_file.read() | |
| def close(self): | |
| self.temp_file.close() | |
| def fileno(self): | |
| return self.temp_file.fileno() | |
| try: | |
| import google.colab | |
| SAFE_STDOUT = StdoutWrapper() | |
| except ImportError: | |
| SAFE_STDOUT = sys.stdout | |