Merge pull request #31 from takluyver/stdout-bytes-py3

Allow importing when sys.stdout is reassigned on Python 3

1 files changed, 8 insertions, 1 deletions

diff --git a/pexpect/__init__.py b/pexpect/__init__.py
index ea39397..18f4102 100644
--- a/pexpect/__init__.py
+++ b/pexpect/__init__.py
@@ -284,7 +284,14 @@ class spawn(object):
         def _chr(c):
             return bytes([c])
         linesep = os.linesep.encode('ascii')
-        write_to_stdout = sys.stdout.buffer.write
+
+        @staticmethod
+        def write_to_stdout(b):
+            try:
+                return sys.stdout.buffer.write(b)
+            except AttributeError:
+                # If stdout has been replaced, it may not have .buffer
+                return sys.stdout.write(b.decode('ascii', 'replace'))
     else:
         allowed_string_types = (basestring,)  # analysis:ignore
         _chr = staticmethod(chr)