Coverage for /pythoncovmergedfiles/medio/medio/usr/local/lib/python3.8/site-packages/rich/_wrap.py: 20%
Shortcuts on this page
r m x toggle line displays
j k next/prev highlighted chunk
0 (zero) top of page
1 (one) first highlighted chunk
Shortcuts on this page
r m x toggle line displays
j k next/prev highlighted chunk
0 (zero) top of page
1 (one) first highlighted chunk
1from __future__ import annotations
3import re
4from typing import Iterable
6from ._loop import loop_last
7from .cells import cell_len, chop_cells
9re_word = re.compile(r"\s*\S+\s*")
12def words(text: str) -> Iterable[tuple[int, int, str]]:
13 """Yields each word from the text as a tuple
14 containing (start_index, end_index, word). A "word" in this context may
15 include the actual word and any whitespace to the right.
16 """
17 position = 0
18 word_match = re_word.match(text, position)
19 while word_match is not None:
20 start, end = word_match.span()
21 word = word_match.group(0)
22 yield start, end, word
23 word_match = re_word.match(text, end)
26def divide_line(text: str, width: int, fold: bool = True) -> list[int]:
27 """Given a string of text, and a width (measured in cells), return a list
28 of cell offsets which the string should be split at in order for it to fit
29 within the given width.
31 Args:
32 text: The text to examine.
33 width: The available cell width.
34 fold: If True, words longer than `width` will be folded onto a new line.
36 Returns:
37 A list of indices to break the line at.
38 """
39 break_positions: list[int] = [] # offsets to insert the breaks at
40 append = break_positions.append
41 cell_offset = 0
42 _cell_len = cell_len
44 for start, _end, word in words(text):
45 word_length = _cell_len(word.rstrip())
46 remaining_space = width - cell_offset
47 word_fits_remaining_space = remaining_space >= word_length
49 if word_fits_remaining_space:
50 # Simplest case - the word fits within the remaining width for this line.
51 cell_offset += _cell_len(word)
52 else:
53 # Not enough space remaining for this word on the current line.
54 if word_length > width:
55 # The word doesn't fit on any line, so we can't simply
56 # place it on the next line...
57 if fold:
58 # Fold the word across multiple lines.
59 folded_word = chop_cells(word, width=width)
60 for last, line in loop_last(folded_word):
61 if start:
62 append(start)
63 if last:
64 cell_offset = _cell_len(line)
65 else:
66 start += len(line)
67 else:
68 # Folding isn't allowed, so crop the word.
69 if start:
70 append(start)
71 cell_offset = _cell_len(word)
72 elif cell_offset and start:
73 # The word doesn't fit within the remaining space on the current
74 # line, but it *can* fit on to the next (empty) line.
75 append(start)
76 cell_offset = _cell_len(word)
78 return break_positions
81if __name__ == "__main__": # pragma: no cover
82 from .console import Console
84 console = Console(width=10)
85 console.print("12345 abcdefghijklmnopqrstuvwyxzABCDEFGHIJKLMNOPQRSTUVWXYZ 12345")
86 print(chop_cells("abcdefghijklmnopqrstuvwxyz", 10))
88 console = Console(width=20)
89 console.rule()
90 console.print("TextualはPythonの高速アプリケーション開発フレームワークです")
92 console.rule()
93 console.print("アプリケーションは1670万色を使用でき")