example-code-2e/08-def-type-hints/clip_annot.py
2020-06-11 14:58:15 -03:00

39 lines
945 B
Python

"""
>>> clip('banana ', 6)
'banana'
>>> clip('banana ', 7)
'banana'
>>> clip('banana ', 5)
'banana'
>>> clip('banana split', 6)
'banana'
>>> clip('banana split', 7)
'banana'
>>> clip('banana split', 10)
'banana'
>>> clip('banana split', 11)
'banana'
>>> clip('banana split', 12)
'banana split'
"""
# tag::CLIP_ANNOT[]
def clip(text: str, max_len: int = 80) -> str:
"""Return new ``str`` clipped at last space before or after ``max_len``.
Return full ``text`` if no space found.
"""
end = None
if len(text) > max_len:
space_before = text.rfind(' ', 0, max_len)
if space_before >= 0:
end = space_before
else:
space_after = text.rfind(' ', max_len)
if space_after >= 0:
end = space_after
if end is None:
end = len(text)
return text[:end].rstrip()
# end::CLIP_ANNOT[]