gh-111201: Allow bracketed paste to work

pablogsal · pablogsal · commit 431566850982 · 2024-05-07T12:52:20.000+01:00
diff --git a/Lib/_pyrepl/commands.py b/Lib/_pyrepl/commands.py
@@ -462,3 +462,13 @@ class paste_mode(Command):
     def do(self) -> None:
         self.reader.paste_mode = not self.reader.paste_mode
         self.reader.dirty = True
+
+
+class enable_bracketed_paste(Command):
+    def do(self) -> None:
+        self.reader.paste_mode = True
+        
+class disable_bracketed_paste(Command):
+    def do(self) -> None:
+        self.reader.paste_mode = False
+        self.reader.insert("\n")
diff --git a/Lib/_pyrepl/reader.py b/Lib/_pyrepl/reader.py
@@ -128,6 +128,8 @@ def make_default_commands() -> dict[CommandName, type[Command]]:
         (r"\M-9", "digit-arg"),
         # (r'\M-\n', 'insert-nl'),
         ("\\\\", "self-insert"),
+        (r"\x1b[200~", "enable_bracketed_paste"),
+        (r"\x1b[201~", "disable_bracketed_paste"),
     ]
     + [(c, "self-insert") for c in map(chr, range(32, 127)) if c != "\\"]
     + [(c, "self-insert") for c in map(chr, range(128, 256)) if c.isalpha()]
diff --git a/Lib/_pyrepl/simple_interact.py b/Lib/_pyrepl/simple_interact.py
@@ -79,6 +79,16 @@ def showtraceback(self):
         super().showtraceback(colorize=self.can_colorize)
 
 
+def _enable_bracketed_paste() -> None:
+    sys.stdout.write("\x1b[?2004h")
+    sys.stdout.flush()
+
+
+def _disable_bracketed_paste() -> None:
+    sys.stdout.write("\x1b[?2004l")
+    sys.stdout.flush()
+
+
 def run_multiline_interactive_console(
     mainmodule: ModuleType | None= None, future_flags: int = 0
 ) -> None:
@@ -137,9 +147,12 @@ def more_lines(unicodetext: str) -> bool:
             ps1 = getattr(sys, "ps1", ">>> ")
             ps2 = getattr(sys, "ps2", "... ")
             try:
+                _enable_bracketed_paste()
                 statement = multiline_input(more_lines, ps1, ps2)
             except EOFError:
                 break
+            finally:
+                _disable_bracketed_paste()
 
             if maybe_run_command(statement):
                 continue
diff --git a/Lib/test/test_pyrepl.py b/Lib/test/test_pyrepl.py
@@ -811,6 +811,46 @@ def test_paste_not_in_paste_mode(self):
         reader = self.prepare_reader(events)
         output = multiline_input(reader)
         self.assertEqual(output, output_code)
+    
+    def test_bracketed_paste(self):
+        """Test that bracketed paste using \x1b[200~ and \x1b[201~ works."""
+        # fmt: off
+        input_code = (
+            'def a():\n'
+            '  for x in range(10):\n'
+            '\n'
+            '    if x%2:\n'
+            '      print(x)\n'
+            '\n'
+            '    else:\n'
+            '      pass\n'
+        )
+        # fmt: on
+
+        output_code = (
+            'def a():\n'
+            '  for x in range(10):\n'
+            '\n'
+            '    if x%2:\n'
+            '      print(x)\n'
+            '\n'
+            '    else:\n'
+            '      pass\n'
+            '\n'
+        )
+
+        paste_start = "\x1b[200~"
+        paste_end = "\x1b[201~"
+
+        events = itertools.chain(
+            code_to_events(paste_start),
+            code_to_events(input_code),
+            code_to_events(paste_end),
+            code_to_events("\n"),
+        )
+        reader = self.prepare_reader(events)
+        output = multiline_input(reader)
+        self.assertEqual(output, output_code)
 
 
 class TestReader(TestCase):

Original file line number	Diff line number	Diff line change
`@@ -128,6 +128,8 @@ def make_default_commands() -> dict[CommandName, type[Command]]:`
`128`	`128`	`(r"\M-9", "digit-arg"),`
`129`	`129`	`# (r'\M-\n', 'insert-nl'),`
`130`	`130`	`("\\\\", "self-insert"),`
	`131`	`+ (r"\x1b[200~", "enable_bracketed_paste"),`
	`132`	`+ (r"\x1b[201~", "disable_bracketed_paste"),`
`131`	`133`	`]`
`132`	`134`	`+ [(c, "self-insert") for c in map(chr, range(32, 127)) if c != "\\"]`
`133`	`135`	`+ [(c, "self-insert") for c in map(chr, range(128, 256)) if c.isalpha()]`