| 50 | return flattened |
| 51 | |
| 52 | class TeeStream: |
| 53 | def __init__(self, queue, original_stream): |
| 54 | self.queue = queue |
| 55 | self.original_stream = original_stream |
| 56 | |
| 57 | def write(self, message): |
| 58 | self.original_stream.write(message) |
| 59 | self.original_stream.flush() |
| 60 | if message.strip(): |
| 61 | self.queue.put(message) |
| 62 | |
| 63 | def flush(self): |
| 64 | self.original_stream.flush() |