diff options
Diffstat (limited to 'cmd2/utils.py')
-rw-r--r-- | cmd2/utils.py | 6 |
1 files changed, 3 insertions, 3 deletions
diff --git a/cmd2/utils.py b/cmd2/utils.py index 29ae332a..697cf6f8 100644 --- a/cmd2/utils.py +++ b/cmd2/utils.py @@ -455,10 +455,10 @@ class ProcReader(object): :param to_write: the bytes being written """ try: - if 'b' in stream.mode: - stream.write(to_write) - else: + if hasattr(stream, 'buffer'): stream.buffer.write(to_write) + else: + stream.write(to_write) except BrokenPipeError: # This occurs if output is being piped to a process that closed pass |