Aluode/PerceptionLabPortable

1from __future__ import annotations2 3import re4from typing import Iterable5 6from ._loop import loop_last7from .cells import cell_len, chop_cells8 9re_word = re.compile(r"\s*\S+\s*")10 11 12def words(text: str) -> Iterable[tuple[int, int, str]]:13    """Yields each word from the text as a tuple14    containing (start_index, end_index, word). A "word" in this context may15    include the actual word and any whitespace to the right.16    """17    position = 018    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, word23        word_match = re_word.match(text, end)24 25 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 list28    of cell offsets which the string should be split at in order for it to fit29    within the given width.30 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.35 36    Returns:37        A list of indices to break the line at.38    """39    break_positions: list[int] = []  # offsets to insert the breaks at40    append = break_positions.append41    cell_offset = 042    _cell_len = cell_len43 44    for start, _end, word in words(text):45        word_length = _cell_len(word.rstrip())46        remaining_space = width - cell_offset47        word_fits_remaining_space = remaining_space >= word_length48 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 simply56                # 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 current74                # line, but it *can* fit on to the next (empty) line.75                append(start)76                cell_offset = _cell_len(word)77 78    return break_positions79 80 81if __name__ == "__main__":  # pragma: no cover82    from .console import Console83 84    console = Console(width=10)85    console.print("12345 abcdefghijklmnopqrstuvwyxzABCDEFGHIJKLMNOPQRSTUVWXYZ 12345")86    print(chop_cells("abcdefghijklmnopqrstuvwxyz", 10))87 88    console = Console(width=20)89    console.rule()90    console.print("TextualはPythonの高速アプリケーション開発フレームワークです")91 92    console.rule()93    console.print("アプリケーションは1670万色を使用でき")94