Spaces:
Sleeping
Sleeping
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)) | |