Fix some regex handling and add another example

prompt-toolkit · Jul 12, 2024 · 58c8acf · 58c8acf
1 parent aa25eb9
commit 58c8acf
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 3 deletions.
diff --git a/examples/full-screen/simple-demos/word-wrapping.py b/examples/full-screen/simple-demos/word-wrapping.py
@@ -87,7 +87,7 @@ def get_line_prefix(lineno, wrap_count):
                 if wrap_count
                 else [],
             ),
-            # Truncating (only wrap the first timle around)
+            # Truncating (only wrap the first time around)
             Window(
                 BufferControl(buffer=buff),
                 wrap_lines=True,
@@ -100,6 +100,16 @@ def get_line_prefix(lineno, wrap_count):
                 if wrap_count > 0
                 else fallback(line, lineno, wrap_count, start, end),
             ),
+            # Split only after vowels
+            Window(
+                BufferControl(buffer=buff),
+                wrap_lines=True,
+                wrap_finder=Window._whitespace_wrap_finder(
+                    sep="[aeiouyAEIOUY]",
+                    split="after",
+                    continuation=to_formatted_text("-"),
+                ),
+            ),
         ]
     ),
     floats=[

diff --git a/src/prompt_toolkit/layout/containers.py b/src/prompt_toolkit/layout/containers.py
@@ -1955,9 +1955,9 @@ def _whitespace_wrap_finder(
                 f"Pattern {sep_re.pattern!r} has capture group – use non-capturing groups instead"
             )
         elif split == "after":
-            sep_re = re.compile("(?<={sep_re.pattern})()")
+            sep_re = re.compile(f"(?={sep_re.pattern})()")
         elif split == "before":
-            sep_re = re.compile("(?={sep_re.pattern})()")
+            sep_re = re.compile(f"(?<={sep_re.pattern})()")
         elif split == "remove":
             sep_re = re.compile(f"({sep_re.pattern})")
         else: