Python: BudouXで指定した幅を目安に改行する (改行後が2文字以内にならないよう調整)

wrap_text.py

import budoux

parser = budoux.load_default_japanese_parser()

def wrap_text(text: str, least_width: int):
    phrases = parser.parse(text)

    lines = ['']
    for phrase in phrases:
        if len(lines) > 0 and len(lines[-1]) >= least_width:
            lines.append('')
        lines[-1] += phrase

    # merge last line if it is too short
    if len(lines) >= 2 and len(lines[-1]) <= 2:
        last_line = lines.pop()
        lines[-1] += last_line

    return '\n'.join(lines)