Add common.trim for strings method
parent
90600abf4a
commit
cdea170178
@ -0,0 +1,31 @@
|
||||
import sys
|
||||
|
||||
|
||||
# source: https://peps.python.org/pep-0257/#handling-docstring-indentation
|
||||
def trim(docstring: str) -> str:
|
||||
"""
|
||||
Trim strings like docstrings are trimmed following PEP 257
|
||||
"""
|
||||
if not docstring:
|
||||
return ""
|
||||
# Convert tabs to spaces (following the normal Python rules)
|
||||
# and split into a list of lines:
|
||||
lines = docstring.expandtabs().splitlines()
|
||||
# Determine minimum indentation (first line doesn't count):
|
||||
indent = sys.maxsize
|
||||
for line in lines[1:]:
|
||||
stripped = line.lstrip()
|
||||
if stripped:
|
||||
indent = min(indent, len(line) - len(stripped))
|
||||
# Remove indentation (first line is special):
|
||||
trimmed = [lines[0].strip()]
|
||||
if indent < sys.maxsize:
|
||||
for line in lines[1:]:
|
||||
trimmed.append(line[indent:].rstrip())
|
||||
# Strip off trailing and leading blank lines:
|
||||
while trimmed and not trimmed[-1]:
|
||||
trimmed.pop()
|
||||
while trimmed and not trimmed[0]:
|
||||
trimmed.pop(0)
|
||||
# Return a single string:
|
||||
return "\n".join(trimmed)
|
Loading…
Reference in New Issue