Arctic/SSH-key-Manager
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

refactored code

Browse source
This commit is contained in:
Arctic 2025-03-08 00:43:33 -06:00
parent 73429771d4
commit 414266eefc
6 changed files with 315 additions and 404 deletions
Download patch file Download diff file Expand all files Collapse all files

35
utils.py
View file

@ -1,25 +1,34 @@
# ssh_manager/utils.py
from enum import Enum
import sys
from functools import lru_cache
class Colors:
GREEN = "\033[0;32m"
RED = "\033[0;31m"
class Colors(Enum):
GREEN = "\033[0;32m"
RED = "\033[0;31m"
YELLOW = "\033[1;33m"
CYAN = "\033[0;36m"
BOLD = "\033[1m"
RESET = "\033[0m"
CYAN = "\033[0;36m"
BOLD = "\033[1m"
RESET = "\033[0m"
def print_error(message):
print(f"{Colors.RED}{Colors.BOLD}[✖] {Colors.RESET}{message}")
def __str__(self):
return self.value
def print_warning(message):
print(f"{Colors.YELLOW}{Colors.BOLD}[⚠] {Colors.RESET}{message}")
@lru_cache(maxsize=32)
def _format_message(prefix: str, color: Colors, message: str) -> str:
return f"{color}{Colors.BOLD}[{prefix}] {Colors.RESET}{message}"
def print_info(message):
print(f"{Colors.GREEN}{Colors.BOLD}[✔] {Colors.RESET}{message}")
def print_error(message: str) -> None:
print(_format_message("✖", Colors.RED, message))
def safe_input(prompt=""):
def print_warning(message: str) -> None:
print(_format_message("⚠", Colors.YELLOW, message))
def print_info(message: str) -> None:
print(_format_message("✔", Colors.GREEN, message))
def safe_input(prompt: str = "") -> str:
"""
A wrapper around input() that exits the entire script on Ctrl+C.
"""