270 lines
7.6 KiB
Python
270 lines
7.6 KiB
Python
"""
|
||
工具类模块 - 提供命令执行、输出格式化等工具函数
|
||
"""
|
||
|
||
import os
|
||
import subprocess
|
||
import sys
|
||
|
||
|
||
class Colors:
|
||
"""控制台颜色"""
|
||
# 基础颜色
|
||
HEADER = '\033[95m' # 紫色
|
||
BLUE = '\033[94m' # 蓝色
|
||
CYAN = '\033[96m' # 青色
|
||
GREEN = '\033[92m' # 绿色
|
||
YELLOW = '\033[93m' # 黄色
|
||
RED = '\033[91m' # 红色
|
||
MAGENTA = '\033[35m' # 品红
|
||
WHITE = '\033[97m' # 白色
|
||
|
||
# 亮色系
|
||
BRIGHT_CYAN = '\033[96m\033[1m' # 亮青色
|
||
BRIGHT_GREEN = '\033[92m\033[1m' # 亮绿色
|
||
BRIGHT_YELLOW = '\033[93m\033[1m' # 亮黄色
|
||
BRIGHT_RED = '\033[91m\033[1m' # 亮红色
|
||
BRIGHT_BLUE = '\033[94m\033[1m' # 亮蓝色
|
||
BRIGHT_MAGENTA = '\033[95m\033[1m' # 亮品红
|
||
|
||
# 背景色
|
||
BG_BLUE = '\033[44m'
|
||
BG_GREEN = '\033[42m'
|
||
BG_YELLOW = '\033[43m'
|
||
BG_RED = '\033[41m'
|
||
BG_CYAN = '\033[46m'
|
||
BG_MAGENTA = '\033[45m'
|
||
|
||
# 样式
|
||
BOLD = '\033[1m'
|
||
UNDERLINE = '\033[4m'
|
||
REVERSE = '\033[7m'
|
||
ENDC = '\033[0m'
|
||
|
||
# 兼容旧代码
|
||
WARNING = '\033[93m'
|
||
FAIL = '\033[91m'
|
||
|
||
|
||
class CommandExecutor:
|
||
"""命令执行器"""
|
||
|
||
@staticmethod
|
||
def run(command: str, show_output: bool = True) -> tuple[bool, str]:
|
||
"""
|
||
执行命令
|
||
|
||
Args:
|
||
command: 要执行的命令
|
||
show_output: 是否显示输出
|
||
|
||
Returns:
|
||
(是否成功, 输出内容)
|
||
"""
|
||
try:
|
||
result = subprocess.run(
|
||
command,
|
||
shell=True,
|
||
capture_output=True,
|
||
text=True,
|
||
encoding='utf-8',
|
||
errors='ignore'
|
||
)
|
||
|
||
output = result.stdout + result.stderr
|
||
|
||
if show_output and output.strip():
|
||
print(output)
|
||
|
||
return result.returncode == 0, output
|
||
except Exception as e:
|
||
OutputFormatter.error(f"命令执行失败: {str(e)}")
|
||
return False, str(e)
|
||
|
||
|
||
class OutputFormatter:
|
||
"""输出格式化器"""
|
||
|
||
@staticmethod
|
||
def header(text: str):
|
||
"""打印标题 - 顶部大标题"""
|
||
line = "=" * 60
|
||
print(f"\n{Colors.BRIGHT_CYAN}{line}")
|
||
print(f"{Colors.BRIGHT_MAGENTA}{Colors.BOLD}>> {text}")
|
||
print(f"{Colors.BRIGHT_CYAN}{line}{Colors.ENDC}")
|
||
|
||
@staticmethod
|
||
def section(text: str):
|
||
"""打印分节标题"""
|
||
print(f"\n{Colors.BRIGHT_YELLOW}{'=' * 60}")
|
||
print(f"{Colors.BRIGHT_BLUE}{Colors.BOLD}>> {text}")
|
||
print(f"{Colors.BRIGHT_YELLOW}{'=' * 60}{Colors.ENDC}")
|
||
|
||
@staticmethod
|
||
def divider():
|
||
"""打印分割线"""
|
||
print(f"{Colors.CYAN}{'-' * 60}{Colors.ENDC}")
|
||
|
||
@staticmethod
|
||
def thin_divider():
|
||
"""打印细分割线"""
|
||
print(f"{Colors.CYAN}{'·' * 60}{Colors.ENDC}")
|
||
|
||
@staticmethod
|
||
def success(text: str):
|
||
"""打印成功信息"""
|
||
print(f"{Colors.BRIGHT_GREEN}[√]{Colors.ENDC} {Colors.GREEN}{text}{Colors.ENDC}")
|
||
|
||
@staticmethod
|
||
def error(text: str):
|
||
"""打印错误信息"""
|
||
print(f"{Colors.BRIGHT_RED}[×]{Colors.ENDC} {Colors.RED}{text}{Colors.ENDC}")
|
||
|
||
@staticmethod
|
||
def info(text: str):
|
||
"""打印提示信息"""
|
||
print(f"{Colors.BRIGHT_CYAN}[i]{Colors.ENDC} {Colors.CYAN}{text}{Colors.ENDC}")
|
||
|
||
@staticmethod
|
||
def warning(text: str):
|
||
"""打印警告信息"""
|
||
print(f"{Colors.BRIGHT_YELLOW}[!]{Colors.ENDC} {Colors.YELLOW}{text}{Colors.ENDC}")
|
||
|
||
@staticmethod
|
||
def step(num: int, text: str):
|
||
"""打印步骤信息"""
|
||
print(f"{Colors.BRIGHT_BLUE}[{num}]{Colors.ENDC} {Colors.WHITE}{text}{Colors.ENDC}")
|
||
|
||
@staticmethod
|
||
def menu_item(num: int, text: str, icon: str = ">>"):
|
||
"""打印菜单项"""
|
||
print(f"{Colors.BRIGHT_CYAN}[{num}]{Colors.ENDC} {Colors.WHITE}{text}{Colors.ENDC}")
|
||
|
||
@staticmethod
|
||
def key_value(key: str, value: str, key_color=None, value_color=None):
|
||
"""打印键值对"""
|
||
if key_color is None:
|
||
key_color = Colors.BRIGHT_CYAN
|
||
if value_color is None:
|
||
value_color = Colors.WHITE
|
||
print(f"{key_color}{key}:{Colors.ENDC} {value_color}{value}{Colors.ENDC}")
|
||
|
||
@staticmethod
|
||
def status(status_type: str, text: str):
|
||
"""打印状态信息"""
|
||
icons = {
|
||
'pending': f"{Colors.YELLOW}[·]{Colors.ENDC}",
|
||
'running': f"{Colors.BRIGHT_CYAN}[>]{Colors.ENDC}",
|
||
'success': f"{Colors.BRIGHT_GREEN}[√]{Colors.ENDC}",
|
||
'error': f"{Colors.BRIGHT_RED}[×]{Colors.ENDC}",
|
||
}
|
||
icon = icons.get(status_type, "[·]")
|
||
print(f"{icon} {text}")
|
||
|
||
|
||
class PlatformUtils:
|
||
"""跨平台工具类"""
|
||
|
||
@staticmethod
|
||
def clear_screen():
|
||
"""清屏 - 跨平台兼容"""
|
||
# Windows
|
||
if os.name == 'nt':
|
||
os.system('cls')
|
||
# Linux/macOS
|
||
else:
|
||
os.system('clear')
|
||
|
||
@staticmethod
|
||
def is_windows() -> bool:
|
||
"""检查是否为 Windows 系统"""
|
||
return os.name == 'nt'
|
||
|
||
@staticmethod
|
||
def is_linux() -> bool:
|
||
"""检查是否为 Linux 系统"""
|
||
return sys.platform.startswith('linux')
|
||
|
||
@staticmethod
|
||
def is_mac() -> bool:
|
||
"""检查是否为 macOS 系统"""
|
||
return sys.platform == 'darwin'
|
||
|
||
@staticmethod
|
||
def get_platform_name() -> str:
|
||
"""获取平台名称"""
|
||
if PlatformUtils.is_windows():
|
||
return "Windows"
|
||
elif PlatformUtils.is_linux():
|
||
return "Linux"
|
||
elif PlatformUtils.is_mac():
|
||
return "macOS"
|
||
else:
|
||
return "Unknown"
|
||
|
||
|
||
class InputValidator:
|
||
"""输入验证器"""
|
||
|
||
@staticmethod
|
||
def get_choice(prompt: str, valid_range: range) -> int:
|
||
"""
|
||
获取用户选择（数字）
|
||
|
||
Args:
|
||
prompt: 提示信息
|
||
valid_range: 有效范围
|
||
|
||
Returns:
|
||
用户选择的数字
|
||
"""
|
||
while True:
|
||
try:
|
||
choice = input(prompt).strip()
|
||
choice_num = int(choice)
|
||
if choice_num in valid_range:
|
||
return choice_num
|
||
else:
|
||
OutputFormatter.error(f"请输入 {valid_range.start}-{valid_range.stop-1} 之间的数字")
|
||
except ValueError:
|
||
OutputFormatter.error("请输入有效的数字")
|
||
|
||
@staticmethod
|
||
def get_input(prompt: str, allow_empty: bool = False) -> str:
|
||
"""
|
||
获取用户输入
|
||
|
||
Args:
|
||
prompt: 提示信息
|
||
allow_empty: 是否允许空输入
|
||
|
||
Returns:
|
||
用户输入的字符串
|
||
"""
|
||
while True:
|
||
user_input = input(prompt).strip()
|
||
if user_input or allow_empty:
|
||
return user_input
|
||
else:
|
||
OutputFormatter.error("输入不能为空")
|
||
|
||
@staticmethod
|
||
def confirm(prompt: str, default: bool = False) -> bool:
|
||
"""
|
||
获取用户确认
|
||
|
||
Args:
|
||
prompt: 提示信息
|
||
default: 默认值
|
||
|
||
Returns:
|
||
用户确认结果
|
||
"""
|
||
suffix = " [Y/n]: " if default else " [y/N]: "
|
||
user_input = input(prompt + suffix).strip().lower()
|
||
|
||
if not user_input:
|
||
return default
|
||
|
||
return user_input in ['y', 'yes']
|