| import re |
| from typing import Iterable, List, Tuple |
| |
| from ._loop import loop_last |
| from .cells import cell_len, chop_cells |
| |
| re_word = re.compile(r"\s*\S+\s*") |
| |
| |
| def words(text: str) -> Iterable[Tuple[int, int, str]]: |
| position = 0 |
| word_match = re_word.match(text, position) |
| while word_match is not None: |
| start, end = word_match.span() |
| word = word_match.group(0) |
| yield start, end, word |
| word_match = re_word.match(text, end) |
| |
| |
| def divide_line(text: str, width: int, fold: bool = True) -> List[int]: |
| divides: List[int] = [] |
| append = divides.append |
| line_position = 0 |
| _cell_len = cell_len |
| for start, _end, word in words(text): |
| word_length = _cell_len(word.rstrip()) |
| if line_position + word_length > width: |
| if word_length > width: |
| if fold: |
| chopped_words = chop_cells(word, max_size=width, position=0) |
| for last, line in loop_last(chopped_words): |
| if start: |
| append(start) |
| |
| if last: |
| line_position = _cell_len(line) |
| else: |
| start += len(line) |
| else: |
| if start: |
| append(start) |
| line_position = _cell_len(word) |
| elif line_position and start: |
| append(start) |
| line_position = _cell_len(word) |
| else: |
| line_position += _cell_len(word) |
| return divides |
| |
| |
| if __name__ == "__main__": # pragma: no cover |
| from .console import Console |
| |
| console = Console(width=10) |
| console.print("12345 abcdefghijklmnopqrstuvwyxzABCDEFGHIJKLMNOPQRSTUVWXYZ 12345") |
| print(chop_cells("abcdefghijklmnopqrstuvwxyz", 10, position=2)) |
