add more methods to fake stdin/err/out

thomasballinger · thomasballinger · commit c81686542ef7 · 2014-03-02T09:13:00.000-05:00
diff --git a/bpython/curtsiesfrontend/coderunner.py b/bpython/curtsiesfrontend/coderunner.py
@@ -115,6 +115,13 @@ def __init__(self, coderunner, please):
     def write(self, *args, **kwargs):
         self.please(*args, **kwargs)
         return self.coderunner.refresh_and_get_value()
+    def writelines(self, l):
+        for s in l:
+            self.write(s)
+    def flush(self):
+        pass
+    def isatty(self):
+        return True
 
 def test_simple():
     orig_stdout = sys.stdout
diff --git a/bpython/curtsiesfrontend/repl.py b/bpython/curtsiesfrontend/repl.py
@@ -7,6 +7,7 @@
 import greenlet
 import subprocess
 import tempfile
+import errno
 
 from bpython.autocomplete import Autocomplete, SIMPLE
 from bpython.repl import Repl as BpythonRepl
@@ -90,6 +91,26 @@ def readline(self):
         self.readline_results.append(value)
         return value
 
+    def readlines(self, size=-1):
+        return list(iter(self.readline, ''))
+
+    def __iter__(self):
+        return iter(self.readlines())
+
+    def isatty(self):
+        return True
+
+    def flush(self):
+        """Flush the internal buffer. This is a no-op. Flushing stdin
+        doesn't make any sense anyway."""
+
+    def write(self, value):
+        # XXX IPython expects sys.stdin.write to exist, there will no doubt be
+        # others, so here's a hack to keep them happy
+        raise IOError(errno.EBADF, "sys.stdin is read-only")
+
+    #TODO write a read() method
+
 class ReevaluateFakeStdin(object):
     def __init__(self, fakestdin, repl):
         self.fakestdin = fakestdin
