Fix FakeStdin.readline() (broke in r217).

Trundle · Trundle · commit a764a74eff56 · 2009-08-11T21:48:43.000+02:00
Also return a bytestring instead of unicode.
diff --git a/bpython/cli.py b/bpython/cli.py
@@ -181,17 +181,21 @@ def readline(self):
         only one I've done anything with. The others are just there in case
         someone does something weird to stop it from blowing up."""
 
+        curses.raw(True)
         buffer = ''
-        while True:
-            key = self.interface.get_key()
-            sys.stdout.write(key)
+        try:
+            while True:
+                key = self.interface.get_key()
+                sys.stdout.write(key)
 # Include the \n in the buffer - raw_input() seems to deal with trailing
 # linebreaks and will break if it gets an empty string.
-            buffer += key
-            if key == '\n':
-                break
+                buffer += key
+                if key == '\n':
+                    break
+        finally:
+            curses.raw(False)
 
-        return buffer
+        return buffer.encode(getpreferredencoding())
 
     def read(self, x):
         pass
