From b1f26d94e63a2af77803c364e137faad14b1a67e Mon Sep 17 00:00:00 2001 From: Anton Hvornum Date: Mon, 15 May 2023 15:50:11 +0200 Subject: Allowing for fg='color' in info, warn, error and debug. (#1818) * Allowing for fg='color' in info, warn, error and debug. * Converted to a more static definition, mypy does not like magic. * Flake8 fixes --- archinstall/lib/output.py | 46 ++++++++++++++++++++++++++++++++++++---------- 1 file changed, 36 insertions(+), 10 deletions(-) (limited to 'archinstall/lib') diff --git a/archinstall/lib/output.py b/archinstall/lib/output.py index bd31b5b3..0395e2e7 100644 --- a/archinstall/lib/output.py +++ b/archinstall/lib/output.py @@ -238,20 +238,46 @@ def _stylize_output( return f'\033[{ansi}m{text}\033[0m' -def info(*msgs: str): - log(*msgs, level=logging.INFO) - - -def debug(*msgs: str): - log(*msgs, level=logging.DEBUG) +def info( + *msgs: str, + level: int = logging.INFO, + fg: str = 'white', + bg: Optional[str] = None, + reset: bool = False, + font: List[Font] = [] +): + log(*msgs, level=level, fg=fg, bg=bg, reset=reset, font=font) +def debug( + *msgs: str, + level: int = logging.DEBUG, + fg: str = 'white', + bg: Optional[str] = None, + reset: bool = False, + font: List[Font] = [] +): + log(*msgs, level=level, fg=fg, bg=bg, reset=reset, font=font) -def error(*msgs: str): - log(*msgs, level=logging.ERROR, fg='red') +def error( + *msgs: str, + level: int = logging.ERROR, + fg: str = 'red', + bg: Optional[str] = None, + reset: bool = False, + font: List[Font] = [] +): + log(*msgs, level=level, fg=fg, bg=bg, reset=reset, font=font) -def warn(*msgs: str): - log(*msgs, level=logging.WARNING, fg='yellow') +def warn( + *msgs: str, + level: int = logging.WARN, + fg: str = 'yellow', + bg: Optional[str] = None, + reset: bool = False, + font: List[Font] = [] +): + log(*msgs, level=level, fg=fg, bg=bg, reset=reset, font=font) def log( -- cgit v1.2.3-54-g00ecf