gh-119434: Fix culmitive errors in wrapping as lines proceed (#119435)

Fix culmitive errors in wrapping as lines proceed
python · May 22, 2024 · e3bf538 · e3bf538
1 parent 9b422fc
commit e3bf538
Showing 1 changed file with 9 additions and 3 deletions.
diff --git a/Lib/_pyrepl/reader.py b/Lib/_pyrepl/reader.py
@@ -307,7 +307,8 @@ def calc_complete_screen(self) -> list[str]:
                 screen.append(prompt + l)
                 screeninfo.append((lp, l2))
             else:
-                for i in range(wrapcount + 1):
+                i = 0
+                while l:
                     prelen = lp if i == 0 else 0
                     index_to_wrap_before = 0
                     column = 0
@@ -317,12 +318,17 @@ def calc_complete_screen(self) -> list[str]:
                         index_to_wrap_before += 1
                         column += character_width
                     pre = prompt if i == 0 else ""
-                    post = "\\" if i != wrapcount else ""
-                    after = [1] if i != wrapcount else []
+                    if len(l) > index_to_wrap_before:
+                        post = "\\"
+                        after = [1]
+                    else:
+                        post = ""
+                        after = []
                     screen.append(pre + l[:index_to_wrap_before] + post)
                     screeninfo.append((prelen, l2[:index_to_wrap_before] + after))
                     l = l[index_to_wrap_before:]
                     l2 = l2[index_to_wrap_before:]
+                    i += 1
         self.screeninfo = screeninfo
         self.cxy = self.pos2xy()
         if self.msg and self.msg_at_bottom: