Suspend REPL colorizing when in a REPL interactive command

Author: olgarithms

Lib/_pyrepl/reader.py

@@ -619,6 +619,16 @@ def suspend(self) -> SimpleContextManager:
                 setattr(self, arg, prev_state[arg])
             self.prepare()
 
+    @contextmanager
+    def suspend_colorization(self) -> SimpleContextManager:
+        try:
+            old_can_colorize = self.can_colorize
+            self.can_colorize = False
+            yield
+        finally:
+            self.can_colorize = old_can_colorize
+
+
     def finish(self) -> None:
         """Called when a command signals that we're finished."""
         pass

Lib/_pyrepl/simple_interact.py

@@ -125,7 +125,7 @@ def maybe_run_command(statement: str) -> bool:
         command = REPL_COMMANDS[statement]
         if callable(command):
             # Make sure that history does not change because of commands
-            with reader.suspend_history():
+            with reader.suspend_history(), reader.suspend_colorization():
                 command()
             return True
         return False