parchlinux-mobile/parchbootstrap
1
0
Fork 0
forked from parchlinux-mobile/kupferbootstrap
Code Issues Pull requests Activity

utils: add colors_supported() and color_str() for terminal colors

Browse source
This commit is contained in:
InsanePrawn 2023-01-04 04:08:48 +01:00
parent 932e739255
commit e269841038
1 changed files with 13 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

13
utils.py
View file

@ -1,4 +1,5 @@
import atexit import atexit
import click
import datetime import datetime
import grp import grp
import hashlib import hashlib
@ -182,3 +183,15 @@ def ellipsize(s: str, length: int = 25, padding: Optional[str] = None, ellipsis:
return s return s
pad = s.rjust if rjust else s.ljust pad = s.rjust if rjust else s.ljust
return pad(length, padding) return pad(length, padding)
def colors_supported(force_colors: Optional[bool] = None) -> bool:
"If force_colors is None, returns isatty(stdout)"
# stdout is fd 1
return force_colors if force_colors is not None else os.isatty(1)
def color_str(s: str, use_colors: Optional[bool] = None, **kwargs) -> str:
if colors_supported(use_colors):
return click.style(s, **kwargs)
return s