forked from varia/varia.website
94 lines
3.3 KiB
Python
94 lines
3.3 KiB
Python
|
from __future__ import annotations
|
||
|
|
|
||
|
|
import re
|
||
|
|
from typing import Iterable
|
||
|
|
|
||
|
|
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]]:
|
||
|
|
"""Yields each word from the text as a tuple
|
||
|
|
containing (start_index, end_index, word). A "word" in this context may
|
||
|
|
include the actual word and any whitespace to the right.
|
||
|
|
"""
|
||
|
|
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]:
|
||
|
|
"""Given a string of text, and a width (measured in cells), return a list
|
||
|
|
of cell offsets which the string should be split at in order for it to fit
|
||
|
|
within the given width.
|
||
|
|
|
||
|
|
Args:
|
||
|
|
text: The text to examine.
|
||
|
|
width: The available cell width.
|
||
|
|
fold: If True, words longer than `width` will be folded onto a new line.
|
||
|
|
|
||
|
|
Returns:
|
||
|
|
A list of indices to break the line at.
|
||
|
|
"""
|
||
|
|
break_positions: list[int] = [] # offsets to insert the breaks at
|
||
|
|
append = break_positions.append
|
||
|
|
cell_offset = 0
|
||
|
|
_cell_len = cell_len
|
||
|
|
|
||
|
|
for start, _end, word in words(text):
|
||
|
|
word_length = _cell_len(word.rstrip())
|
||
|
|
remaining_space = width - cell_offset
|
||
|
|
word_fits_remaining_space = remaining_space >= word_length
|
||
|
|
|
||
|
|
if word_fits_remaining_space:
|
||
|
|
# Simplest case - the word fits within the remaining width for this line.
|
||
|
|
cell_offset += _cell_len(word)
|
||
|
|
else:
|
||
|
|
# Not enough space remaining for this word on the current line.
|
||
|
|
if word_length > width:
|
||
|
|
# The word doesn't fit on any line, so we can't simply
|
||
|
|
# place it on the next line...
|
||
|
|
if fold:
|
||
|
|
# Fold the word across multiple lines.
|
||
|
|
folded_word = chop_cells(word, width=width)
|
||
|
|
for last, line in loop_last(folded_word):
|
||
|
|
if start:
|
||
|
|
append(start)
|
||
|
|
if last:
|
||
|
|
cell_offset = _cell_len(line)
|
||
|
|
else:
|
||
|
|
start += len(line)
|
||
|
|
else:
|
||
|
|
# Folding isn't allowed, so crop the word.
|
||
|
|
if start:
|
||
|
|
append(start)
|
||
|
|
cell_offset = _cell_len(word)
|
||
|
|
elif cell_offset and start:
|
||
|
|
# The word doesn't fit within the remaining space on the current
|
||
|
|
# line, but it *can* fit on to the next (empty) line.
|
||
|
|
append(start)
|
||
|
|
cell_offset = _cell_len(word)
|
||
|
|
|
||
|
|
return break_positions
|
||
|
|
|
||
|
|
|
||
|
|
if __name__ == "__main__": # pragma: no cover
|
||
|
|
from .console import Console
|
||
|
|
|
||
|
|
console = Console(width=10)
|
||
|
|
console.print("12345 abcdefghijklmnopqrstuvwyxzABCDEFGHIJKLMNOPQRSTUVWXYZ 12345")
|
||
|
|
print(chop_cells("abcdefghijklmnopqrstuvwxyz", 10))
|
||
|
|
|
||
|
|
console = Console(width=20)
|
||
|
|
console.rule()
|
||
|
|
console.print("TextualはPythonの高速アプリケーション開発フレームワークです")
|
||
|
|
|
||
|
|
console.rule()
|
||
|
|
console.print("アプリケーションは1670万色を使用でき")
|