mirror of
https://github.com/caelestia-dots/cli.git
synced 2026-06-16 05:49:59 -05:00
2 * r + 2 * t
89 lines
2.2 KiB
Python
89 lines
2.2 KiB
Python
import sys
|
|
from typing import Never
|
|
|
|
LOG_COLOUR: int = 2
|
|
INFO_COLOUR: int = 0
|
|
PROMPT_COLOUR: int = 36
|
|
WARNING_COLOUR: int = 33
|
|
ERROR_COLOUR: int = 31
|
|
|
|
_disable_input: bool = False
|
|
|
|
|
|
def disable_input() -> None:
|
|
global _disable_input
|
|
_disable_input = True
|
|
|
|
|
|
def log_exception(func):
|
|
"""Log exceptions to stdout instead of raising
|
|
|
|
Used by the `apply_()` functions so that an exception, when applying
|
|
a theme, does not prevent the other themes from being applied.
|
|
"""
|
|
|
|
def wrapper(*args, **kwargs):
|
|
try:
|
|
func(*args, **kwargs)
|
|
except Exception as e:
|
|
error(f'exception during "{func.__name__}()": {str(e)}')
|
|
|
|
return wrapper
|
|
|
|
|
|
def format_msg(colour: int, prefix: bool, msg: str) -> str:
|
|
return f"\033[{colour}m{':: ' if prefix else ''}{msg}\033[0m"
|
|
|
|
|
|
def log(msg: str, prefix: bool = True) -> None:
|
|
print(format_msg(LOG_COLOUR, prefix, msg))
|
|
|
|
|
|
def info(msg: str, prefix: bool = True) -> None:
|
|
print(format_msg(INFO_COLOUR, prefix, msg))
|
|
|
|
|
|
def warn(msg: str, prefix: bool = True) -> None:
|
|
print(format_msg(WARNING_COLOUR, prefix, f"Warning: {msg}"))
|
|
|
|
|
|
def error(err: str | Exception, prefix: bool = True) -> None:
|
|
print(format_msg(ERROR_COLOUR, prefix, f"Error: {err}"), file=sys.stderr)
|
|
|
|
|
|
def fatal(err: str | Exception, prefix: bool = True) -> Never:
|
|
print(format_msg(ERROR_COLOUR, prefix, f"Fatal: {err}"), file=sys.stderr)
|
|
sys.exit(1)
|
|
|
|
|
|
def _input(prompt: str) -> str:
|
|
if _disable_input:
|
|
print(prompt, end="")
|
|
return ""
|
|
|
|
try:
|
|
return input(prompt)
|
|
except (KeyboardInterrupt, EOFError):
|
|
print()
|
|
raise KeyboardInterrupt()
|
|
|
|
|
|
def prompt(msg: str, prefix: bool = True, end: str = " ") -> str:
|
|
return _input(format_msg(PROMPT_COLOUR, prefix, msg) + end)
|
|
|
|
|
|
def confirm(msg: str, prefix: bool = True, default: bool = True) -> bool:
|
|
suffix = " [Y/n]" if default else " [y/N]"
|
|
answer = prompt(msg + suffix, prefix=prefix).strip().lower()
|
|
if not answer:
|
|
return default
|
|
return answer in ("y", "yes")
|
|
|
|
|
|
def pause() -> None:
|
|
if _disable_input:
|
|
return
|
|
|
|
_input("\n\033[2m\033[3m(Ctrl+C to exit, enter to continue)\033[0m")
|
|
print("\033[1A\r\033[2K\033[1A\r\033[2K", end="") # Clear pause prompt
|