151 lines
3.8 KiB
Python
151 lines
3.8 KiB
Python
"""
|
||
工具类模块 - 提供命令执行、输出格式化等工具函数
|
||
"""
|
||
|
||
import subprocess
|
||
|
||
|
||
class Colors:
|
||
"""控制台颜色"""
|
||
HEADER = '\033[95m'
|
||
BLUE = '\033[94m'
|
||
CYAN = '\033[96m'
|
||
GREEN = '\033[92m'
|
||
WARNING = '\033[93m'
|
||
FAIL = '\033[91m'
|
||
ENDC = '\033[0m'
|
||
BOLD = '\033[1m'
|
||
|
||
|
||
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):
|
||
"""打印标题"""
|
||
print(f"\n{Colors.HEADER}{Colors.BOLD}{'='*50}")
|
||
print(f" {text}")
|
||
print(f"{'='*50}{Colors.ENDC}\n")
|
||
|
||
@staticmethod
|
||
def success(text: str):
|
||
"""打印成功信息"""
|
||
print(f"{Colors.GREEN}✓ {text}{Colors.ENDC}")
|
||
|
||
@staticmethod
|
||
def error(text: str):
|
||
"""打印错误信息"""
|
||
print(f"{Colors.FAIL}✗ {text}{Colors.ENDC}")
|
||
|
||
@staticmethod
|
||
def info(text: str):
|
||
"""打印提示信息"""
|
||
print(f"{Colors.CYAN}ℹ {text}{Colors.ENDC}")
|
||
|
||
@staticmethod
|
||
def warning(text: str):
|
||
"""打印警告信息"""
|
||
print(f"{Colors.WARNING}⚠ {text}{Colors.ENDC}")
|
||
|
||
|
||
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']
|