summaryrefslogtreecommitdiff
path: root/cmd2/utils.py
diff options
context:
space:
mode:
Diffstat (limited to 'cmd2/utils.py')
-rw-r--r--cmd2/utils.py6
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