add bracketed paste mode for Python3.13 and above

python_files/normalizeSelection.py

@@ -8,6 +8,8 @@
 import textwrap
 from typing import Iterable
 
+attach_bracket_paste = sys.version_info >= (3, 13)
+
 
 def split_lines(source):
     """
@@ -279,14 +281,20 @@ def get_next_block_lineno(which_line_next):
         normalized = result["normalized_smart_result"]
         which_line_next = result["which_line_next"]
         if normalized == "deprecated":
-            data = json.dumps({"normalized": normalized})
+            data = json.dumps(
+                {"normalized": normalized, "attach_bracket_paste": attach_bracket_paste}
+            )
         else:
             data = json.dumps(
-                {"normalized": normalized, "nextBlockLineno": result["which_line_next"]}
+                {
+                    "normalized": normalized,
+                    "nextBlockLineno": result["which_line_next"],
+                    "attach_bracket_paste": attach_bracket_paste,
+                }
             )
     else:
         normalized = normalize_lines(contents["code"])
-        data = json.dumps({"normalized": normalized})
+        data = json.dumps({"normalized": normalized, "attach_bracket_paste": attach_bracket_paste})
 
     stdout = sys.stdout if sys.version_info < (3,) else sys.stdout.buffer
     stdout.write(data.encode("utf-8"))

src/client/terminals/codeExecution/helper.ts

@@ -118,6 +118,16 @@ export class CodeExecutionHelper implements ICodeExecutionHelper {
                 const lineOffset = object.nextBlockLineno - activeEditor!.selection.start.line - 1;
                 await this.moveToNextBlock(lineOffset, activeEditor);
             }
+            // For new _pyrepl for Python3.13 and above, we need to send code via bracketed paste mode.
+            if (object.attach_bracket_paste) {
+                // return `\u001b[200~${object.normalized.trim()}\u001b[201~`;
+                let trimmedNormalized = object.normalized.replace(/\n$/, '');
+                if (trimmedNormalized.endsWith(':\n')) {
+                    // In case where statement is unfinished via :, truncate so auto-indentation lands nicely.
+                    trimmedNormalized = trimmedNormalized.replace(/\n$/, '');
+                }
+                return `\u001b[200~${trimmedNormalized}\u001b[201~`;
+            }
 
             return parse(object.normalized);
         } catch (ex) {

src/test/terminals/codeExecution/helper.test.ts

@@ -8,6 +8,7 @@ import * as path from 'path';
 import { SemVer } from 'semver';
 import * as TypeMoq from 'typemoq';
 import { Position, Range, Selection, TextDocument, TextEditor, TextLine, Uri } from 'vscode';
+import * as sinon from 'sinon';
 import * as fs from '../../../client/common/platform/fs-paths';
 import {
     IActiveResourceService,
@@ -35,7 +36,7 @@ import { ICodeExecutionHelper } from '../../../client/terminals/types';
 import { PYTHON_PATH } from '../../common';
 
 const TEST_FILES_PATH = path.join(EXTENSION_ROOT_DIR, 'src', 'test', 'python_files', 'terminalExec');
-
+// TODO: tests for 3.13 relevant sequences
 suite('Terminal - Code Execution Helper', () => {
     let activeResourceService: TypeMoq.IMock<IActiveResourceService>;
     let documentManager: TypeMoq.IMock<IDocumentManager>;
@@ -49,10 +50,11 @@ suite('Terminal - Code Execution Helper', () => {
     let workspaceService: TypeMoq.IMock<IWorkspaceService>;
     let configurationService: TypeMoq.IMock<IConfigurationService>;
     let pythonSettings: TypeMoq.IMock<IPythonSettings>;
+    let jsonParseStub: sinon.SinonStub;
     const workingPython: PythonEnvironment = {
         path: PYTHON_PATH,
-        version: new SemVer('3.6.6-final'),
-        sysVersion: '1.0.0.0',
+        version: new SemVer('3.13.0'),
+        sysVersion: '3.13.0',
         sysPrefix: 'Python',
         displayName: 'Python',
         envType: EnvironmentType.Unknown,
@@ -134,7 +136,38 @@ suite('Terminal - Code Execution Helper', () => {
         editor.setup((e) => e.document).returns(() => document.object);
     });
 
+    test('normalizeLines should handle attach_bracket_paste correctly', async () => {
+        configurationService
+            .setup((c) => c.getSettings(TypeMoq.It.isAny()))
+            .returns({
+                REPL: {
+                    EnableREPLSmartSend: false,
+                    REPLSmartSend: false,
+                },
+                // eslint-disable-next-line @typescript-eslint/no-explicit-any
+            } as any);
+        const actualProcessService = new ProcessService();
+        processService
+            .setup((p) => p.execObservable(TypeMoq.It.isAny(), TypeMoq.It.isAny(), TypeMoq.It.isAny()))
+            .returns((file, args, options) =>
+                actualProcessService.execObservable.apply(actualProcessService, [file, args, options]),
+            );
+
+        jsonParseStub = sinon.stub(JSON, 'parse');
+        const mockResult = {
+            normalized: 'print("Looks like you are on 3.13")',
+            attach_bracket_paste: true,
+        };
+        jsonParseStub.returns(mockResult);
+
+        const result = await helper.normalizeLines('print("Looks like you are on 3.13")');
+
+        expect(result).to.equal(`\u001b[200~print("Looks like you are on 3.13")\u001b[201~`);
+        jsonParseStub.restore();
+    });
+
     test('normalizeLines should call normalizeSelection.py', async () => {
+        jsonParseStub.restore();
         let execArgs = '';
 
         processService
@@ -186,7 +219,7 @@ suite('Terminal - Code Execution Helper', () => {
                 path.join(TEST_FILES_PATH, `sample${fileNameSuffix}_normalized_selection.py`),
                 'utf8',
             );
-
+            // python3 -m pythonFiles.tests
             await ensureCodeIsNormalized(code, expectedCode);
         });
     });