return string
-def justify_text(text: str, *, width: int = 80, alignment: str = "c") -> str:
+def justify_text(text: str, *, width: int = 80, alignment: str = "c", indent_by: int = 0) -> str:
"""
- Justifies text.
+ Justifies text optionally with initial indentation.
>>> justify_text('This is a test of the emergency broadcast system. This is only a test.',
... width=40, alignment='j') #doctest: +NORMALIZE_WHITESPACE
'This is a test of the emergency\\nbroadcast system. This is only a test.'
"""
- retval = ""
- line = ""
+ retval = ''
+ indent = ''
+ if indent_by > 0:
+ indent += ' ' * indent_by
+ line = indent
+
for word in text.split():
if (
len(string_utils.strip_ansi_sequences(line))
) > width:
line = line[1:]
line = justify_string(line, width=width, alignment=alignment)
- retval = retval + "\n" + line
- line = ""
- line = line + " " + word
+ retval = retval + '\n' + line
+ line = indent
+ line = line + ' ' + word
if len(string_utils.strip_ansi_sequences(line)) > 0:
- retval += "\n" + line[1:]
+ if alignment != 'j':
+ retval += "\n" + justify_string(line[1:], width=width, alignment=alignment)
+ else:
+ retval += "\n" + line[1:]
return retval[1:]
print(self.pad_prefix + self.padding * self.level + text, end='')
-def header(title: str, *, width: int = 80, color: str = ''):
+def header(
+ title: str,
+ *,
+ width: Optional[int] = None,
+ align: Optional[str] = None,
+ style: Optional[str] = 'solid',
+):
"""
Returns a nice header line with a title.
- >>> header('title', width=60, color='')
+ >>> header('title', width=60, style='ascii')
'----[ title ]-----------------------------------------------'
"""
- w = width
- w -= len(string_utils.strip_ansi_sequences(title)) + 4
- if w >= 4:
- left = 4 * '-'
- right = (w - 4) * '-'
- if color != '' and color is not None:
- r = reset()
- else:
- r = ''
- return f'{left}[ {color}{title}{r} ]{right}'
+ if not width:
+ width = get_console_rows_columns().columns
+ if not align:
+ align = 'left'
+ if not style:
+ style = 'ascii'
+
+ text_len = len(string_utils.strip_ansi_sequences(title))
+ if align == 'left':
+ left = 4
+ right = width - (left + text_len + 4)
+ elif align == 'right':
+ right = 4
+ left = width - (right + text_len + 4)
+ else:
+ left = int((width - (text_len + 4)) / 2)
+ right = left
+ while left + text_len + 4 + right < width:
+ right += 1
+
+ if style == 'solid':
+ line_char = '━'
+ begin = ''
+ end = ''
+ elif style == 'dashed':
+ line_char = '┅'
+ begin = ''
+ end = ''
else:
- return ''
+ line_char = '-'
+ begin = '['
+ end = ']'
+ return line_char * left + begin + ' ' + title + ' ' + end + line_char * right
def box(