voussoir/voussoirkit
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

Handle the case in which colorama is not installed.

Browse source
This commit is contained in:
voussoir 2022-03-23 14:38:15 -07:00
parent 5ddc83e3ff
commit 956b3c6053
No known key found for this signature in database
GPG key ID: 5F7554F8C26DACCB
1 changed files with 6 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

9
voussoirkit/betterhelp.py
View file

@ -1,5 +1,8 @@
import argparse
import colorama
try:
import colorama
except ImportError:
colorama = None
import io
import os
import re
@ -131,7 +134,7 @@ def make_helptext(
# Even though this text is going out on stderr, we only colorize it if
# both stdout and stderr are tty because as soon as pipe buffers are
# involved, even on stdout, things start to get weird.
if do_colors and pipeable.stdout_tty() and pipeable.stderr_tty():
if do_colors and colorama and pipeable.stdout_tty() and pipeable.stderr_tty():
colorama.init()
color = dotdict.DotDict(
positional=colorama.Style.BRIGHT + colorama.Fore.CYAN,
@ -148,7 +151,7 @@ def make_helptext(
flag='',
command='',
reset='',
required_asterisk=colorama.Style.BRIGHT + colorama.Fore.RED + '(*)' + colorama.Style.RESET_ALL,
required_asterisk='(*)',
)
if program_name is None: