mirrors/cpython
1
0
Fork 0
mirror of https://github.com/python/cpython.git synced 2025-08-31 22:18:28 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 9

gh-118835: pyrepl: Fix prompt length computation for custom prompts containing ANSI escape codes (#119942)

Browse source
This commit is contained in:
Daniel Hollas 2024-06-03 18:07:06 +01:00 • committed by GitHub
parent 4e8aa32245
commit 2e0aa731ae
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
3 changed files with 41 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

10
Lib/_pyrepl/reader.py
View file

@ -28,7 +28,7 @@ from _colorize import can_colorize, ANSIColors # type: ignore[import-not-found]
from . import commands, console, input
from .utils import ANSI_ESCAPE_SEQUENCE, wlen
from .utils import ANSI_ESCAPE_SEQUENCE, wlen, str_width
from .trace import trace
@ -339,7 +339,8 @@ class Reader:
screeninfo.append((0, []))
return screen
def process_prompt(self, prompt: str) -> tuple[str, int]:
@staticmethod
def process_prompt(prompt: str) -> tuple[str, int]:
"""Process the prompt.
This means calculate the length of the prompt. The character \x01
@ -351,6 +352,11 @@ class Reader:
# sequences if they were not explicitly within \x01...\x02.
# They are CSI (or ANSI) sequences ( ESC [ ... LETTER )
# wlen from utils already excludes ANSI_ESCAPE_SEQUENCE chars,
# which breaks the logic below so we redefine it here.
def wlen(s: str) -> int:
return sum(str_width(i) for i in s)
out_prompt = ""
l = wlen(prompt)
pos = 0