模块化重构代码

quickgit/utils.py

@@ -0,0 +1,150 @@
+"""
+工具类模块 - 提供命令执行、输出格式化等工具函数
+"""
+
+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']