diff options
Diffstat (limited to 'text_utils.py')
| -rw-r--r-- | text_utils.py | 22 |
1 files changed, 22 insertions, 0 deletions
diff --git a/text_utils.py b/text_utils.py index 8ea6e19..49ff9b9 100644 --- a/text_utils.py +++ b/text_utils.py @@ -268,6 +268,28 @@ class Indenter: print(self.pad_prefix + self.padding * self.level + text, end='') +def header(title: str, *, width: int = 80, color: str = ''): + """ + Returns a nice header line with a title. + + >>> header('title', width=60, color='') + '----[ title ]-----------------------------------------------' + + """ + w = width + w -= (len(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}' + else: + return '' + + if __name__ == '__main__': import doctest doctest.testmod() |
