fix: custom REPL input/output

Sébastien Grignard · jonathanslenders · commit d8b4eae00f2c · 2020-04-14T17:46:06.000+02:00
diff --git a/ptpython/python_input.py b/ptpython/python_input.py
@@ -324,7 +324,7 @@ def __init__(
             extra_toolbars=self._extra_toolbars,
         )
 
-        self.app = self._create_application()
+        self.app = self._create_application(input, output)
 
         if vi_mode:
             self.app.editing_mode = EditingMode.VI
@@ -728,7 +728,11 @@ def get_values():
             ),
         ]
 
-    def _create_application(self) -> Application:
+    def _create_application(
+        self,
+        input: Optional[Input],
+        output: Optional[Output]
+    ) -> Application:
         """
         Create an `Application` instance.
         """
@@ -758,6 +762,8 @@ def _create_application(self) -> Application:
             style_transformation=self.style_transformation,
             include_default_pygments_style=False,
             reverse_vi_search_direction=True,
+            input=input,
+            output=output,
         )
 
     def _create_buffer(self) -> Buffer:
diff --git a/ptpython/repl.py b/ptpython/repl.py
@@ -187,6 +187,7 @@ def compile_with_flags(code: str, mode: str):
                         style=self._current_style,
                         style_transformation=self.style_transformation,
                         include_default_pygments_style=False,
+                        output=output,
                     )
 
             # If not a valid `eval` expression, run using `exec` instead.
@@ -233,6 +234,7 @@ def _handle_exception(self, e: Exception) -> None:
             style=self._current_style,
             style_transformation=self.style_transformation,
             include_default_pygments_style=False,
+            output=output,
         )
 
         output.write("%s\n" % e)

Original file line number	Diff line number	Diff line change
`@@ -187,6 +187,7 @@ def compile_with_flags(code: str, mode: str):`
`187`	`187`	`style=self._current_style,`
`188`	`188`	`style_transformation=self.style_transformation,`
`189`	`189`	`include_default_pygments_style=False,`
	`190`	`+ output=output,`
`190`	`191`	`)`
`191`	`192`
`192`	`193`	# If not a valid `eval` expression, run using `exec` instead.
`@@ -233,6 +234,7 @@ def _handle_exception(self, e: Exception) -> None:`
`233`	`234`	`style=self._current_style,`
`234`	`235`	`style_transformation=self.style_transformation,`
`235`	`236`	`include_default_pygments_style=False,`
	`237`	`+ output=output,`
`236`	`238`	`)`
`237`	`239`
`238`	`240`	`output.write("%s\n" % e)`