Allowing sys.stind.readline receive size parameter

ulises-blacklane · sebastinas · commit ebc920677874 · 2022-05-28T12:20:42.000+02:00
diff --git a/bpython/curtsiesfrontend/repl.py b/bpython/curtsiesfrontend/repl.py
@@ -179,12 +179,17 @@ def add_input_character(self, e: str) -> None:
         )
         self.cursor_offset += 1
 
-    def readline(self):
-        self.has_focus = True
-        self.repl.send_to_stdin(self.current_line)
-        value = self.coderunner.request_from_main_context()
-        self.readline_results.append(value)
-        return value
+    def readline(self, size=-1):
+        if not isinstance(size, int):
+            raise TypeError(f"'{type(size).__name__}' object cannot be interpreted as an integer")
+        elif size == 0:
+            return ''
+        else:
+            self.has_focus = True
+            self.repl.send_to_stdin(self.current_line)
+            value = self.coderunner.request_from_main_context()
+            self.readline_results.append(value)
+            return value if size <= -1 else value[:size]
 
     def readlines(self, size=-1):
         return list(iter(self.readline, ""))
