diff options
Diffstat (limited to 'cmd2/utils.py')
| -rw-r--r-- | cmd2/utils.py | 18 |
1 files changed, 9 insertions, 9 deletions
diff --git a/cmd2/utils.py b/cmd2/utils.py index 310f0284..dd71cfc2 100644 --- a/cmd2/utils.py +++ b/cmd2/utils.py @@ -399,6 +399,13 @@ class ProcReader(object): self._out_thread.join() self._err_thread.join() + # Handle case where the process ended before the last read could be done + out, err = self._proc.communicate() + if out: + self._write_bytes(self._stdout, out) + if err: + self._write_bytes(self._stderr, err) + def _reader_thread_func(self, read_stdout: bool) -> None: """ Thread function that reads a stream from the process @@ -416,15 +423,8 @@ class ProcReader(object): # noinspection PyUnresolvedReferences available = read_stream.peek() if available: - out = read_stream.read(len(available)) - self._write_bytes(write_stream, out) - - # Handle case where the process ended before the last could be done - # noinspection PyUnresolvedReferences - available = read_stream.peek() - if available: - out = read_stream.read(len(available)) - self._write_bytes(write_stream, out) + read_stream.read(len(available)) + self._write_bytes(write_stream, available) @staticmethod def _write_bytes(stream: Union[BinaryIO, TextIO], to_write: bytes) -> None: |